a
    LcòfB  ã                
   @   s`  d Z ddlmZ ddlmZ ddlmZ ddlm	Z	 ddl
mZ ddlmZ ddlmZ zdd	lZW n. ey’ Z zed
ƒe‚W Y d	Z[n
d	Z[0 0 ddlmZmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z  ddl!m"Z" ddl#m$Z$ ddl%m&Z& ej'Z(e(dk �r"edej) ƒ‚i e¥ej*ej+i¥Z,edƒZ-G dd„ dƒZ.G dd„ de	ƒZ/d	S )za
MySQL database backend for Django.

Requires mysqlclient: https://pypi.org/project/mysqlclient/
é    )ÚImproperlyConfigured)ÚIntegrityError)Úutils)ÚBaseDatabaseWrapper)Úasync_unsafe)Úcached_property)Ú_lazy_re_compileNz:Error loading MySQLdb module.
Did you install mysqlclient?)ÚCLIENTÚ
FIELD_TYPE)Úconversionsé   )ÚDatabaseClient)ÚDatabaseCreation)ÚDatabaseFeatures)ÚDatabaseIntrospection)ÚDatabaseOperations)ÚDatabaseSchemaEditor)ÚDatabaseValidation)r   é   é   z4mysqlclient 1.4.3 or newer is required; you have %s.z(\d{1,2})\.(\d{1,2})\.(\d{1,2})c                   @   s>   e Zd ZdZdZdd„ Zddd„Zdd	„ Zd
d„ Zdd„ Z	dS )ÚCursorWrappera6  
    A thin wrapper around MySQLdb's normal cursor class that catches particular
    exception instances and reraises them with the correct types.

    Implemented as a wrapper, rather than a subclass, so that it isn't stuck
    to the particular underlying representation returned by Connection.cursor().
    )i  iš  ië  i¹  c                 C   s
   || _ d S ©N)Úcursor©Úselfr   © r   úZ/home/httpd/docs/test/DocsMgr/lib/python3.9/site-packages/django/db/backends/mysql/base.pyÚ__init__E   s    zCursorWrapper.__init__Nc              
   C   s\   z| j  ||¡W S  tjyV } z,|jd | jv r@tt|jƒŽ ‚‚ W Y d }~n
d }~0 0 d S ©Nr   )r   ÚexecuteÚDatabaseÚOperationalErrorÚargsÚcodes_for_integrityerrorr   Útuple©r   Úqueryr"   Úer   r   r   r   H   s    zCursorWrapper.executec              
   C   s\   z| j  ||¡W S  tjyV } z,|jd | jv r@tt|jƒŽ ‚‚ W Y d }~n
d }~0 0 d S r   )r   Úexecutemanyr    r!   r"   r#   r   r$   r%   r   r   r   r(   S   s    zCursorWrapper.executemanyc                 C   s   t | j|ƒS r   )Úgetattrr   )r   Úattrr   r   r   Ú__getattr__]   s    zCursorWrapper.__getattr__c                 C   s
   t | jƒS r   )Úiterr   ©r   r   r   r   Ú__iter__`   s    zCursorWrapper.__iter__)N)
Ú__name__Ú
__module__Ú__qualname__Ú__doc__r#   r   r   r(   r+   r.   r   r   r   r   r   5   s   

