U
    ecIU                  
   @   s  d Z ddl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mZ dd	lmZ zddlZdd
lmZ W n2 ek
r Z zede eW 5 dZ[X Y nX zddlmZ W n ek
r   dZY nX 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& ej'j(Z(ej'j)Z)ej'j*Z*ej'j+Z+ej'j,Z,dd Z-G dd dZ.G dd de
Z/G dd deZ0dS )a  Django database Backend using MySQL Connector/Python.

This Django database backend is heavily based on the MySQL backend from Django.

Changes include:
* Support for microseconds (MySQL 5.6.3 and later)
* Using INFORMATION_SCHEMA where possible
* Using new defaults for, for example SQL_AUTO_IS_NULL

Requires and comes with MySQL Connector/Python v8.0.22 and later:
    http://dev.mysql.com/downloads/connector/python/
    N)datetime)settings)ImproperlyConfigured)IntegrityError)BaseDatabaseWrapper)	dateparsetimezone)cached_property)MySQLConverterz&Error loading mysql.connector module: )datetime_to_mysqlFT   )DatabaseClient)DatabaseCreation)DatabaseFeatures)DatabaseIntrospection)DatabaseOperations)DatabaseSchemaEditor)DatabaseValidationc                 C   sb   t jrLt| r8td|  dt t }t| |} | 	tj
jdd} trXt| S | dS )zDEquivalent to DateTimeField.get_db_prep_value. Used only by raw SQL.z!MySQL received a naive datetime (z$) while time zone support is active.Ntzinfoz%Y-%m-%d %H:%M:%S.%f)r   USE_TZr   is_naivewarningswarnRuntimeWarningZget_default_timezoneZ
make_aware
astimezoneutcreplace	HAVE_CEXTr   strftime)valueZdefault_timezone r!   ?/tmp/pip-unpacked-wheel-ix8dg7li/mysql/connector/django/base.py$adapt_datetime_with_timezone_supportR   s    

r#   c                   @   sV   e Zd ZdZdZdd Zedd Zedd Zdd
dZ	dd Z
dd Zdd Zd	S )CursorWrapperzWrapper around MySQL Connector/Python's cursor class.

    The cursor class is defined by the options passed to MySQL
    Connector/Python. If buffered option is True in those options,
    MySQLCursorBuffered will be used.
    )i  i  i  i  c                 C   s
   || _ d S N)cursorselfr&   r!   r!   r"   __init__r   s    zCursorWrapper.__init__c                 C   s<   | s| S t | }|  D ]\}}t|trt|||< q|S r%   )dictitems
isinstancer   r#   )argsnew_argskeyr    r!   r!   r"   _adapt_execute_args_dictu   s    
z&CursorWrapper._adapt_execute_args_dictc                 C   s@   | s| S t | }t| D ]\}}t|trt|||< qt|S r%   )list	enumerater,   r   r#   tuple)r-   r.   iargr!   r!   r"   _adapt_execute_args   s    
z!CursorWrapper._adapt_execute_argsNc              
   C   s~   t |tr| |}n
| |}z| j||W S  tjjk
rx } z&|j	d | j
krftt|j	 d W 5 d}~X Y nX dS )zExecutes the given operation

        This wrapper method around the execute()-method of the cursor is
        mainly needed to re-raise using different exceptions.
        r   N)r,   r*   r0   r6   r&   executemysql	connectorOperationalErrorr-   codes_for_integrityerrorr   r3   )r(   queryr-   r.   excr!   r!   r"   r7      s    

zCursorWrapper.executec              
   C   s^   z| j ||W S  tjjk
rX } z&|jd | jkrFtt|j d W 5 d}~X Y nX dS )zExecutes the given operation

        This wrapper method around the executemany()-method of the cursor is
        mainly needed to re-raise using different exceptions.
        r   N)	r&   executemanyr8   r9   r:   r-   r;   r   r3   )r(   r<   r-   r=   r!   r!   r"   r>      s    zCursorWrapper.executemanyc                 C   s   t | j|S )z%Return an attribute of wrapped cursor)getattrr&   r(   attrr!   r!   r"   __getattr__   s    zCursorWrapper.__getattr__c                 C   s
   t | jS )z&Return an iterator over wrapped cursor)iterr&   r(   r!   r!   r"   __iter__   s    zCursorWrapper.__iter__)N)__name__
__module____qualname____doc__r;   r)   staticmethodr0   r6   r7   r>   rB   rE   r!   r!   r!   r"   r$   c   s   




