File: //opt/cloudlinux/venv/lib/python3.11/site-packages/clcommon/__pycache__/mysql_lib.cpython-311.pyc
�
�>1f�
� � � d dl mZ d dlZd dlmZ d dlZd dlmZ d dlm Z dZ
dZ dZ G d � d
ej
� � Z G d � d � � Zd
� Ze
e e fd�dd�ZdS )� )� annotationsN)�Iterable)�
DictCursor)� load_fastz/etc/psa/psa.confz /etc/my.cnfz/etc/mysql/my.cnfc � � e Zd ZdS )�
MySQLErrorN)�__name__�
__module__� __qualname__� � ��/builddir/build/BUILDROOT/alt-python27-cllib-3.4.3-1.el8.cloudlinux.x86_64/opt/cloudlinux/venv/lib/python3.11/site-packages/clcommon/mysql_lib.pyr r s � � � � � ��Dr
r c �z � e Zd Zddd�Zd� Zd� Zdd�Zd d
�Zd d �Zd � Z e
d
� � � Z d d d�Z e
d d�� � ZdS ) �MySQLConnectorF�as_dict�boolc � � | � |� dd� � |� � d | _ |rt nd | _ || _ d S )N�host� localhost)� add_unix_socket_if_localhost�get� _connectionr � _cursor_type�_connect_kwargs)�selfr �kwargss r �__init__zMySQLConnector.__init__ sM � � �)�)�&�*�*�V�[�*I�*I�6�R�R�R� ���*1� ;�J�J�t��� %�� � � r
c �* � | � � � S �N)�connect�r s r � __enter__zMySQLConnector.__enter__$ s � ��|�|�~�~� r
c �. � | � � � d S r )�close)r �exc_type� exc_value� tracebacks r �__exit__zMySQLConnector.__exit__' s � � �
�
� � � � � r
�returnc �< � t j di | j ��| _ | S )Nr )�pymysqlr r r r! s r r zMySQLConnector.connect* s" � �"�?�B�B�T�-A�B�B���� r
�Nonec �J � | j �| j � � � d S d S r )r r$ r! s r r$ zMySQLConnector.close. s. � � � � '� � � "� "� $� $� $� $� $� (� 'r
c �J � | j �| j � � � d S d S r )r �commitr! s r r/ zMySQLConnector.commit2 s. � � � � '� � � #� #� %� %� %� %� %� (� 'r
c �@ � | j � | j � � S r )�
connection�cursorr r! s r r2 zMySQLConnector.cursor6 s � ���%�%�d�&7�8�8�8r
c �F � | j �| � � � | j S r )r r r! s r r1 zMySQLConnector.connection9 s! � � � � #� �L�L�N�N�N�� � r
N� sql_query�str�args�Iterable[object] | None�&tuple[object, ...] | dict[str, object]c �� � | j � | j � � 5 }|� ||�� � |� � � cddd� � S # 1 swxY w Y dS )z:
Execute SQL query and return the result.
)r6 N)r1 r2 r �execute�fetchall)r r4 r6 r2 s r �
execute_queryz MySQLConnector.execute_query? s� � � �_�
#�
#�D�$5�
6�
6� %�&� �N�N�9�4�N� 0� 0� 0��?�?�$�$� %� %� %� %� %� %� %� %� %� %� %� %���� %� %� %� %� %� %s �+A�A � A r r �dictc �D � | dk rd|vrt � � |d< dS dS 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 � unix_socketN)�get_unix_socket_path)r r s r r z+MySQLConnector.add_unix_socket_if_localhostG s= � � �;� � �=��#>�#>�$8�$:�$:�F�=� !� !� !� � �#>�#>r
)F)r r )r) r )r) r, r )r4 r5 r6 r7 r) r8 )r r5 r r= r) r, )r r
r r r" r( r r$ r/ r2 �propertyr1 r< � staticmethodr r r
r r r s� � � � � � �&� &� &� &� &� � � �� � �� � � �%� %� %� %�&� &� &� &�9� 9� 9� � � � �X� �
%� %� %� %� %� � ;� ;� ;� �\� ;� ;� ;r
r c �P � t � | j | j � � | S )zF
Get a modified RFC 1738 URL for a MySQL database connection.
)r r r �query)�urls r �get_rfc1738_db_urirF V s# � � �/�/���#�)�D�D�D� �Jr
)�plesk_config_path�
mysql_configsrG r5 rH �
Iterable[str]r) c �: � t j � | � � r*t | d�� � }|� d� � }|�|S |D ]N}t j � |� � s�"t |d�� � }|� d� � }|�|c S �OdS ) 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.
� )� delimiter� MYSQL_SOCKETNT)� strip_quotes�socketz/var/lib/mysql/mysql.sock)�os�path�isfiler r )rG rH �psa_conf�psa_conf_socket� my_conf_path�my_conf�my_conf_sockets r r@ r@ ^ s� � �
�w�~�~�'�(�(� #� � .�#�>�>�>��"�,�,�~�6�6�� � &�"� "�%� "� "� ��w�~�~�l�+�+� � ��L�t�<�<�<�� ���X�.�.�� � %�!� !� !� !� &� '� &r
)rG r5 rH rI r) r5 )�
__future__r rP �typingr r+ �pymysql.cursorsr �clcommon.clconfparsr �PLESK_CONFIG_PATH�MY_CONFIG_PATH�MY_CONFIG_PATH2�Errorr r rF r@ r r
r �<module>r` s
�� #� "� "� "� "� "� � � � � � � � � � � ���� &� &� &� &� &� &� )� )� )� )� )� )�'� � ��%�� � � � � �� � � � �8;� 8;� 8;� 8;� 8;� 8;� 8;� 8;�v� � � /�$2�O�#D�'� '� '� '� '� '� '� 'r