o
    h"^                  
   @   s&  d Z ddlZddlmZmZ ddlmZmZmZmZm	Z	m
Z
mZ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mZ dd
lmZ 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( ddl)m*Z* ddl+m,Z,m-Z-m.Z.m/Z/m0Z0 W n e1y Z2 zede2 e2dZ2[2ww zddl3m4Z4 W n e1y   dZ5Y nw dZ5ddl6m7Z7 ddl8m9Z9 ddl:m;Z; ddl<m=Z= ddl>m?Z? ddl@mAZA ddlBmCZC ejDjEZEejDjFZFejDjGZGejDjHZHejDjIZIded e0fd!d"ZJG d#d$ d$ZKG d%d& d&eZLG d'd( d(e"ZMd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time)	AnyDict	GeneratorIteratorListOptionalSetTupleUnion)settings)ImproperlyConfigured)IntegrityError)BaseDatabaseWrapper)	dateparsetimezone)cached_property)MySQLConnection)CMySQLConnection)MySQLConverter)MySQLCursor)CMySQLCursor)
HexLiteral)PooledMySQLConnection)ParamsDictTypeParamsSequenceOrDictTypeParamsSequenceTypeRowType
StrOrBytesz&Error loading mysql.connector module: )datetime_to_mysqlFT   )DatabaseClient)DatabaseCreation)DatabaseFeatures)DatabaseIntrospection)DatabaseOperations)DatabaseSchemaEditor)DatabaseValidationvaluereturnc                 C   sf   t jr&t| rtd|  dt t }t| |} | 	tj
jdd} tr.t| }|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RuntimeWarningget_default_timezone
make_aware
astimezoneutcreplace	HAVE_CEXTr    strftime)r)   default_timezonemysql_datetime r:   V/var/www/html/scripts/venv/lib/python3.10/site-packages/mysql/connector/django/base.py$adapt_datetime_with_timezone_supportb   s   


r<   c                   @   s   e Zd ZdZdZdeeef ddfddZe	de
de
fd	d
Ze	dee dee fddZ	ddedee deeeeef ddf  fddZdedeeedf ee f deeeeef ddf  fddZdedefddZdee f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  cursorr*   Nc                 C   s
   || _ d S N)r>   selfr>   r:   r:   r;   __init__      
zCursorWrapper.__init__argsc                 C   s<   | s| S t | }|  D ]\}}t|trt|||< q|S r?   )dictitems
isinstancer   r<   )rD   new_argskeyr)   r:   r:   r;   _adapt_execute_args_dict   s   
z&CursorWrapper._adapt_execute_args_dictc                 C   s@   | s| S t | }t| D ]\}}t|trt|||< qt|S r?   )list	enumeraterG   r   r<   tuple)rD   rH   iargr:   r:   r;   _adapt_execute_args   s   
z!CursorWrapper._adapt_execute_argsqueryc              
   C   st   d}t |tr| |}n| |}z| j||W S  tjjy9 } z|j	d | j
v r4tt|j	 d d}~ww )zExecutes the given operation

        This wrapper method around the execute()-method of the cursor is
        mainly needed to re-raise using different exceptions.
        Nr   )rG   rE   rJ   rP   r>   executemysql	connectorOperationalErrorrD   codes_for_integrityerrorr   rM   )rA   rQ   rD   rH   excr:   r:   r;   rR      s   

zCursorWrapper.execute.c              
   C   sP   z| j ||W S  tjjy' } z|jd | jv r"tt|j d d}~ww )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>   executemanyrS   rT   rU   rD   rV   r   rM   )rA   rQ   rD   rW   r:   r:   r;   rX      s   zCursorWrapper.executemanyattrc                 C   s   t | j|S )z%Return an attribute of wrapped cursor)getattrr>   rA   rY   r:   r:   r;   __getattr__   s   zCursorWrapper.__getattr__c                 C   s
   t | jS )z&Return an iterator over wrapped cursor)iterr>   rA   r:   r:   r;   __iter__   s   