r   c                       sl  e Zd ZdZdddddddd	d
ddddd
ddddddddddddddœZdZdddddddd ddddd!œZd"Zd#d$d%d&d'd(d)œZh d*£Z	e
Z
eZeZeZeZeZeZeZd+d,„ Zd-d.„ Zed/d0„ ƒZ‡ fd1d2„ZedPd4d5„ƒZd6d7„ Zd8d9„ Z d:d;„ Z!d<d=„ Z"dQd>d?„Z#d@dA„ Z$e%dBdC„ ƒZ&e%dDdE„ ƒZ'e%dFdG„ ƒZ(e%dHdI„ ƒZ)e%dJdK„ ƒZ*e%dLdM„ ƒZ+e%dNdO„ ƒZ,‡  Z-S )RÚDatabaseWrapperÚmysqlzinteger AUTO_INCREMENTzbigint AUTO_INCREMENTÚlongblobÚboolzvarchar(%(max_length)s)Údatezdatetime(6)z+numeric(%(max_digits)s, %(decimal_places)s)Zbigintzdouble precisionÚintegerzchar(15)zchar(39)Újsonzbigint UNSIGNEDzinteger UNSIGNEDzsmallint UNSIGNEDzsmallint AUTO_INCREMENTZsmallintÚlongtextztime(6)zchar(32))Z	AutoFieldZBigAutoFieldZBinaryFieldZBooleanFieldZ	CharFieldZ	DateFieldZDateTimeFieldZDecimalFieldZDurationFieldZ	FileFieldZFilePathFieldZ
FloatFieldZIntegerFieldZBigIntegerFieldZIPAddressFieldZGenericIPAddressFieldÚ	JSONFieldZOneToOneFieldÚPositiveBigIntegerFieldÚPositiveIntegerFieldÚPositiveSmallIntegerFieldZ	SlugFieldZSmallAutoFieldZSmallIntegerFieldZ	TextFieldZ	TimeFieldZ	UUIDField)	ZtinyblobZblobZ
mediumblobr5   ZtinytextÚtextZ
mediumtextr:   r9   z= %szLIKE %szLIKE BINARY %sz> %sz>= %sz< %sz<= %s)ÚexactZiexactÚcontainsÚ	icontainsÚgtZgteÚltZlteÚ
startswithÚendswithÚistartswithÚ	iendswithzCREPLACE(REPLACE(REPLACE({}, '\\', '\\\\'), '%%', '\%%'), '_', '\_')z"LIKE BINARY CONCAT('%%', {}, '%%')zLIKE CONCAT('%%', {}, '%%')zLIKE BINARY CONCAT({}, '%%')zLIKE CONCAT({}, '%%')zLIKE BINARY CONCAT('%%', {})zLIKE CONCAT('%%', {}))rA   rB   rE   rG   rF   rH   >   úread committedzread uncommittedZserializablezrepeatable readc                 C   s   | j S r   )Úmysql_versionr-   r   r   r   Úget_database_versionË   s    z$DatabaseWrapper.get_database_versionc                 C   s  t ddœ}| j}|d r$|d |d< |d r8|d |d< |d rL|d |d< |d	  d
¡rh|d	 |d< n|d	 r||d	 |d< |d r”t|d ƒ|d< tj|d< |d  ¡ }| dd¡}|rò| ¡ }|| j	vròt
d|d dd„ t| j	ƒD ƒ¡f ƒ‚|| _| |¡ |S )NÚutf8)ÚconvÚcharsetÚUSERÚuserÚNAMEZdatabaseZPASSWORDÚpasswordZHOSTú/Zunix_socketÚhostZPORTÚportZclient_flagÚOPTIONSÚisolation_levelrI   zKInvalid transaction isolation level '%s' specified.
Use one of %s, or None.z, c                 s   s   | ]}d | V  qdS )z'%s'Nr   )Ú.0Úsr   r   r   Ú	<genexpr>î   ó    z8DatabaseWrapper.get_connection_params.<locals>.<genexpr>)Údjango_conversionsÚsettings_dictrE   Úintr	   Z
FOUND_ROWSÚcopyÚpopÚlowerÚisolation_levelsr   ÚjoinÚsortedrW   Úupdate)r   Úkwargsr]   ÚoptionsrW   r   r   r   Úget_connection_paramsÎ   s@    þ

