File: //opt/alt/python37/lib/python3.7/site-packages/clcommon/__pycache__/mysql_lib.cpython-37.pyc
B
wi�"��в� @ s� d dl mZ d dlZd dlmZ y d dlZW nF ek
rr y d dlZW n ek
r` dZY nX d dlm Z Y nX d dl
m Z d dl m Z d dl
mZ dZdZd ZG d
d � d �Zd d
� Zeeefd�dddd�dd�ZdS )� )� annotationsN)�Iterable)�
DictCursor)� load_fast)� NoPackagez/etc/psa/psa.confz /etc/my.cnfz/etc/mysql/my.cnfc @ s� e Zd Zd!dd�dd�Zdd� Zdd � Zd d
�d d �Zd
d
�dd�Zd
d
�dd�Zdd� Z e
dd� �Z d"dddd�dd �Z e
dd d
d �d d ��ZdS )#�MySQLConnectorF�bool)�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.4.1-1.el9.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__1 s zMySQLConnector.__enter__c C s | � � d S )N)�close)r �exc_type� exc_value� tracebackr r r �__exit__4 s zMySQLConnector.__exit__)�returnc C s t jf | j�| _| S )N)r r r r )r r r r r 7 s zMySQLConnector.connect�Nonec C s | j d k r| j �� d S )N)r r )r r r r r ; 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" C s zMySQLConnector.cursorc C s | j d kr| �� | j S )N)r r )r r r r r! F s
zMySQLConnector.connectionN�strzIterable[object] | Nonez&tuple[object, ...] | dict[str, object])� 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_queryL s z MySQLConnector.execute_query�dict)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
T s z+MySQLConnector.add_unix_socket_if_localhost)F)N)�__name__�
__module__� __qualname__r r r r r r r" �propertyr! r& � staticmethodr
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_uric s r. )�plesk_config_path�
mysql_configsr# z
Iterable[str])r/ r0 r c C sp t j�| �r.t| dd�}|�d�}|dk r.|S x<|D ]4}t j�|�sFq4t|dd�}|�d�}|dk r4|S q4W 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_SOCKETNT)Z strip_quotesZsocketz/var/lib/mysql/mysql.sock)�os�path�isfiler r )r/ r0 Zpsa_confZpsa_conf_socketZ my_conf_pathZmy_confZmy_conf_socketr r r r( k s
r( )Z
__future__r r2 �typingr r � ImportErrorZMySQLdbZMySQLdb.cursorsr Zpymysql.cursorsZclcommon.clconfparsr Z clcommon.cpapi.cpapiexceptionsr ZPLESK_CONFIG_PATHZMY_CONFIG_PATHZMY_CONFIG_PATH2r r. r( r r r r �<module> s(
@