zCursorWrapper.__iter__r?   )__name__
__module____qualname____doc__rV   r   r   r   rB   staticmethodr   rJ   r	   r   rP   strr   r   rR   r   r   rX   r   r\   r   r   r_   r:   r:   r:   r;   r=   t   sB    



r=   c                       s  e Zd ZU dZdZi 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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/d0d1d2d3d4d.d.d-d-d5Zd6Zd7d8d9d:d;d<d=Z	d>Z
ee ed?< h d@ZejZeZeZeZeZeZeZeZdAe dBe dCd>f fdDdEZ!dFedCe"fdGdHZ#dCe$ee f fdIdJZ%dKe$ee f dCe&e'e(e)f fdLdMZ*dpdNdOZ+dqdPe dCe,fdQdRZ-dpdSdTZ.dUe"dCd>fdVdWZ/dCe"fdXdYZ0dpdZd[Z1dqd\ee2e  dCd>fd]d^Z3dCe"fd_d`Z4e5e6dCefdadbZ7e5dCe$eef fdcddZ8e5dCe$ee f fdedfZ9e5dCe fdgdhZ:e5dCe;e<dif fdjdkZ=e5dCe>e fdldmZ?e@dCe"fdndoZA  ZBS )rDatabaseWrapperz Represent a database connection.rS   	AutoFieldzinteger AUTO_INCREMENTBigAutoFieldzbigint AUTO_INCREMENTBinaryFieldlongblobBooleanFieldbool	CharFieldzvarchar(%(max_length)s)	DateFielddateDateTimeFieldzdatetime(6)DecimalFieldz+numeric(%(max_digits)s, %(decimal_places)s)DurationFieldbigint	FileFieldFilePathField
FloatFieldzdouble precisionIntegerFieldintegerBigIntegerFieldIPAddressFieldzchar(15)GenericIPAddressFieldzchar(39)	JSONFieldjsonzbigint UNSIGNEDzinteger UNSIGNEDzsmallint UNSIGNEDzsmallint AUTO_INCREMENTsmallintlongtextztime(6)zchar(32))NullBooleanFieldOneToOneFieldPositiveBigIntegerFieldPositiveIntegerFieldPositiveSmallIntegerField	SlugFieldSmallAutoFieldSmallIntegerField	TextField	TimeField	UUIDField)	tinyblobblob
mediumblobrj   tinytexttext
mediumtextr   r}   z= %szLIKE %szLIKE BINARY %szREGEXP BINARY %sz	REGEXP %sz> %sz>= %sz< %sz<= %s)exactiexactcontains	icontainsregexiregexgtgtelt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   r   r   r   r   Nisolation_level>   read committedrepeatable readread uncommittedserializablerD   kwargsr*   c                    sp   t  j|i | | jd}|r.|dt | _|dt}t|ts(td| | _	d S t | _	t | _d S )NOPTIONSuse_pureconverter_classzXConverter class should be a subclass of mysql.connector.django.base.DjangoMySQLConverter)
superrB   settings_dictgetr6   	_use_pureDjangoMySQLConverter
issubclassProgrammingError	converter)rA   rD   r   optionsr   	__class__r:   r;   rB   <  s   
zDatabaseWrapper.__init__rY   c                 C   s   | drdS t)Nmysql_isF)r   AttributeErrorr[   r:   r:   r;   r\   P  s   
zDatabaseWrapper.__getattr__c                 C   s^  ddddd}| j }|d r|d |d< |d r|d |d< |d	 r(|d	 |d
< |d dr6|d |d< n
|d r@|d |d< |d rLt|d |d< |di dr]|d d |d< tj|d< tjjj	j
g|d< z9|d  }|d}|r| }|| jvrddd t| jD }td| d| d|| _|| W |S  ty   Y |S w )Nutf8TF)charsetuse_unicodebufferedconsume_resultsUSERuserNAMEdatabasePASSWORDpasswdHOST/unix_sockethostPORTportr   init_commandraise_on_warningsclient_flagsr   z, 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.)r   r   intr   r   DEBUGrS   rT   	constants
ClientFlag
FOUND_ROWScopypoplowerisolation_levelsjoinsortedr   r   updateKeyError)rA   r   r   r   r   valid_levelsr:   r:   r;   get_connection_paramsU  sX   




z%DatabaseWrapper.get_connection_paramsconn_paramsc                 C   s&   d|vrt |d< tjjdi |}|S )Nr   r:   )r   rS   rT   connect)rA   r   cnxr:   r:   r;   get_new_connection  s   z"DatabaseWrapper.get_new_connectionc                 C   s   g }| j jr|d | jr|d| j   |r7|  }|d| W d    n1 s2w   Y  d| jv rZz| 	| jd  W d S  t
yY   | | jd  Y d S w d S )NzSET SQL_AUTO_IS_NULL = 0z(SET SESSION TRANSACTION ISOLATION LEVEL z; 
AUTOCOMMIT)featuresis_sql_auto_is_null_enabledappendr   upperr>   rR   r   r   set_autocommitr   _set_autocommit)rA   assignmentsr>   r:   r:   r;   init_connection_state  s(   


z%DatabaseWrapper.init_connection_statenamec                 C   s   | j  }t|S r?   )
connectionr>   r=   )rA   r   r>   r:   r:   r;   create_cursor  s   
zDatabaseWrapper.create_cursorc                 C   s&   zt |  W d S  ty   Y d S w r?   )r   	_rollbackNotSupportedErrorr^   r:   r:   r;   r     s
   zDatabaseWrapper._rollback
autocommitc                 C   s4   | j  || j_W d    d S 1 sw   Y  d S r?   )wrap_database_errorsr   r   )rA   r   r:   r:   r;   r     s   
"zDatabaseWrapper._set_autocommitc                 C   s8   |   }|d W d   dS 1 sw   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>   rR   r@   r:   r:   r;   disable_constraint_checking  s   

z+DatabaseWrapper.disable_constraint_checkingc                 C   sb   d| j | _ }z%|  }|d W d   n1 sw   Y  W || _ dS W || _ dS || _ w )zM
        Re-enable foreign key checks after they have been disabled.
        FzSET foreign_key_checks=1N)needs_rollbackr>   rR   )rA   r   r>   r:   r:   r;   enable_constraint_checking  s   

z*DatabaseWrapper.enable_constraint_checkingtable_namesc           
      C   s  |   s}|du r| j|}|D ]]}| j||}|sq| j||}|D ]G\}}}|d| d| d| d| d| d| d| d	| d
 | D ]}	td| d|	d  d| d| d|	d  d| d| dq&qW d   dS 1 szw   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.
        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>   introspectionr   get_primary_key_columnget_key_columnsrR   fetchallr   )
rA   r   r>   
table_nameprimary_key_column_namekey_columnscolumn_namereferenced_table_namereferenced_column_namebad_rowr:   r:   r;   check_constraints  sl   
	
"z!DatabaseWrapper.check_constraintsc                 C   s&   z| j   W dS  ty   Y dS w )NFT)r   pingErrorr^   r:   r:   r;   	is_usable  s   zDatabaseWrapper.is_usablec                   C   s   dS )zDisplay name.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)r   r   r   )r   !supports_column_check_constraints)rA   r   r:   r:   r;   data_type_check_constraints  s   z+DatabaseWrapper.data_type_check_constraintsc                 C   sp   |   }|d | }W d   n1 sw   Y  |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
            Nr   r!               )versionsql_modedefault_storage_enginesql_auto_is_nulllower_case_table_nameshas_zoneinfo_database)temporary_connectionrR   fetchonerl   )rA   r>   rowr:   r:   r;   mysql_server_data  s   





z!DatabaseWrapper.mysql_server_datac                 C   sB   |   }|d | d W  d   S 1 sw   Y  dS )Return MySQL version.zSELECT VERSION()r   N)r  rR   r  r@   r:   r:   r;   mysql_server_info5  s   


$z!DatabaseWrapper.mysql_server_info.c                 C   sH   |   }tjjdi |}| }W d   |S 1 sw   Y  |S )r  Nr:   )r   rS   rT   r   get_server_version)rA   configconnserver_versionr:   r:   r;   mysql_version<  s   

zDatabaseWrapper.mysql_versionc                 C   sV   |   }|d | }W d   n1 sw   Y  t|r(|d dS dS )zReturn SQL mode.zSELECT @@sql_modeNr   ,r:   )r>   rR   r  setsplit)rA   r>   r  r:   r:   r;   r  D  s
   


zDatabaseWrapper.sql_modec                 C   s
   | j }|S )z1Return True if pure Python version is being used.)r   )rA   ansr:   r:   r;   r   L  s   zDatabaseWrapper.use_pure)r*   Nr?   )Cr`   ra   rb   rc   vendor