r$   c                       s  e Zd 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dZdZ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,Z	d-Z
d.d/d0d1hZejZeZeZeZeZeZeZeZ fd2d3Zd4d5 Zd6d7 Zd8d9 Z d:d; Z!dXd<d=Z"d>d? Z#d@dA Z$dBdC Z%dDdE Z&dYdFdGZ'dHdI Z(e)e*dJdK Z+e)dLdM Z,e)dNdO Z-e)dPdQ Z.e)dRdS Z/e)dTdU Z0e1dVdW Z2  Z3S )ZDatabaseWrapperz Represent a database connection.r8   z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GenericIPAddressFieldZ	JSONFieldZNullBooleanFieldZOneToOneFieldPositiveBigIntegerFieldPositiveIntegerFieldPositiveSmallIntegerFieldZ	SlugFieldZSmallAutoFieldZSmallIntegerFieldZ	TextFieldZ	TimeFieldZ	UUIDField)	ZtinyblobZblobZ
mediumblobrL   ZtinytexttextZ
mediumtextrQ   rP   z= %szLIKE %szLIKE BINARY %szREGEXP BINARY %sz	REGEXP %sz> %sz>= %sz< %sz<= %s)exactZiexactcontains	icontainsregexZiregex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('%%', {}))rW   rX   r\   r^   r]   r_   Nzread uncommittedzread committedzrepeatable readZserializablec                    sj   t  j|| | jd}|rV|dt | _|dt}t|tsLtd| | _	nt | _	t | _d S )NOPTIONSuse_pureconverter_classzXConverter class should be a subclass of mysql.connector.django.base.DjangoMySQLConverter)
superr)   settings_dictgetr   	_use_pureDjangoMySQLConverter
issubclassProgrammingError	converter)r(   r-   kwargsoptionsrb   	__class__r!   r"   r)     s    

zDatabaseWrapper.__init__c                 C   s   | drdS td S )NZmysql_isF)r\   AttributeErrorr@   r!   r!   r"   rB   3  s    
zDatabaseWrapper.__getattr__c                 C   sD  ddddd}| j }|d r(|d |d< |d r<|d |d< |d	 rP|d	 |d
< |d drl|d |d< n|d r|d |d< |d rt|d |d< tj|d< tjjjj	g|d< zt|d 
 }|d}|r| }|| jkrddd t| jD }td| d| d|| _|| W n tk
r>   Y nX |S )Nutf8TF)charsetZuse_unicodeZbufferedZconsume_resultsUSERuserNAMEZdatabaseZPASSWORDpasswdZHOST/Zunix_sockethostZPORTportZraise_on_warningsZclient_flagsr`   isolation_levelz, c                 s   s   | ]}d | d V  qdS )'Nr!   ).0levelr!   r!   r"   	<genexpr>]  s    z8DatabaseWrapper.get_connection_params.<locals>.<genexpr>z%Invalid transaction isolation level 'z' specified.
Use one of z
, or None.)rd   r\   intr   DEBUGr8   r9   	constantsZ
ClientFlagZ
FOUND_ROWScopypoplowerisolation_levelsjoinsortedr   ry   updateKeyError)r(   rk   rd   rl   ry   Zvalid_levelsr!   r!   r"   get_connection_params8  sL    



z%DatabaseWrapper.get_connection_paramsc                 C   s"   d|krt |d< tjjf |}|S )Nrb   )rg   r8   r9   connect)r(   Zconn_paramscnxr!   r!   r"   get_new_connectionk  s    z"DatabaseWrapper.get_new_connectionc              	   C   s   g }| j jr|d | jr2|d| j   |rZ|  }|d| W 5 Q R X d| jkrz| 	| jd  W n$ t
k
r   | | jd  Y nX d S )NzSET SQL_AUTO_IS_NULL = 0z(SET SESSION TRANSACTION ISOLATION LEVEL z; Z
AUTOCOMMIT)featuresZis_sql_auto_is_null_enabledappendry   upperr&   r7   r   rd   Zset_autocommitro   _set_autocommit)r(   Zassignmentsr&   r!   r!   r"   init_connection_stater  s    


z%DatabaseWrapper.init_connection_statec                 C   s   | j  }t|S r%   )
connectionr&   r$   )r(   namer&   r!   r!   r"   create_cursor  s    
zDatabaseWrapper.create_cursorc                 C   s(   zt |  W n tk
r"   Y nX d S r%   )r   	_rollbackNotSupportedErrorrD   r!   r!   r"   r     s    zDatabaseWrapper._rollbackc              	   C   s   | j  || j_W 5 Q R X d S r%   )Zwrap_database_errorsr   
autocommit)r(   r   r!   r!   r"   r     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&   r7   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&   r7   )r(   r   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 ]\}}}|d| d| d| d| d| d| d| d	| d
 | D ]@}	td| d|	d  d| d| d|	d  d| d| dqqLq"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.
        Nz+
                        SELECT REFERRING.`z&`,
                        REFERRING.`z `
                        FROM `z2` as REFERRING
                        LEFT JOIN `zR` as REFERRED
                        ON (
                            REFERRING.`z*` =
                            REFERRED.`zE`
                        )
                        WHERE REFERRING.`z4` IS NOT NULL
                        AND REFERRED.`z"` IS NULL
                        zThe row in table 'z' with primary key 'r   z' has an invalid foreign key: .z contains a value 'r   z.' that does not have a corresponding value in )r&   introspectiontable_namesZget_primary_key_columnZget_key_columnsr7   Zfetchallr   )