þþÿ
z%DatabaseWrapper.get_connection_paramsc                 C   s0   t jf i |¤Ž}|j t¡tu r,|j t¡ |S r   )r    ÚconnectÚencodersÚgetÚbytesr`   )r   Zconn_paramsÚ
connectionr   r   r   Úget_new_connectionõ   s    z"DatabaseWrapper.get_new_connectionc                    sz   t ƒ  ¡  g }| jjr | d¡ | jr:| d| j ¡  ¡ |rv|  ¡ � }| d 	|¡¡ W d   ƒ n1 sl0    Y  d S )NzSET SQL_AUTO_IS_NULL = 0z*SET SESSION TRANSACTION ISOLATION LEVEL %sz; )
ÚsuperÚinit_connection_stateÚfeaturesZis_sql_auto_is_null_enabledÚappendrW   Úupperr   r   rc   )r   Zassignmentsr   ©Ú	__class__r   r   rp      s    

ÿÿ
z%DatabaseWrapper.init_connection_stateNc                 C   s   | j  ¡ }t|ƒS r   )rm   r   r   )r   Únamer   r   r   r   Úcreate_cursor  s    
zDatabaseWrapper.create_cursorc                 C   s(   zt  | ¡ W n tjy"   Y n0 d S r   )r   Ú	_rollbackr    ZNotSupportedErrorr-   r   r   r   rx     s    zDatabaseWrapper._rollbackc                 C   s6   | j � | j |¡ W d   ƒ n1 s(0    Y  d S r   )Zwrap_database_errorsrm   Ú
autocommit)r   ry   r   r   r   Ú_set_autocommit  s    zDatabaseWrapper._set_autocommitc                 C   s6   |   ¡ �}| d¡ W d  ƒ n1 s(0    Y  dS )z¿
        Disable foreign key checks, primarily for use in adding rows with
        forward references. Always return True to indicate constraint checks
        need to be re-enabled.
        zSET foreign_key_checks=0NT)r   r   r   r   r   r   Údisable_constraint_checking#  s    
(z+DatabaseWrapper.disable_constraint_checkingc                 C   sX   d| j  | _ }z<|  ¡ �}| d¡ W d  ƒ n1 s80    Y  W || _ n|| _ 0 dS )zM
        Re-enable foreign key checks after they have been disabled.
        FzSET foreign_key_checks=1N)Úneeds_rollbackr   r   )r   r|   r   r   r   r   Úenable_constraint_checking-  s
    
*z*DatabaseWrapper.enable_constraint_checkingc           
      C   sÐ   |   ¡ �´}|du r| j |¡}|D ]Š}| j ||¡}|s:q"| j ||¡}| ¡ D ]Z\}\}}| d||||||||f ¡ | ¡ D ]&}	td||	d |||	d ||f ƒ‚q‚qPq"W d  ƒ n1 sÂ0    Y  dS )a]  
        Check each table name in `table_names` for rows with invalid foreign
        key references. This method is intended to be used in conjunction with
        `disable_constraint_checking()` and `enable_constraint_checking()`, to
        determine if rows with invalid references were entered while constraint
        checks were off.
        Na0  
                        SELECT REFERRING.`%s`, REFERRING.`%s` FROM `%s` as REFERRING
                        LEFT JOIN `%s` as REFERRED
                        ON (REFERRING.`%s` = REFERRED.`%s`)
                        WHERE REFERRING.`%s` IS NOT NULL AND REFERRED.`%s` IS NULL
                        z–The row in table '%s' with primary key '%s' has an invalid foreign key: %s.%s contains a value '%s' that does not have a corresponding value in %s.%s.r   r   )	r   ÚintrospectionÚtable_namesZget_primary_key_columnZget_relationsÚitemsr   Zfetchallr   )
r   r   r   Z
table_nameZprimary_key_column_nameZ	relationsZcolumn_nameZreferenced_column_nameZreferenced_table_nameZbad_rowr   r   r   Úcheck_constraints:  sP    
ÿý
øúÿùýÿz!DatabaseWrapper.check_constraintsc                 C   s.   z| j  ¡  W n tjy$   Y dS 0 dS d S )NFT)rm   Zpingr    ÚErrorr-   r   r   r   Ú	is_usabler  s
    zDatabaseWrapper.is_usablec                 C   s   | j r
dS dS )NZMariaDBÚMySQL)Úmysql_is_mariadbr-   r   r   r   Údisplay_namez  s    zDatabaseWrapper.display_namec                 C   s4   | j jr0ddddœ}| jr,| jdk r,d|d< |S i S )Nz`%(column)s` >= 0)r<   r=   r>   )é
   r   r   zJSON_VALID(`%(column)s`)r;   )rq   Z!supports_column_check_constraintsr…   rJ   )r   r�   r   r   r   Údata_type_check_constraints~  s    ýz+DatabaseWrapper.data_type_check_constraintsc                 C   sp   |   ¡ �"}| d¡ | ¡ }W d   ƒ n1 s00    Y  |d |d |d t|d ƒt|d ƒt|d ƒdœS )	Na2  
                SELECT VERSION(),
                       @@sql_mode,
                       @@default_storage_engine,
                       @@sql_auto_is_null,
                       @@lower_case_table_names,
                       CONVERT_TZ('2001-01-01 01:00:00', 'UTC', 'UTC') IS NOT NULL
            r   r   é   r   r   é   )ÚversionÚsql_modeZdefault_storage_engineZsql_auto_is_nullZlower_case_table_namesZhas_zoneinfo_database)Ztemporary_connectionr   Zfetchoner6   )r   r   Úrowr   r   r   Úmysql_server_data�  s    
ÿ
&


úz!DatabaseWrapper.mysql_server_datac                 C   s
   | j d S )Nr‹   )rŽ   r-   r   r   r   Úmysql_server_info§  s    z!DatabaseWrapper.mysql_server_infoc                 C   s4   t  | j¡}|std| j ƒ‚tdd„ | ¡ D ƒƒS )Nz8Unable to determine MySQL version from version string %rc                 s   s   | ]}t |ƒV  qd S r   )r^   )rX   Úxr   r   r   rZ   ³  r[   z0DatabaseWrapper.mysql_version.<locals>.<genexpr>)Úserver_version_reÚmatchr�   Ú	Exceptionr$   Úgroups)r   r’   r   r   r   rJ   «  s    ÿÿzDatabaseWrapper.mysql_versionc                 C   s   d| j  ¡ v S )NZmariadb)r�   ra   r-   r   r   r   r…   µ  s    z DatabaseWrapper.mysql_is_mariadbc                 C   s    | j d }t|r| d¡ndƒS )NrŒ   ú,r   )rŽ   ÚsetÚsplit)r   rŒ   r   r   r   rŒ   ¹  s    
zDatabaseWrapper.sql_mode)N)N).r/   r0   r1   ÚvendorÚ
data_typesZ_limited_data_typesÚ	operatorsZpattern_escZpattern_opsrb   r    r   ZSchemaEditorClassr   Zclient_classr   Zcreation_classr   Zfeatures_classr   Zintrospection_classr   Z	ops_classr   Zvalidation_classrK   rh   r   rn   rp   rw   rx   rz   r{   r}   r�   rƒ   r   r†   rˆ   rŽ   r�   rJ   r…   rŒ   Ú__classcell__r   r   rt   r   r3   d   s®   å#ôú	'



8




	
r3   )0r2   Zdjango.core.exceptionsr   Z	django.dbr   Zdjango.db.backendsr   Zbackend_utilsZdjango.db.backends.base.baser   Zdjango.utils.asyncior   Zdjango.utils.functionalr   Zdjango.utils.regex_helperr   ZMySQLdbr    ÚImportErrorÚerrZMySQLdb.constantsr	   r
   ZMySQLdb.convertersr   Úclientr   Zcreationr   rq   r   r~   r   Ú
operationsr   Zschemar   Z
validationr   Úversion_infor‹   Ú__version__ZTIMEZtypecast_timer\   r‘   r   r3   r   r   r   r   Ú<module>   sJ   ÿþ
ÿÿ
þ/