data_types_limited_data_types	operatorspattern_escpattern_opsr   r	   re   __annotations__r   rS   rT   Databaser'   SchemaEditorClassr"   client_classr#   creation_classr$   features_classr%   introspection_classr&   	ops_classr(   validation_classr   rB   rl   r\   r   r   r   r   r   r   r   r   r=   r   r   r   r   r   r   r   r  r   rd   r  r  r  r  r   r   r  r
   r  propertyr   __classcell__r:   r:   r   r;   rf      s   
 	
$	5


	


0rf   c                	   @   s   e Zd ZdZeddededee fddZ	eddededee
 fdd	Zdedeeef fd
dZdedeeef fddZdedefddZdS )r   zCustom converter for Django.Nr)   dscr*   c                 C   s   t | dS )zXReturn MySQL TIME data type as datetime.time()

        Returns datetime.time()
        zutf-8)r   
parse_timedecoder)   r2  r:   r:   r;   _time_to_pythonX  s   z$DjangoMySQLConverter._time_to_pythonc                 C   s   | rt | S dS )ao  Connector/Python always returns naive datetime.datetime

        Connector/Python always returns naive timestamps since MySQL has
        no time zone support.

        - A naive datetime is a datetime that doesn't know its own timezone.

        Django needs a non-naive datetime, but in this method we don't need
        to make a datetime value time zone aware since Django itself at some
        point will make it aware (at least in versions 3.2.16 and 4.1.2) when
        USE_TZ=True. This may change in a future release, we need to keep an
        eye on this behaviour.

        Returns datetime.datetime()
        N)r   _datetime_to_pythonr5  r:   r:   r;   r7  `  s   z(DjangoMySQLConverter._datetime_to_pythonc                 C   
   |  |S r?   _str_to_mysqlrA   r)   r:   r:   r;   _safestring_to_mysqlu  rC   z)DjangoMySQLConverter._safestring_to_mysqlc                 C   r8  r?   r9  r;  r:   r:   r;   _safetext_to_mysqlx  rC   z'DjangoMySQLConverter._safetext_to_mysqlc                 C   r8  r?   )_bytes_to_mysqlr;  r:   r:   r;   _safebytes_to_mysql{  rC   z(DjangoMySQLConverter._safebytes_to_mysqlr?   )r`   ra   rb   rc   rd   bytesr   r	   r   r6  r   r7  re   r   r   r<  r=  r?  r:   r:   r:   r;   r   S  s    r   )Nrc   r.   r   r   typingr   r   r   r   r   r	   r
   r   r   django.confr   django.core.exceptionsr   	django.dbr   django.db.backends.base.baser   django.utilsr   r   django.utils.functionalr   mysql.connectorrS   mysql.connector.connectionr   mysql.connector.connection_cextr   mysql.connector.conversionr   mysql.connector.cursorr   mysql.connector.cursor_cextr   mysql.connector.custom_typesr   mysql.connector.poolingr   mysql.connector.typesr   r   r   r   r   ImportErrorerr_mysql_connectorr    r6   clientr"   creationr#   r   r$   r   r%   
operationsr&   schemar'   
validationr(   rT   r  DatabaseErrorr   rU   r   r<   r=   rf   r   r:   r:   r:   r;   <module>   sb   , [   