|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/django/db/backends/base/__pycache__/ |
Upload File : |
o
3�a�` � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dl Z d dl
m Z d dl m
Z
d dlmZmZ d dlmZ d dlmZ d d lmZ d d
lmZ d d lmZ d d lm Z d d
l m Z d dl m Z dZ!G dd� d�Z"dS )� N)�deque)�contextmanager)�settings)�ImproperlyConfigured)�DEFAULT_DB_ALIAS�
DatabaseError)�utils)�BaseDatabaseValidation)�connection_created)�TransactionManagementError��DatabaseErrorWrapper)�timezone)� async_unsafe)�cached_property� __no_db__c @ s~ e Zd ZdZi Zi Zi ZdZdZdZ dZ
dZ dZ dZ
dZdZeZdZefdd�Zdd� Zed d
� �Zed d � �Zed
d� �Zedd� �Zdd� Z dd� Z dd� Z d|dd�Z e dd� �Z!dd � Z"e d d � �Z#d d � Z$d|d!d"�Z%d#d$� Z&d%d&� Z'd'd(� Z(e d)d*� �Z)e d+d,� �Z*e d-d.� �Z+e d/d0� �Z,d1d2� Z-d3d4� Z.d5d6� Z/d7d8� Z0e d9d:� �Z1e d;d<� �Z2e d=d>� �Z3e d?d@� �Z4dAdB� Z5dCdD� Z6d}dFdG�Z7dHdI� Z8dJdK� Z9dLdM� Z:dNdO� Z;e<dPdQ� �Z=dRdS� Z>dTdU� Z?d|dVdW�Z@dXdY� ZAdZd[� ZBed\d]� �ZCd^d_� ZDd`da� ZEdbdc� ZFddde� ZGedfdg� �ZHdhdi� ZIdjdk� ZJdldm� ZKe<dndo� �ZLe<dpdq� �ZMdrds� ZNdtdu� ZOdvdw� ZPe<dxdy� �ZQd|dzd{�ZRdS )~�BaseDatabaseWrapperz Represent a database connection.N�unknowni(# c C s� d | _ || _|| _t| jd�| _d| _d| _d| _d| _ g | _
d| _ d| _ d | _
d| _d| _t�� | _d| _t�� | _g | _d| _g | _| �| �| _| � | �| _ | � | �| _ | � | �| _!| �"| �| _#| �$| �| _%d S )N)�maxlenFr T)&�
connection�
settings_dict�aliasr �
queries_limit� queries_log�force_debug_cursor�
autocommit�in_atomic_block�savepoint_state�
savepoint_ids�commit_on_exit�needs_rollback�close_at�closed_in_transaction�errors_occurred� threading�Lock�_thread_sharing_lock�_thread_sharing_count�_thread� get_ident�
_thread_ident�
run_on_commit�%run_commit_hooks_on_set_autocommit_on�execute_wrappers� client_class�client�creation_class�creation�features_class�features�introspection_class�
introspection� ops_class�ops�validation_class�
validation)�selfr r � r; �>/usr/lib/python3/dist-packages/django/db/backends/base/base.py�__init__0 s4
z BaseDatabaseWrapper.__init__c C � dS )z
Ensure the connection's timezone is set to `self.timezone_name` and
return whether it changed or not.
Fr; �r: r; r; r< �ensure_timezonen � z#BaseDatabaseWrapper.ensure_timezonec C s. t jsdS | jd du rtjS t�| jd �S )a
Return a tzinfo of the database connection time zone.
This is only used when time zone support is enabled. When a datetime is
read from the database, it is always returned in this time zone.
When the database backend supports time zones, it doesn't matter which
time zone Django uses, as long as aware datetimes are used everywhere.
Other users connecting to the database can choose their own time zone.
When the database backend doesn't support time zones, the time zone
Django uses may be constrained by the requirements of other users of
the database.
N� TIME_ZONE)r �USE_TZr r �utc�pytzr? r; r; r< r u s
z BaseDatabaseWrapper.timezonec C s( t jst jS | jd du rdS | jd S )zC
Name of the time zone of the database connection.
rB N�UTC)r rC rB r r? r; r; r< �
timezone_name� s
z!BaseDatabaseWrapper.timezone_namec C s | j ptjS �N)r r �DEBUGr? r; r; r< �queries_logged� � z"BaseDatabaseWrapper.queries_loggedc C s0 t | j�| jjkrt�d�| jj�� t| j�S )NzLLimit for query logging exceeded, only the last {} queries will be returned.)�lenr r �warnings�warn�format�listr? r; r; r< �queries� s
�
zBaseDatabaseWrapper.queriesc C � t d��)z<Return a dict of parameters suitable for get_new_connection.zNsubclasses of BaseDatabaseWrapper may require a get_connection_params() method��NotImplementedErrorr? r; r; r< �get_connection_params� � z)BaseDatabaseWrapper.get_connection_paramsc C rR )z"Open a connection to the database.zKsubclasses of BaseDatabaseWrapper may require a get_new_connection() methodrS )r: � conn_paramsr; r; r< �get_new_connection� rV z&BaseDatabaseWrapper.get_new_connectionc C rR )z,Initialize the database connection settings.zOsubclasses of BaseDatabaseWrapper may require an init_connection_state() methodrS r? r; r; r< �init_connection_state� rV z)BaseDatabaseWrapper.init_connection_statec C rR )z9Create a cursor. Assume that a connection is established.zFsubclasses of BaseDatabaseWrapper may require a create_cursor() methodrS �r: �namer; r; r< �
create_cursor� rV z!BaseDatabaseWrapper.create_cursorc C s� | � � d| _g | _d| _| jd }|du rdnt�� | | _d| _d| _ | �
� }| � |�| _ | �
| jd � | �� tj| j| d� g | _dS )z>Connect to the database. Assume that the connection is closed.F� CONN_MAX_AGEN�
AUTOCOMMIT)�senderr )�check_settingsr r r r �time� monotonicr! r" r# rU rX r �set_autocommitrY r
�send� __class__r+ )r: �max_agerW r; r; r< �connect� s
zBaseDatabaseWrapper.connectc C s* | j d d urtjstd| j ��d S d S )NrB z=Connection '%s' cannot set TIME_ZONE because USE_TZ is False.)r r rC r r r? r; r; r< r` � s ���z"BaseDatabaseWrapper.check_settingsc C sB | j du r | j�
| �� W d � dS 1 sw Y dS dS )z;Guarantee that a connection to the database is established.N)r �wrap_database_errorsrg r? r; r; r< �ensure_connection� s
"��z%BaseDatabaseWrapper.ensure_connectionc C s* | � � | jr| �|�}|S | �|�}|S )zY
Validate the connection is usable and perform database cursor wrapping.
)�validate_thread_sharingrJ �make_debug_cursor� make_cursor)r: �cursor�wrapped_cursorr; r; r< �_prepare_cursor� s
�z#BaseDatabaseWrapper._prepare_cursorc C sB | � � | j� | �| �|��W d � S 1 sw Y d S rH )ri rh ro r\ rZ r; r; r< �_cursor� s $�zBaseDatabaseWrapper._cursorc C �B | j d ur | j�
| j �� W d � S 1 sw Y d S d S rH )r rh �commitr? r; r; r< �_commit� �
$��zBaseDatabaseWrapper._commitc C rq rH )r rh �rollbackr? r; r; r< � _rollback� rt z BaseDatabaseWrapper._rollbackc C rq rH )r rh �closer? r; r; r< �_close� rt zBaseDatabaseWrapper._closec C � | � � S )z3Create a cursor, opening a connection if necessary.)rp r? r; r; r< rm s zBaseDatabaseWrapper.cursorc C s( | � � | �� | �� d| _d| _dS )z.Commit a transaction and reset the dirty flag.FTN)rj �validate_no_atomic_blockrs r# r, r? r; r; r< rr s
zBaseDatabaseWrapper.commitc C s. | � � | �� | �� d| _d| _g | _dS )z1Roll back a transaction and reset the dirty flag.FN)rj rz rv r# r r+ r? r; r; r< ru s
z BaseDatabaseWrapper.rollbackc C sj | � � g | _| js| jdu rdS z| �� W | jr"d| _d| _dS d| _dS | jr1d| _d| _w d| _w )z%Close the connection to the database.NT)rj r+ r" r rx r r r? r; r; r< rw s
�
zBaseDatabaseWrapper.closec C �@ | � � �}|�| j�|�� W d � d S 1 sw Y d S rH )rm �executer7 �savepoint_create_sql�r: �sidrm r; r; r< �
_savepoint0 �
"�z BaseDatabaseWrapper._savepointc C r{ rH )rm r| r7 �savepoint_rollback_sqlr~ r; r; r< �_savepoint_rollback4 r� z'BaseDatabaseWrapper._savepoint_rollbackc C r{ rH )rm r| r7 �savepoint_commit_sqlr~ r; r; r< �_savepoint_commit8 r� z%BaseDatabaseWrapper._savepoint_commitc C s | j jo| �� S rH )r3 �uses_savepoints�get_autocommitr? r; r; r< �_savepoint_allowed< s z&BaseDatabaseWrapper._savepoint_allowedc C sV | � � sdS t�� }t|��dd�}| jd7 _d|| jf }| �� | �|� |S )z�
Create a savepoint inside the current transaction. Return an
identifier for the savepoint that will be used for the subsequent
rollback or commit. Do nothing if savepoints are not supported.
N�-� � zs%s_x%d)r� r( r) �str�replacer rj r� )r: � thread_ident�tidr r; r; r< � savepointB s
z BaseDatabaseWrapper.savepointc s8 | � � sdS | �� | �� � � fdd�| jD �| _dS )zW
Roll back to a savepoint. Do nothing if savepoints are not supported.
Nc s g | ] \}}� |vr||f�qS r; r; )�.0�sids�func�r r; r< �
<listcomp>c s z:BaseDatabaseWrapper.savepoint_rollback.<locals>.<listcomp>)r� rj r� r+ �r: r r; r� r< �savepoint_rollbackW s
�z&BaseDatabaseWrapper.savepoint_rollbackc C s" | � � sdS | �� | �|� dS )zR
Release a savepoint. Do nothing if savepoints are not supported.
N)r� rj r� r� r; r; r< �savepoint_commitg s z$BaseDatabaseWrapper.savepoint_commitc C s
d| _ dS )zY
Reset the counter used to generate unique savepoint ids in this thread.
r N)r r? r; r; r< �clean_savepointsr s
z$BaseDatabaseWrapper.clean_savepointsc C rR )zR
Backend-specific implementation to enable or disable autocommit.
zHsubclasses of BaseDatabaseWrapper may require a _set_autocommit() methodrS )r: r r; r; r< �_set_autocommit{ s z#BaseDatabaseWrapper._set_autocommitc C s | � � | jS )zGet the autocommit state.)ri r r? r; r; r< r� � s z"BaseDatabaseWrapper.get_autocommitFc C sf | � � | �� |o| ot| d�}|r| �� n| �|� || _|r/| jr1| �� d| _dS dS dS )a�
Enable or disable autocommit.
The usual way to start a transaction is to turn autocommit off.
SQLite does not properly start a transaction when disabling
autocommit. To avoid this buggy behavior and to actually enter a new
transaction, an explicit BEGIN is required. Using
force_begin_transaction_with_broken_autocommit=True will issue an
explicit BEGIN with SQLite. This option will be ignored for other
backends.
�#_start_transaction_under_autocommitFN)rz ri �hasattrr� r� r r, �run_and_clear_commit_hooks)r: r �.force_begin_transaction_with_broken_autocommit�"start_transaction_under_autocommitr; r; r< rc � s
�
�z"BaseDatabaseWrapper.set_autocommitc C s | j std��| jS )z9Get the "needs rollback" flag -- for *advanced use* only.�<The rollback flag doesn't work outside of an 'atomic' block.�r r r r? r; r; r< � get_rollback� s
�z BaseDatabaseWrapper.get_rollbackc C s | j std��|| _dS )zT
Set or unset the "needs rollback" flag -- for *advanced use* only.
r� Nr� )r: ru r; r; r< � set_rollback� s
�
z BaseDatabaseWrapper.set_rollbackc C s | j rtd��dS )z,Raise an error if an atomic block is active.z3This is forbidden when an 'atomic' block is active.N)r r r? r; r; r< rz � s
��z,BaseDatabaseWrapper.validate_no_atomic_blockc C s | j rtd��d S )NzlAn error occurred in the current transaction. You can't execute queries until the end of the 'atomic' block.)r r r? r; r; r< � validate_no_broken_transaction� s
��z2BaseDatabaseWrapper.validate_no_broken_transactionc c s8 � | � � }zdV W |r| �� dS dS |r| �� w w )z:
Disable foreign key constraint checking.
N)�disable_constraint_checking�enable_constraint_checking)r: �disabledr; r; r< �constraint_checks_disabled� s � �
�z.BaseDatabaseWrapper.constraint_checks_disabledc C r> )z�
Backends can implement as needed to temporarily disable foreign key
constraint checking. Should return True if the constraints were
disabled and will need to be reenabled.
Fr; r? r; r; r< r� � � z/BaseDatabaseWrapper.disable_constraint_checkingc C r> )zh
Backends can implement as needed to re-enable foreign key constraint
checking.
Nr; r? r; r; r< r� � rA z.BaseDatabaseWrapper.enable_constraint_checkingc C r> )z�
Backends can override this method if they can apply constraint
checking (e.g. via "SET CONSTRAINTS ALL IMMEDIATE"). Should raise an
IntegrityError if any invalid foreign key references are encountered.
Nr; )r: � table_namesr; r; r< �check_constraints� r� z%BaseDatabaseWrapper.check_constraintsc C rR )a
Test if the database connection is usable.
This method may assume that self.connection is not None.
Actual implementations should take care not to raise exceptions
as that may prevent Django from recycling unusable connections.
zCsubclasses of BaseDatabaseWrapper may require an is_usable() methodrS r? r; r; r< � is_usable� s �z BaseDatabaseWrapper.is_usablec C sz | j dur7| �� | jd kr| �� dS | jr%| �� r d| _n| �� dS | jdur9t�� | jkr;| �� dS dS dS dS )z
Close the current connection if unrecoverable errors have occurred
or if it outlived its maximum age.
Nr^ F) r r� r rw r# r� r! ra rb r? r; r; r< � close_if_unusable_or_obsolete� s
�z1BaseDatabaseWrapper.close_if_unusable_or_obsoletec C s4 | j �
| jdkW d � S 1 sw Y d S �Nr �r& r' r? r; r; r< �allow_thread_sharing s $�z(BaseDatabaseWrapper.allow_thread_sharingc C s: | j � | jd7 _W d � d S 1 sw Y d S )Nr� r� r? r; r; r< �inc_thread_sharing s "�z&BaseDatabaseWrapper.inc_thread_sharingc C sL | j � | jdkr
td��| jd8 _W d � d S 1 s w Y d S )Nr z5Cannot decrement the thread sharing count below zero.r� )r& r' � RuntimeErrorr? r; r; r< �dec_thread_sharing s
"�z&BaseDatabaseWrapper.dec_thread_sharingc C s6 | j s| jt�� kstd| j| jt�� f ��dS dS )a3
Validate that the connection isn't accessed by another thread than the
one which originally created it, unless the connection was explicitly
authorized to be shared between threads (via the `inc_thread_sharing()`
method). Raise an exception if the validation fails.
z�DatabaseWrapper objects created in a thread can only be used in that same thread. The object with alias '%s' was created in thread id %s and this is thread id %s.N)r� r* r( r) r r r? r; r; r< rj s ���z+BaseDatabaseWrapper.validate_thread_sharingc C r> )z~
Hook to do any database check or preparation, generally called before
migrating a project or an app.
Nr; r? r; r; r< �prepare_database2 rA z$BaseDatabaseWrapper.prepare_databasec C s t | �S )z�
Context manager and decorator that re-throws backend-specific database
exceptions using Django's common wrappers.
r r? r; r; r< rh 9 s z(BaseDatabaseWrapper.wrap_database_errorsc C ry )z�
Return a cursor that tries to avoid caching in the database (if
supported by the database), otherwise return a regular cursor.
)rm r? r; r; r< �chunked_cursorA s z"BaseDatabaseWrapper.chunked_cursorc C � t �|| �S )z:Create a cursor that logs all queries in self.queries_log.)r �CursorDebugWrapper�r: rm r; r; r< rk H rK z%BaseDatabaseWrapper.make_debug_cursorc C r� )z&Create a cursor without debug logging.)r �
CursorWrapperr� r; r; r< rl L rK z BaseDatabaseWrapper.make_cursorc c sb � | j du }z"| �� � }|V W d � n1 sw Y W |r'| �� dS dS |r0| �� w w )a8
Context manager that ensures that a connection is established, and
if it opened one, closes it to avoid leaving a dangling connection.
This is useful for operations outside of the request-response cycle.
Provide a cursor: with self.temporary_connection() as cursor: ...
N)r rm rw )r: �
must_closerm r; r; r< �temporary_connectionP s �
�� �
�z(BaseDatabaseWrapper.temporary_connectionc c st � | j i | j�ddi�td�}z%|�� � }|V W d � n1 s"w Y W |�� dS W |�� dS |�� w )aq
Return a cursor from an alternative connection to be used when there is
no need to access the main database, specifically for test db
creation/deletion. This also prevents the production database from
being exposed to potential child threads while (or after) the test
database is destroyed. Refs #10868, #17786, #16969.
�NAMEN)r )re r � NO_DB_ALIASrm rw )r: �connrm r; r; r< � _nodb_cursora s �
� �z BaseDatabaseWrapper._nodb_cursorc O s* | j du r td��| j | g|�R i |��S )zG
Return a new instance of this backend's SchemaEditor.
NzFThe SchemaEditorClass attribute of this database wrapper is still None)�SchemaEditorClassrT )r: �args�kwargsr; r; r< �
schema_editorq s
�z!BaseDatabaseWrapper.schema_editorc C sJ t |�std��| jr| j�t| j�|f� d S | �� s td��|� d S )Nz*on_commit()'s callback must be a callable.z;on_commit() cannot be used in manual transaction management) �callable� TypeErrorr r+ �append�setr r� r )r: r� r; r; r< � on_commitz s
z BaseDatabaseWrapper.on_commitc C s8 | � � | j}g | _|r|�d�\}}|� |s d S d S r� )rz r+ �pop)r: �current_run_on_commitr� r� r; r; r< r� � s �z.BaseDatabaseWrapper.run_and_clear_commit_hooksc c s2 � | j �|� z dV W | j �� dS | j �� w )z|
Return a context manager under which the wrapper is applied to suitable
database query executions.
N)r- r� r� )r: �wrapperr; r; r< �execute_wrapper� s
� z#BaseDatabaseWrapper.execute_wrapperc C s( t �| j�}|du r
| j}t| �||�S )zy
Return a copy of this connection.
For tests that require two connections to the same database.
N)�copy�deepcopyr r �type)r: r r r; r; r< r� � s zBaseDatabaseWrapper.copyrH )F)S�__name__�
__module__� __qualname__�__doc__�
data_types�data_types_suffix�data_type_check_constraintsr7 �vendor� display_namer� r. r0 r2 r4 r6 r r8 r r r= r@ r r rG �propertyrJ rQ rU rX rY r\ r rg r` ri ro rp rs rv rx rm rr ru rw r� r� r� r� r� r� r� r� r� r� rc r� r� rz r� r r� r� r� r� r� r� r� r� r� rj r� rh r� rk rl r� r� r� r� r� r� r� r; r; r; r< r s� >
r )#r( r� r$ ra rM � collectionsr �
contextlibr rE � django.confr �django.core.exceptionsr � django.dbr r �django.db.backendsr �"django.db.backends.base.validationr �django.db.backends.signalsr
�django.db.transactionr �django.db.utilsr
� django.utilsr �django.utils.asyncior �django.utils.functionalr r� r r; r; r; r< �<module> s*