]j` d dl Z d dlmZmZ d dlmZ d dlZd dlmZ d dl m
Z
dZ dZ dZ
G d d
e Z G d d Zd
Ze e e
fddedee defdZdedee fdZdedee fdZdS ) N)IterableList)deque)
DictCursor) load_fastz/etc/psa/psa.confz /etc/my.cnfz/etc/mysql/my.cnfc e Zd ZdZdS )
MySQLErrorz4
Exception related to operation with MySQL.
N)__name__
__module__ __qualname____doc__ r/builddir/build/BUILD/imunify360-venv-2.6.3/opt/imunify360/venv/lib/python3.11/site-packages/clcommon/mysql_lib.pyr r s r r c
e Zd ZddefdZd Zd Zd ZddZdd
Z dd Z
d Z e d
Z
ddedee d z deedf eeef z fdZedededd fd Zd S )MySQLConnectorFas_dictc | | dd | d | _ |rt nd | _ || _ d S )Nhost localhost) add_unix_socket_if_localhostget _connectionr _cursor_type_connect_kwargs)selfr kwargss r __init__zMySQLConnector.__init__ sM ))&**V[*I*I6RRR *1 ;JJt % r c . | d S Ncloser s r __del__zMySQLConnector.__del__&
r c * | S r )connectr# s r __enter__zMySQLConnector.__enter__) s ||~~ r c . | d S r r! )r exc_type exc_value tracebacks r __exit__zMySQLConnector.__exit__, r% r returnc | j | S t j di | j | _ | S # t j $ r"}t t
| |d }~ww xY w)Nr )r pymysqlr' r Errorr str)r es r r' zMySQLConnector.connect/ sl 'K , & F F1E F FD K} , , , SVV$$! + ,s ( A AANc X | j "| j d | _ d S d S r )r r" r# s r r" zMySQLConnector.close8 s6 ' " " $ $ $ #D ( 'r c J | j | j d S d S r )r commitr# s r r6 zMySQLConnector.commit= s. ' # # % % % % % ( 'r c @ | j | j S r )
connectioncursorr r# s r r9 zMySQLConnector.cursorA s %%d&7888r c F | j | | j S r )r r' r# s r r8 zMySQLConnector.connectionD s! # LLNNN r sql_queryargs.c | j | j 5 }| || | cddd S # 1 swxY w Y dS )z:
Execute SQL query and return the result.
)r< N)r8 r9 r executefetchall)r r; r< r9 s r
execute_queryz MySQLConnector.execute_queryJ s _
#
#D$5
6
6 %& NN94N 0 0 0??$$ % % % % % % % % % % % % % % % % % %s +AA A r r c 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_localhostV s= ; =#>#>$8$:$:F= ! ! ! #>#>r )F)r. r )r. Nr )r
r r boolr r$ r( r- r' r" r6 r9 propertyr8 r2 r objecttupledictr@ staticmethodr r r r r r sW & & & & & & , , , ,$ $ $ $
& & & &9 9 9 X )-
%
%
% v %
%
vs{ d3; / /
%
%
%
% ;3 ; ; ; ; ; \ ; ; ;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_urirM e s# //#)DDD Jr )plesk_config_path
mysql_configsrN rO r. c t j | r*t | d }| d }||S t | }t
| dk r| }t j | sGt |d }| d }||S | t | t
| dk d S )
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_SOCKETNr T) strip_quotessocketz/var/lib/mysql/mysql.sock)
ospathisfiler r r lenpopleftextend
get_cnf_paths)rN rO psa_confpsa_conf_socketmysql_configs_queue my_conf_pathmy_confmy_conf_sockets r rC rC m s
w~~'(( # .#>>>",,~66 &" "
..
!
"
"Q
&
&*2244 w~~l++ Lt<<< X.. %! ! ""= #>#>??? !
"
"Q
&
& ' &r rW c g }g } t | ddd 5 }|D ] }| } | d r\| d }t j | r| | np| d r[| d }t j | r| | # t $ r Y w xY w|D ]}|t | z
} ddd n # 1 swxY w Y n/# t t f$ r}t d | d
| |d}~ww xY w|S ) z
Get all included config files paths in the given config file
marked with !include and !includedir.
Returns an empty list if there are no paths.
rzutf-8surrogateescape)encodingerrorsz !includedirz !include z!includeNzError reading config file z: )openstrip
startswith removeprefixrV rW isdirappendrX
IndexErrorget_cnf_files_in_dirOSErrorIOErrorr ) rW include_dirs
list_of_paths config_filelineincluded_dir_pathincluded_file_pathdir_pathr3 s r r\ r\ s !L !MJ
$g6G
H
H
H @K#
zz|| }55 E,0,=,=m,L,L,R,R,T,T) 7==):;; C ( / /0A B B B 55 E-1->->z-J-J-P-P-R-R* 7>>*<== E ) 0 01C D D D! D )
@
@ !5h!?!??
@ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @"
W J J JAdAAaAABBIJ s^ E E C!DE
D$ E #D$$E ? E EE EE F(E>>Fc g }t j | rt j | D ]}t j t j | | r\t j | d dk r3| t j | | |S )zT
Get paths list in given dir
Returns an empty list if there is no paths
z.cnf)rV rW rl listdirrX joinsplitextrm )rW rs items r ro ro s
M w}}T ?Jt$$ ? ?Dw~~bgll4 6 677
?BG<L<LT<R<RST<UY_<_<_ $$RW\\$%=%=>>> r )rV typingr r collectionsr r0 pymysql.cursorsr clcommon.clconfparsr PLESK_CONFIG_PATHMY_CONFIG_PATHMY_CONFIG_PATH2 Exceptionr r rM r2 rC listr\ ro r r r <module>r s
! ! ! ! ! ! ! ! & & & & & & ) ) ) ) ) )' % E; E; E; E; E; E; E; E;P /$2O#D ' ' ' ' C= ' ' ' ' 'F S B
s
tCy
r