|
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/__pycache__/ |
Upload File : |
o
�a~& � @ s, d dl Z d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dl
m Z d dl m
Z
d Zd
ZG d d � d e�ZG d
d� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd � d e�ZG d d � d �Zd d � ZG d!d"� d"e �Z G d#d$� d$�Z dS )%� N)�
import_module��settings)�ImproperlyConfigured)�ConnectionDoesNotExist)�BaseConnectionHandler)�cached_property)�
import_string�default�_django_versionc @ � e Zd ZdS )�ErrorN��__name__�
__module__� __qualname__� r r �1/usr/lib/python3/dist-packages/django/db/utils.pyr
� r
c @ r )�InterfaceErrorNr r r r r r r r c @ r )�
DatabaseErrorNr r r r r r r r c @ r )� DataErrorNr r r r r r r r c @ r )�OperationalErrorNr r r r r r r r c @ r )�IntegrityErrorNr r r r r r $ r r c @ r )�
InternalErrorNr r r r r r ( r r c @ r )�ProgrammingErrorNr r r r r r , r r c @ r )�NotSupportedErrorNr r r r r r 0 r r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �DatabaseErrorWrapperz~
Context manager and decorator that reraises backend-specific database
exceptions using Django's common wrappers.
c C �
|| _ dS )zx
wrapper is a database wrapper.
It must have a Database attribute defining PEP-249 exceptions.
N)�wrapper)�selfr r r r �__init__: s
z DatabaseErrorWrapper.__init__c C s d S �Nr �r r r r � __enter__B s z DatabaseErrorWrapper.__enter__c C sp |d u rd S t ttttttttf D ]$}t | j
j |j �}t
||�r5||j� }|t tfvr/d| j
_|�|�|�qd S �NT)r r r r r r r r r
�getattrr �Databaser �
issubclass�args�errors_occurred�with_traceback)r �exc_type� exc_value� traceback� dj_exc_type� db_exc_type� dj_exc_valuer r r �__exit__E s( �
��z DatabaseErrorWrapper.__exit__c s � �fdd�}|S )Nc s6 �� � | i |��W d � S 1 sw Y d S r" r )r) �kwargs��funcr r r �inner_ s $�z,DatabaseErrorWrapper.__call__.<locals>.innerr )r r5 r6 r r4 r �__call__\ s z DatabaseErrorWrapper.__call__N)r r r �__doc__r! r$ r2 r7 r r r r r 4 s r c
C s� | dkrd} zt d| �W S tyG } z.ddl}dd� t�|jjj�D �}| dd� |D �vrBtt t
|��}t d | d
� |�f �|�� d}~ww ) z�
Return a database backend's "base" module given a fully qualified database
backend name, or raise an error if it doesn't exist.
z&django.db.backends.postgresql_psycopg2z django.db.backends.postgresqlz%s.baser Nc S s" g | ]
\}}}|r|d vr|�qS )> �base�dummyr )�.0�_�name�ispkgr r r �
<listcomp>t s ��z load_backend.<locals>.<listcomp>c S s g | ]}d | �qS )zdjango.db.backends.%sr )r; �br r r r? x s z�%r isn't an available database backend or couldn't be imported. Check the above exception. To use one of the built-in backends, use 'django.db.backends.XXX', where XXX is one of:
%sz, )
r � ImportError�django.db.backends�pkgutil� iter_modules�db�backends�__path__�map�repr�sortedr �join)� backend_name�e_user�django�builtin_backends�
backend_reprsr r r � load_backende s* � ��� ��rQ c sP e Zd ZdZdZ� fdd�Zedd� �Zdd� Zd d
� Z d d � Z
d
d� Z � Z S )�ConnectionHandler� DATABASESTc sX t � �|�}|i krddi|t< |S t|vr tdt� d���|t i kr*d|t d< |S )N�ENGINE�django.db.backends.dummyzYou must define a 'z ' database.)�super�configure_settings�DEFAULT_DB_ALIASr )r � databases�� __class__r r rW � s �
� z$ConnectionHandler.configure_settingsc C s | j S r" r r# r r r rY � s zConnectionHandler.databasesc C s� z| j | }W n ty | �d|� d���w |�dd� |�dd� |�dd� |d d ks4|d s8d|d< |�d
d � |�d i � |�d
d� dD ]}|�|d� qLdS )z}
Put the defaults into the settings dictionary for a given connection
where no settings is provided.
�The connection '�' doesn't exist.�ATOMIC_REQUESTSF�
AUTOCOMMITTrT rU zdjango.db.backends.� CONN_MAX_AGEr �OPTIONS� TIME_ZONEN)�NAME�USER�PASSWORD�HOST�PORT� �rY �KeyError�exception_class�
setdefault)r �alias�conn�settingr r r �ensure_defaults� s � �z!ConnectionHandler.ensure_defaultsc C sb z| j | }W n ty | �d|� d���w |�di �}g d�}|D ]
\}}|�||� q$dS )zY
Make sure the test settings are available in the 'TEST' sub-dictionary.
r\ r] �TEST))�CHARSETN)� COLLATIONN)�MIGRATET)�MIRRORN)rc NNri )r rm rn �
test_settings�default_test_settings�key�valuer r r �prepare_test_settings� s � �z'ConnectionHandler.prepare_test_settingsc C s6 | � |� | �|� | j| }t|d �}|�||�S )NrT )rp rz rY rQ �DatabaseWrapper)r rm rE �backendr r r �create_connection� s
z#ConnectionHandler.create_connectionc C s: | D ]}zt | j|�}W n ty Y qw |�� qd S r" )r& � _connections�AttributeError�close)r rm �
connectionr r r � close_all� s �
�zConnectionHandler.close_all)
r r r �
settings_name�thread_criticalrW �propertyrY rp rz r} r� �
__classcell__r r rZ r rR � s
rR c @ s\ e Zd Zddd�Zedd� �Zdd� Zed�Zed �Zd
d � Z d d
� Z
dd� Z ddd�Z dS )�ConnectionRouterNc C r )zT
If routers is not specified, default to settings.DATABASE_ROUTERS.
N)�_routers)r �routersr r r r! � s
zConnectionRouter.__init__c C sJ | j d u r tj| _ g }| j D ]}t|t�rt|�� }n|}|�|� q|S r" )r� r �DATABASE_ROUTERS�
isinstance�strr �append)r r� �r�routerr r r r� � s
zConnectionRouter.routersc s � fdd�}|S )Nc st d }| j D ]!}zt|� �}W n ty Y qw ||fi |��}|r&| S q|�d�}|d ur8|jjr8|jjS tS )N�instance)r� r& r �get�_staterE rX )r �model�hints� chosen_dbr� �methodr� ��actionr r � _route_db� s
��
z0ConnectionRouter._router_func.<locals>._route_dbr )r� r� r r� r � _router_func� s z ConnectionRouter._router_func� db_for_read� db_for_writec K s\ | j D ]"}z|j}W n ty Y qw |||fi |��}|d ur%| S q|jj|jjkS r" )r� �allow_relationr r� rE )r �obj1�obj2r� r� r� �allowr r r r� s
��z ConnectionRouter.allow_relationc K sP | j D ]"}z|j}W n ty Y qw |||fi |��}|d ur%| S qdS r% )r� �
allow_migrater )r rE � app_labelr� r� r� r� r r r r� s
��z ConnectionRouter.allow_migratec C s | j ||jj|jj|d�S )N)�
model_namer� )r� �_metar� r� )r rE r� r r r �allow_migrate_model s �z$ConnectionRouter.allow_migrate_modelFc s |j |d�}� �fdd�|D �S )z8Return app models allowed to be migrated on provided db.)�include_auto_createdc s g | ]
}�� � |�r|�qS r )r� )r; r� �rE r r r r? ( s z:ConnectionRouter.get_migratable_models.<locals>.<listcomp>)�
get_models)r �
app_configrE r� �modelsr r� r �get_migratable_models% s z&ConnectionRouter.get_migratable_modelsr" )F)
r r r r! r r� r� r� r� r� r� r� r� r r r r r� � s
r� ) rC � importlibr � django.confr �django.core.exceptionsr �django.utils.connectionr r �django.utils.functionalr �django.utils.module_loadingr rX �DJANGO_VERSION_PICKLE_KEY� Exceptionr
r r r r r r r r r rQ rR r� r r r r �<module> s. 1!R