U
    dB                  
   @   sZ  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k
r Z zed
eW 5 d	Z[X Y nX 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) 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      r   z4mysqlclient 1.4.0 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   A/tmp/pip-unpacked-wheel-7vj2equ7/django/db/backends/mysql/base.py__init__E   s    zCursorWrapper.__init__Nc              
   C   sZ   z| j ||W S  tjk
rT } z$|jd | jkrBtt|j  W 5 d }~X Y nX d S Nr   )r   executeDatabaseOperationalErrorargscodes_for_integrityerrorr   tupler   queryr!   er   r   r   r   H   s    zCursorWrapper.executec              
   C   sZ   z| j ||W S  tjk
rT } z$|jd | jkrBtt|j  W 5 d }~X Y nX 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                       sp  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d*d+d,d-hZ	e
Z
eZeZeZeZeZeZeZd.d/ Zd0d1 Zed2d3 Z fd4d5ZedSd7d8Zd9d: Zd;d< Z d=d> Z!d?d@ Z"dTdAdBZ#dCdD Z$e%dEdF Z&e%dGdH Z'e%dIdJ Z(e%dKdL Z)e%dMdN Z*e%dOdP Z+e%dQdR Z,  Z-S )UDatabaseWrapperZ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
mediumblobr3   ZtinytexttextZ
mediumtextr8   r7   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('%%', {}))r?   r@   rC   rE   rD   rF   zread uncommittedread committedzrepeatable readZserializablec                 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 rt|d |d< tj|d< |d  }|dd}|r| }|| j	krt
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_levelrG   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>   s     z8DatabaseWrapper.get_connection_params.<locals>.<genexpr>)django_conversionssettings_dictrC   intr	   Z
FOUND_ROWScopypoplowerisolation_levelsr   joinsortedrU   update)r   kwargsrZ   optionsrU   r   r   r   get_connection_params   s@    


z%DatabaseWrapper.get_connection_paramsc                 C   s,   t jf |}|jttk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              	      sf   t    g }| jjr |d | jr:|d| j   |rb|  }|d	| W 5 Q R X 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appendrU   upperr   r   r`   )r   Zassignmentsr   	__class__r   r   rm      s    


z%DatabaseWrapper.init_connection_stateNc                 C   s   | j  }t|S r   )rj   r   r   )r   namer   r   r   r   create_cursor  s    
zDatabaseWrapper.create_cursorc                 C   s*   zt |  W n tjk
r$   Y nX d S r   )r   	_rollbackr   ZNotSupportedErrorr,   r   r   r   ru     s    zDatabaseWrapper._rollbackc              	   C   s"   | j  | j| W 5 Q R X d S r   )Zwrap_database_errorsrj   
autocommit)r   rv   r   r   r   _set_autocommit  s    zDatabaseWrapper._set_autocommitc              	   C   s"   |   }|d W 5 Q R X 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=0T)r   r   r   r   r   r   disable_constraint_checking#  s    
z+DatabaseWrapper.disable_constraint_checkingc              	   C   s>   d| j  | _ }z"|  }|d W 5 Q R X W 5 || _ X dS )zM
        Re-enable foreign key checks after they have been disabled.
        FzSET foreign_key_checks=1N)needs_rollbackr   r   )r   ry   r   r   r   r   enable_constraint_checking-  s
    
z*DatabaseWrapper.enable_constraint_checkingc           
      C   s   |   }|dkr| j|}|D ]}| j||}|s:q"| j||}| D ]Z\}\}}|d||||||||f  | D ]&}	td||	d |||	d ||f qqPq"W 5 Q R X 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
                        zThe 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:  sR    
 
z!DatabaseWrapper.check_constraintsc                 C   s0   z| j   W n tjk
r&   Y dS X dS d S )NFT)rj   Zpingr   Errorr,   r   r   r   	is_usabler  s
    zDatabaseWrapper.is_usablec                 C   s   | j r
dS dS )NZMariaDBZ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      zJSON_VALID(`%(column)s`)r9   )rn   Z!supports_column_check_constraintsr   rH   )r   r~   r   r   r   data_type_check_constraints~  s    z+DatabaseWrapper.data_type_check_constraintsc              	   C   s\   |   }|d | }W 5 Q R X |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fetchoner4   )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[   )rV   xr   r   r   rX     s     z0DatabaseWrapper.mysql_version.<locals>.<genexpr>)server_version_rematchr   	Exceptionr#   groups)r   r   r   r   r   rH     s    zDatabaseWrapper.mysql_versionc                 C   s   d| j  kS )NZmariadb)r   r^   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.   r/   r0   vendor
data_typesZ_limited_data_types	operatorsZpattern_escZpattern_opsr_   r   r   ZSchemaEditorClassr   Zclient_classr   Zcreation_classr   Zfeatures_classr   Zintrospection_classr   Z	ops_classr   Zvalidation_classrI   re   r   rk   rm   rt   ru   rw   rx   rz   r~   r   r   r   r   r   r   rH   r   r   __classcell__r   r   rq   r   r2   d   s   #
'



8




	
r2   )0r1   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   rn   r   r{   r   
operationsr   Zschemar   Z
validationr   version_infor   __version__ZTIMEZtypecast_timerY   r   r   r2   r   r   r   r   <module>   sF   
	
/