r(   r   r&   Z
table_nameZprimary_key_column_nameZkey_columnsZcolumn_nameZreferenced_table_nameZreferenced_column_nameZbad_rowr!   r!   r"   check_constraints  sN    
 6z!DatabaseWrapper.check_constraintsc                 C   s.   z| j   W n tk
r$   Y dS X dS d S )NFT)r   ZpingErrorrD   r!   r!   r"   	is_usable  s
    zDatabaseWrapper.is_usablec                   C   s   dS )zDisplay name.ZMySQLr!   r!   r!   r!   r"   display_name  s    zDatabaseWrapper.display_namec                 C   s   | j jrdddd}|S i S )z<Mapping of Field objects to their SQL for CHECK constraints.z`%(column)s` >= 0)rR   rS   rT   )r   Z!supports_column_check_constraints)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 )	zReturn MySQL server data.a2  
                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               )versionsql_modeZdefault_storage_engineZsql_auto_is_nullZlower_case_table_namesZhas_zoneinfo_database)temporary_connectionr7   fetchonerM   )r(   r&   rowr!   r!   r"   mysql_server_data  s    




z!DatabaseWrapper.mysql_server_datac              
   C   s6   |   $}|d | d W  5 Q R  S Q R X dS )Return MySQL version.zSELECT VERSION()r   N)r   r7   r   r'   r!   r!   r"   mysql_server_info  s    

z!DatabaseWrapper.mysql_server_infoc              	   C   s.   |   }tjjf |}| }W 5 Q R X |S )r   )r   r8   r9   r   Zget_server_version)r(   configconnZserver_versionr!   r!   r"   mysql_version  s    zDatabaseWrapper.mysql_versionc              	   C   s@   |   }|d | }W 5 Q R X t|r:|d dndS )zReturn SQL mode.zSELECT @@sql_moder   ,r!   )r&   r7   r   setsplit)r(   r&   r   r!   r!   r"   r   $  s    

zDatabaseWrapper.sql_modec                 C   s   | j S )z1Return True if pure Python version is being used.)rf   rD   r!   r!   r"   ra   ,  s    zDatabaseWrapper.use_pure)N)N)4rF   rG   rH   rI   vendor
data_typesZ_limited_data_types	operatorsZpattern_escZpattern_opsry   r   r8   r9   ZDatabaser   ZSchemaEditorClassr   Zclient_classr   Zcreation_classr   Zfeatures_classr   Zintrospection_classr   Z	ops_classr   Zvalidation_classr)   rB   r   r   r   r   r   r   r   r   r   r   r	   rJ   r   r   r   r   r   r   propertyra   __classcell__r!   r!   rm   r"   rK      s   $	3


0




rK   c                   @   sD   e Zd ZdZedddZedddZdd Zd	d
 Zdd Z	dS )rg   zCustom converter for Django.Nc                 C   s   t | dS )zXReturn MySQL TIME data type as datetime.time()

        Returns datetime.time()
        zutf-8)r   Z
parse_timedecode)r    dscr!   r!   r"   _time_to_python7  s    z$DjangoMySQLConverter._time_to_pythonc                 C   s@   | sdS t | }|dkrdS tjr<t|r<|jtjd}|S )a  Connector/Python always returns naive datetime.datetime

        Connector/Python always returns naive timestamps since MySQL has
        no time zone support. Since Django needs non-naive, we need to add
        the UTC time zone.

        Returns datetime.datetime()
        Nr   )r
   _datetime_to_pythonr   r   r   r   r   r   )r    r   dtr!   r!   r"   r   ?  s    

z(DjangoMySQLConverter._datetime_to_pythonc                 C   s
   |  |S r%   Z_str_to_mysqlr(   r    r!   r!   r"   _safestring_to_mysqlU  s    z)DjangoMySQLConverter._safestring_to_mysqlc                 C   s
   |  |S r%   r   r   r!   r!   r"   _safetext_to_mysqlX  s    z'DjangoMySQLConverter._safetext_to_mysqlc                 C   s
   |  |S r%   )Z_bytes_to_mysqlr   r!   r!   r"   _safebytes_to_mysql[  s    z(DjangoMySQLConverter._safebytes_to_mysql)N)N)
rF   rG   rH   rI   rJ   r   r   r   r   r   r!   r!   r!   r"   rg   2  s   rg   )1rI   r   r   Zdjango.confr   Zdjango.core.exceptionsr   Z	django.dbr   Zdjango.db.backends.base.baser   Zdjango.utilsr   r   Zdjango.utils.functionalr	   Zmysql.connectorr8   Zmysql.connector.conversionr
   ImportErrorerrZ_mysql_connectorr   r   clientr   Zcreationr   r   r   r   r   
operationsr   Zschemar   Z
validationr   r9   r   ZDatabaseErrorr   r:   ri   r#   r$   rK   rg   r!   r!   r!   r"   <module>   sJ   "
O   