File: //usr/local/CyberPanel/lib64/python3.10/site-packages/django/db/__pycache__/utils.cpython-310.pyc
o
�h?$ � @ 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 �E/usr/local/CyberPanel/lib/python3.10/site-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 1 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__C 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__F 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 5 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>w s ��z load_backend.<locals>.<listcomp>c S s g | ]}d | �qS )zdjango.db.backends.%sr )r; �br r r r? | 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_backendg s* � ��� ��rQ c s8 e Zd ZdZdZ� fdd�Zedd� �Zdd� Z� Z S ) �ConnectionHandler� DATABASESTc s t � �|�}|i krddi|t< nt|vr tdt� d���|t i kr)d|t d< |�� D ]\}|�dd� |�dd� |�dd� |d d ksK|d sOd|d< |�d
d � |�d d� |�d
i � |�dd � dD ]}|�|d� qi|�di �}g d�}|D ]
\}}|�||� q~q-|S )N�ENGINEzdjango.db.backends.dummyzYou must define a 'z ' database.�ATOMIC_REQUESTSF�
AUTOCOMMITTzdjango.db.backends.� CONN_MAX_AGEr �CONN_HEALTH_CHECKS�OPTIONS� TIME_ZONE)�NAME�USER�PASSWORD�HOST�PORT� �TEST))�CHARSETN)� COLLATIONN)�MIGRATET)�MIRRORN)r[ N)�super�configure_settings�DEFAULT_DB_ALIASr �values�
setdefault)r � databases�conn�setting�
test_settings�default_test_settings�key�value�� __class__r r rg � s6
� �z$ConnectionHandler.configure_settingsc C s | j S r" r r# r r r rk � s zConnectionHandler.databasesc C s" | j | }t|d �}|�||�S )NrT )r rQ �DatabaseWrapper)r �aliasrE �backendr r r �create_connection� s
z#ConnectionHandler.create_connection)
r r r �
settings_name�thread_criticalrg �propertyrk rw �
__classcell__r r rr 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& �AttributeError�get�_staterE rh )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 rh �DJANGO_VERSION_PICKLE_KEY� Exceptionr
r r r r r r r r r rQ rR r| r r r r �<module> s. 2#;