File: //var/opt/nydus/ops/mysql/connector/django/__pycache__/base.cpython-39.pyc
a
�,�h�^ �
@ s& d Z ddlZddlmZmZ ddlmZmZmZmZm Z m
Z
m Z m Z m
Z
mZmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZmZ dd
lm Z dd l m Z zDddl Z dd l!m"Z" dd
l#m$Z$ ddl%m&Z& ddl m'Z'm(Z(m)Z) W n6 e*�y" Z+ z ede+� ��e+�W Y dZ+[+n
dZ+[+0 0 zddl,m-Z- W n e*�yL dZ.Y n0 dZ.ddl/m0Z0 ddl1m2Z2 ddl3m4Z4 ddl5m6Z6 ddl7m8Z8 ddl9m:Z: ddl;m<Z< e j=j>Z>e j=j?Z?e j=j@Z@e j=jAZAe j=jBZBe�r�dd lCmDZDmEZE ee)d �d d �ZFG d d!� d!�ZGG d"d#� d#e�ZHG d$d%� d%e"�ZIdS )&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) �
TYPE_CHECKING�Any�Dict� Generator�Iterator�List�Optional�Sequence�Set�Tuple�Union)�settings)�ImproperlyConfigured)�IntegrityError)�BaseDatabaseWrapper)� dateparse�timezone)�cached_property)�MySQLConvertibleType)�MySQLConverter)�
HexLiteral)�PooledMySQLConnection)�ParamsSequenceOrDictType�RowType�
StrOrBytesz&Error loading mysql.connector module: )�datetime_to_mysqlFT� )�DatabaseClient)�DatabaseCreation)�DatabaseFeatures)�DatabaseIntrospection)�DatabaseOperations)�DatabaseSchemaEditor)�DatabaseValidation)�MySQLConnectionAbstract�MySQLCursorAbstract��value�returnc C sf t jrLt�| �r8t�d| � d�t� t�� }t�| |�} | � tj
�j dd�} 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 ZUSE_TZr Zis_naive�warnings�warn�RuntimeWarningZget_default_timezoneZ
make_aware�
astimezone�utc�replace� HAVE_CEXTr �strftime)r) Zdefault_timezoneZmysql_datetime� r4 �N/opt/nydus/tmp/pip-target-wkfpz8uv/lib64/python/mysql/connector/django/base.py�$adapt_datetime_with_timezone_supporti s
� �r6 c @ s� e Zd ZdZdZddd�dd�Zeeee f eee f d�d d
��Z
ee e e e e e d�d d ��Z
dee ee e eee f f e ed
d�dd�Zee ee e eee f f e ed
d�dd�Zeed�dd�Zee d�dd�ZdS )�
CursorWrapperz�Wrapper 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� r' N)�cursorr* c C s
|| _ d S �N)r8 ��selfr8 r4 r4 r5 �__init__� s zCursorWrapper.__init__)�argsr* c C s< | s| S t | �}| �� D ] \}}t|t�rt|�||< q|S r9 )�dict�items�
isinstancer r6 )r= �new_args�keyr) r4 r4 r5 �_adapt_execute_args_dict� s
z&CursorWrapper._adapt_execute_args_dictc C s@ | s| S t | �}t| �D ] \}}t|t�rt|�||< qt|�S r9 )�list� enumerater@ r r6 �tuple)r= rA �i�argr4 r4 r5 �_adapt_execute_args� s
z!CursorWrapper._adapt_execute_args)r' NN)�queryr= r* c
C s� d}t |t�r| �|�}n
| �|�}z| j�||�W S tjjy~ } z.|j d | j
v rht t |j �� d�� W Y d}~n
d}~0 0 dS )z�Executes the given operation
This wrapper method around the execute()-method of the cursor is
mainly needed to re-raise using different exceptions.
Nr )
r@ r> rC rI r8 �execute�mysql� connector�OperationalErrorr= �codes_for_integrityerrorr rF )r; rJ r= rA �excr4 r4 r5 rK � s
zCursorWrapper.executec
C s` z| j �||�W S tjjyZ } z.|jd | jv rDtt|j�� d�� W Y d}~n
d}~0 0 dS )z�Executes the given operation
This wrapper method around the executemany()-method of the cursor is
mainly needed to re-raise using different exceptions.
r N) r8 � executemanyrL rM rN r= rO r rF )r; rJ r= rP r4 r4 r5 rQ � s zCursorWrapper.executemany��attrr* c C s t | j|�S )z%Return an attribute of wrapped cursor)�getattrr8 �r; rS r4 r4 r5 � __getattr__� s zCursorWrapper.__getattr__�r* c C s
t | j�S )z&Return an iterator over wrapped cursor)�iterr8 �r; r4 r4 r5 �__iter__� s zCursorWrapper.__iter__)N)�__name__�
__module__� __qualname__�__doc__rO r< � staticmethodr �strr rC r
r rI r r rK rQ r rV r r rZ r4 r4 r4 r5 r7 } s6
�
� ��
� �
� r7 c sJ e Zd ZU dZdZddddddd d
d ddd d
d ddddd
dddddddddd� ZdZdd d d d d d d!d"d#d d d d d$�Zd%Zd&d'd(d)d*d+d,�Z d-Z
e e e
d.< h d/�ZejZeZeZeZeZeZe Z e Z e e d-d0�� fd1d2� Z!e e"d3�d4d5�Z#e$e e f d6�d7d8�Z%e$e e f e&e'd9f d:�d;d<�Z(d-d6�d=d>�Z)d_e e*d?�d@dA�Z+d-d6�dBdC�Z,e"d-dD�dEdF�Z-e"d6�dGdH�Z.d-d6�dIdJ�Z/d`e e0e d-dK�dLdM�Z1e"d6�dNdO�Z2e3e4e d6�dPdQ���Z5e3e$e e f d6�dRdS��Z6e3e$e e f d6�dTdU��Z7e3e d6�dVdW��Z8e3e9e:dXf d6�dYdZ��Z;e3e<e d6�d[d\��Z=e>e"d6�d]d^��Z?� Z@S )a�DatabaseWrapperz Represent a database connection.rL zinteger AUTO_INCREMENTzbigint AUTO_INCREMENT�longblob�boolzvarchar(%(max_length)s)�datez datetime(6)z+numeric(%(max_digits)s, %(decimal_places)s)Zbigintzdouble precision�integerzchar(15)zchar(39)�jsonzbigint UNSIGNEDzinteger UNSIGNEDzsmallint UNSIGNEDzsmallint AUTO_INCREMENTZsmallint�longtextztime(6)zchar(32)) Z AutoFieldZ BigAutoFieldZ BinaryFieldZ BooleanFieldZ CharFieldZ DateFieldZ
DateTimeFieldZ DecimalFieldZ
DurationFieldZ FileFieldZ
FilePathFieldZ
FloatFieldZ IntegerFieldZBigIntegerFieldZIPAddressFieldZGenericIPAddressFieldZ JSONFieldZNullBooleanFieldZ
OneToOneField�PositiveBigIntegerField�PositiveIntegerField�PositiveSmallIntegerFieldZ SlugFieldZSmallAutoFieldZSmallIntegerFieldZ TextFieldZ TimeFieldZ UUIDField) ZtinyblobZblobZ
mediumblobrb Ztinytext�textZ
mediumtextrg rf z= %szLIKE %szLIKE BINARY %szREGEXP BINARY %sz REGEXP %sz> %sz>= %sz< %sz<= %s)�exactZiexact�contains� icontains�regexZiregex�gtZgte�ltZlte�
startswith�endswith� istartswith� iendswithzCREPLACE(REPLACE(REPLACE({}, '\\', '\\\\'), '%%', '\%%'), '_', '\_')z"LIKE BINARY CONCAT('%%', {}, '%%')zLIKE CONCAT('%%', {}, '%%')z LIKE BINARY CONCAT({}, '%%')zLIKE CONCAT({}, '%%')z LIKE BINARY CONCAT('%%', {})zLIKE CONCAT('%%', {}))rm rn rr rt rs ru N�isolation_level> zread uncommittedzread committedzrepeatable readZ serializable)r= �kwargsr* c sn t � j|i |�� | j�d�}|rZ|�dt �| _|�dt�}t|t�sPtd��|� | _ nt� | _ t | _d S )N�OPTIONS�use_pure�converter_classzXConverter class should be a subclass of mysql.connector.django.base.DjangoMySQLConverter)
�superr< �
settings_dict�getr2 � _use_pure�DjangoMySQLConverter�
issubclass�ProgrammingError� converter)r; r= rw �optionsrz �� __class__r4 r5 r< J s �
�
zDatabaseWrapper.__init__rR c C s |� d�rdS t�d S )NZmysql_isF)rr �AttributeErrorrU r4 r4 r5 rV ^ s
zDatabaseWrapper.__getattr__rW c C sf ddddd�}| j }|d r(|d |d< |d r<|d |d< |d rP|d |d
< |d �d �rl|d |d
< n|d r�|d |d< |d r�t|d �|d< |�di ��d�r�|d d |d< tj|d< tjjj j
g|d< zv|d � � }|� dd �}|�r:|�
� }|| jv�r:d�dd� t| j�D ��}td|� d|� d���|| _|�|� W n t�y` Y n0 |S ) N�utf8TF)�charsetZ use_unicodeZbufferedZconsume_results�USER�user�NAMEZdatabaseZPASSWORD�passwdZHOST�/Z unix_socket�hostZPORT�portrx Z init_commandZraise_on_warningsZ client_flagsrv z, c s s | ]}d |� d �V qdS )�'Nr4 )�.0�levelr4 r4 r5 � <genexpr>� s z8DatabaseWrapper.get_connection_params.<locals>.<genexpr>z%Invalid transaction isolation level 'z' specified.
Use one of z
, or None.)r| rr �intr} r �DEBUGrL rM � constantsZ
ClientFlagZ
FOUND_ROWS�copy�pop�lower�isolation_levels�join�sortedr rv �update�KeyError)r; rw r| r� rv Z valid_levelsr4 r4 r5 �get_connection_paramsc sT �
�
���z%DatabaseWrapper.get_connection_paramsr& )� conn_paramsr* c C s& d|vrt |d< tjjf i |��}|S )Nrz )r rL rM �connect)r; r� Zcnxr4 r4 r5 �get_new_connection� s z"DatabaseWrapper.get_new_connectionc C s� g }| j jr|�d� | jr2|�d| j�� � �� |rn| �� � }|�d�|�� W d � n1 sd0 Y d| jv r�z| � | jd � W n" t
y� | � | jd � Y n0 d S )NzSET SQL_AUTO_IS_NULL = 0z(SET SESSION TRANSACTION ISOLATION LEVEL z; Z
AUTOCOMMIT) �featuresZis_sql_auto_is_null_enabled�appendrv �upperr8 rK r� r| Zset_autocommitr� �_set_autocommit)r; Z assignmentsr8 r4 r4 r5 �init_connection_state� s"
��
.
z%DatabaseWrapper.init_connection_state)�namer* c C s | j �� }t|�S r9 )�
connectionr8 r7 )r; r� r8 r4 r4 r5 �
create_cursor� s
z DatabaseWrapper.create_cursorc C s& zt �| � W n ty Y n0 d S r9 )r � _rollback�NotSupportedErrorrY r4 r4 r5 r� � s zDatabaseWrapper._rollback)�
autocommitr* c C s2 | j � || j_W d � n1 s$0 Y d S r9 )Zwrap_database_errorsr� r� )r; r� r4 r4 r5 r� � s z DatabaseWrapper._set_autocommitc C s6 | � � �}|�d� W d � n1 s(0 Y dS )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.
zSET foreign_key_checks=0NT)r8 rK r: r4 r4 r5 �disable_constraint_checking� s
(z+DatabaseWrapper.disable_constraint_checkingc C sX d| j | _ }z<| �� �}|�d� W d � n1 s80 Y W || _ n|| _ 0 dS )zM
Re-enable foreign key checks after they have been disabled.
FzSET foreign_key_checks=1N)�needs_rollbackr8 rK )r; r� r8 r4 r4 r5 �enable_constraint_checking� s
*z*DatabaseWrapper.enable_constraint_checking)� table_namesr* c
C s | � � ��}|du r | j�|�}|D ]�}| j�||�}|s:q"| j�||�}|D ]�\}}}|�d|� d|� d|� d|� d|� d|� d|� d |� d
�� |�� D ]@} td |� d | d
� d|� d|� d| d � d|� d|� d���q�qLq"W d � n1 s�0 Y dS )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
zThe 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 )r8 �
introspectionr� Zget_primary_key_columnZget_key_columnsrK Zfetchallr )
r; r� r8 Z
table_nameZprimary_key_column_nameZ key_columnsZ column_nameZreferenced_table_nameZreferenced_column_nameZbad_rowr4 r4 r5 �check_constraints� sh
��������� �
�� �������z!DatabaseWrapper.check_constraintsc C s( z| j �� W n ty" Y dS 0 dS )NFT)r� Zping�ErrorrY r4 r4 r5 � is_usable s
zDatabaseWrapper.is_usablec C s dS )z
Display name.ZMySQLr4 r4 r4 r4 r5 � display_name s z DatabaseWrapper.display_namec C s | j jrdddd�}|S i S )z<Mapping of Field objects to their SQL for CHECK constraints.z`%(column)s` >= 0)rh ri rj )r� Z!supports_column_check_constraints)r; r� r4 r4 r5 �data_type_check_constraints s �z+DatabaseWrapper.data_type_check_constraintsc C sp | � � �"}|�d� |�� }W d � n1 s00 Y |d |d |d t|d �t|d �t|d �d �S )
zReturn 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_modeZdefault_storage_engineZsql_auto_is_nullZlower_case_table_namesZhas_zoneinfo_database)�temporary_connectionrK �fetchonerc )r; r8 �rowr4 r4 r5 �mysql_server_data* s
�
&
�z!DatabaseWrapper.mysql_server_datac C sB | � � �&}|�d� |�� d W d � S 1 s40 Y dS )�Return MySQL version.zSELECT VERSION()r N)r� rK r� r: r4 r4 r5 �mysql_server_infoE s
z!DatabaseWrapper.mysql_server_info.c C sF | � � }tjjf i |���}|�� }W d � n1 s80 Y |S )r� N)r� rL rM r� Zget_server_version)r; �config�connZserver_versionr4 r4 r5 �
mysql_versionL s &z DatabaseWrapper.mysql_versionc C sT | � � �"}|�d� |�� }W d � n1 s00 Y t|rN|d �d�nd�S )zReturn SQL mode.zSELECT @@sql_modeNr �,r4 )r8 rK r� �set�split)r; r8 r� r4 r4 r5 r� T s
&zDatabaseWrapper.sql_modec C s
| j }|S )z1Return True if pure Python version is being used.)r~ )r; Zansr4 r4 r5 ry \ s zDatabaseWrapper.use_pure)N)N)Ar[ r\ r] r^ �vendor�
data_typesZ_limited_data_types� operatorsZ pattern_escZ pattern_opsrv r
r` �__annotations__r� rL rM ZDatabaser$ ZSchemaEditorClassr Z client_classr Zcreation_classr! Zfeatures_classr" Zintrospection_classr# Z ops_classr% Zvalidation_classr r< rc rV r r� r r r� r� r7 r� r� r� r� r� r r� r� r r_ r� r� r� r� r
r� r� r r� �propertyry �
__classcell__r4 r4 r� r5 ra � s�
�$
�� 6
�
2 ra c @ s� e Zd ZdZedeeee d�dd��Z edeeee
d�dd��Z e e
eef d�d d
�Ze e
eef d�d d �Zeed�d
d�ZdS )r z Custom converter for Django.N)r) �dscr* c C s t �| �d��S )zXReturn MySQL TIME data type as datetime.time()
Returns datetime.time()
zutf-8)r Z
parse_time�decode�r) r� r4 r4 r5 �_time_to_pythonh s z$DjangoMySQLConverter._time_to_pythonc C s | rt �| �S dS )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_pythonr� r4 r4 r5 r� p s z(DjangoMySQLConverter._datetime_to_pythonr( c C s
| � |�S r9 �Z
_str_to_mysql�r; r) r4 r4 r5 �_safestring_to_mysql� s z)DjangoMySQLConverter._safestring_to_mysqlc C s
| � |�S r9 r� r� r4 r4 r5 �_safetext_to_mysql� s z'DjangoMySQLConverter._safetext_to_mysqlc C s
| � |�S r9 )Z_bytes_to_mysqlr� r4 r4 r5 �_safebytes_to_mysql� s z(DjangoMySQLConverter._safebytes_to_mysql)N)N)r[ r\ r] r^ r_ �bytesr r
r r� r r� r` r r r� r� r� r4 r4 r4 r5 r c s r )Jr^ r, r r �typingr r r r r r r
r r r
r Z django.confr Zdjango.core.exceptionsr Z django.dbr Z django.db.backends.base.baser Z django.utilsr r Zdjango.utils.functionalr Zmysql.connector.typesr Zmysql.connectorrL Zmysql.connector.conversionr Z mysql.connector.custom_typesr Zmysql.connector.poolingr r r r � ImportError�errZ_mysql_connectorr r2 �clientr Zcreationr r� r! r� r" �
operationsr# Zschemar$ Z
validationr% rM r� Z
DatabaseErrorr� rN r� Zmysql.connector.abstractsr& r' r6 r7 ra r r4 r4 r4 r5 �<module> sX
4 &
`