Edit File: mysql_lib.cpython-37.pyc
B � e � @ s� d dl Z d dlmZ y d dlZW nF ek rf y d dlZW n ek rT dZY nX d dlmZ Y nX d dlmZ d dl m Z d dl m Z dZ dZdZG d d � d �Zd d � Zed �dd�ZdS )� N)�Optional)� DictCursor)� load_fast)� NoPackagez/etc/psa/psa.confz /etc/my.cnfz/etc/mysql/my.cnfc @ s� e Zd Zded�dd�Zdd� Zdd� Zd d �d d �Zd d �d d�Zd d �dd�Z dd� Z e dd� �Z d e ee ed�dd�Zee ed d�dd��Zd S ) �MySQLConnectorF)�as_dictc K sB t d krtd��| �|�dd�|� d | _|r2tnd | _|| _d S )NzFCan not connect to database; MySQL-client libraries are not installed.�host� localhost)�pymysqlr � add_unix_socket_if_localhost�get� _connectionr � _cursor_type�_connect_kwargs)�selfr �kwargs� r ��/builddir/build/BUILDROOT/alt-python27-cllib-3.3.5-1.el8.cloudlinux.x86_64/opt/alt/python37/lib/python3.7/site-packages/clcommon/mysql_lib.py�__init__# s zMySQLConnector.__init__c C s | � � S )N)�connect)r r r r � __enter__/ s zMySQLConnector.__enter__c C s | � � d S )N)�close)r �exc_type� exc_value� tracebackr r r �__exit__2 s zMySQLConnector.__exit__)�returnc C s t jf | j�| _| S )N)r r r r )r r r r r 5 s zMySQLConnector.connectNc C s | j d k r| j �� d S )N)r r )r r r r r 9 s zMySQLConnector.closec C s | j d k r| j �� d S )N)r �commit)r r r r r = s zMySQLConnector.commitc C s | j �| j�S )N)� connection�cursorr )r r r r r A s zMySQLConnector.cursorc C s | j d kr| �� | j S )N)r r )r r r r r D s zMySQLConnector.connection)� sql_query�argsr c C s0 | j �| j��}|j||d� |�� S Q R X dS )z: Execute SQL query and return the result. )r! N)r r r ZexecuteZfetchall)r r r! r r r r � execute_queryJ s z MySQLConnector.execute_query)r r r c C s | dkrd|krt � |d< dS )ag Add 'unix_socket' to kwargs if host is 'localhost'. It seems that when the host is set to 'localhost', establishing a connection through TCP/IP might encounter issues due to some MySQL configuration options. To prioritize a Unix socket connection, we should include the 'unix_socket' query parameter. r Z unix_socketN)�get_unix_socket_path)r r r r r r R s z+MySQLConnector.add_unix_socket_if_localhost)F)N)�__name__� __module__� __qualname__�boolr r r r r r r �propertyr �strr �tupler" � staticmethod�dictr r r r r r ! s r c C s t �| j| j� | S )zF Get a modified RFC 1738 URL for a MySQL database connection. )r r r Zquery)Zurlr r r �get_rfc1738_db_uria s r- )r c C sn t j�t�r.ttdd�} | �d�}|dk r.|S x:ttfD ].}t j�|�r8t|�}|�d�}|dk r8|S q8W dS )z� Get the Unix socket path for MySQL connection. Check Plesk and MySQL config files for the socket path. If found, return that path. If not, return default socket path. � )Z delimiterZ MYSQL_SOCKETNZsocketz/var/lib/mysql/mysql.sock)�os�path�isfile�PLESK_CONFIG_PATHr r �MY_CONFIG_PATH�MY_CONFIG_PATH2)Zpsa_confZpsa_conf_socketZ my_conf_pathZmy_confZmy_conf_socketr r r r# i s r# )r/ �typingr r � ImportErrorZMySQLdbZMySQLdb.cursorsr Zpymysql.cursorsZclcommon.clconfparsr Z clcommon.cpapi.cpapiexceptionsr r2 r3 r4 r r- r) r# r r r r �<module> s$ @