File: //lib/python3/dist-packages/django/db/backends/base/__pycache__/creation.cpython-310.pyc
o
�a39 � @ s� d dl Z d dlZd dlmZ d dlmZmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ d dlmZ d d lmZ d
ZG d d � d �ZdS )
� N)�StringIO)�expectedFailure�skip)�apps)�settings)� serializers)�router)�atomic)�
import_string�test_c @ s� e Zd ZdZdd� Zdd� Zdd� Zd,d d �Zd
d� Zdd� Z dd� Z
dd� Z dd� Z d-dd�Z
d-dd�Zd.dd �Zd d � Zd-d d �Zd/d"d#�Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd!S )0�BaseDatabaseCreationzs
Encapsulate backend-specific differences pertaining to creation and
destruction of the test database.
c C s
|| _ d S �N)�
connection)�selfr � r �B/usr/lib/python3/dist-packages/django/db/backends/base/creation.py�__init__ �
z BaseDatabaseCreation.__init__c C s
| j �� S r
)r � _nodb_cursor�r r r r r r z!BaseDatabaseCreation._nodb_cursorc C s t j�|tj � d S r
)�sys�stderr�write�os�linesep)r �msgr r r �log s zBaseDatabaseCreation.log� FTc C sH ddl m} | �� }|dkr!d}|rd}| �d|| �||�f � | �|||� | j�� |tj | jj
d< || jj d< z6| jj d d d
u rTtj }d d � t
�� D �t_ |d
t|d d�d
| jj
dd� W | jj d d d
u rr|t_ n| jj d d d
u r�|t_ w |r�| �� | j_|d| jj
d� | j�� tj�d�dkr�| �� |S )z�
Create a test database, prompting the user for confirmation if the
database already exists. Return the name of the test database created.
r )� call_commandr �CreatingzUsing existing� %s test database for alias %s...�NAME�TEST�MIGRATEFc S s i | ]}|j d �qS r
)�label)�.0�appr r r �
<dictcomp>C s ��z7BaseDatabaseCreation.create_test_db.<locals>.<dictcomp>�migrateT)� verbosity� interactive�database�
run_syncdb�createcachetable)r+ �RUNNING_DJANGOS_TEST_SUITE�true)�django.core.managementr �_get_test_db_namer �_get_database_display_str�_create_test_dbr �closer � DATABASES�alias�
settings_dict�MIGRATION_MODULESr �get_app_configs�max�serialize_db_to_string�_test_serialized_contents�ensure_connectionr �environ�get� mark_expected_failures_and_skips) r r) � autoclobber� serialize�keepdbr �test_database_name�action�old_migration_modulesr r r �create_test_db sN
�
� ���
z#BaseDatabaseCreation.create_test_dbc C s |d | j jd< dS )z�
Set this database up to be used in testing as a mirror of a primary
database whose settings are given.
r! N)r r7 )r �primary_settings_dictr r r �set_as_test_mirrorf s z'BaseDatabaseCreation.set_as_test_mirrorc s. � fdd�}t � }tjd|� d|d� |�� S )z�
Serialize all data in the database into a JSON string.
Designed only for test runner usage; will not handle large
amounts of data.
c 3 s� � ddl m} | � j�}t�� D ]>}|jd urN|j|jv rN|jt j
vrN|� � D ]&}|j �
� j�rMt�� jj|�rM|j�� jj��|j jj�}|�� E d H q'qd S )Nr )�MigrationLoader)�django.db.migrations.loaderrJ r r r9 �
models_moduler$ �
migrated_apps�namer �TEST_NON_SERIALIZED_APPS�
get_models�_meta� can_migrater �allow_migrate_modelr6 �
_base_manager�using�order_by�pk�iterator)rJ �loader�
app_config�model�querysetr r r � get_objectst s( �
��� ���z@BaseDatabaseCreation.serialize_db_to_string.<locals>.get_objects�jsonN)�indent�stream)r r rB �getvalue)r r] �outr r r r; m s z+BaseDatabaseCreation.serialize_db_to_stringc C s� t |�}t� }t| jjd��? | j�� �" tjd|| jjd�D ]}|�� |� |j
j j j
� q W d � n1 s9w Y | jj|d� W d � dS 1 sPw Y dS )zw
Reload the database with data from a string generated by
the serialize_db_to_string() method.
)rU r^ N)� table_names)r �setr r r6 �constraint_checks_disabledr � deserialize�save�add�object� __class__rQ �db_table�check_constraints)r �datarc �objr r r �deserialize_db_from_string� s � �"�z/BaseDatabaseCreation.deserialize_db_from_stringc C s$ d| j j|dkrd| f S df S )zR
Return display string for a database for use in various actions.
z'%s'%s� z ('%s')� )r r6 )r r) �
database_namer r r r2 � s ��z.BaseDatabaseCreation._get_database_display_strc C s0 | j jd d r| j jd d S t| j jd S )z�
Internal implementation - return the name of the test DB that will be
created. Only useful when called from create_test_db() and
_create_test_db() and when no external munging is done with the 'NAME'
settings.
r" r! )r r7 �TEST_DATABASE_PREFIXr r r r r1 � s z&BaseDatabaseCreation._get_test_db_namec C s |� d| � d S )Nz%CREATE DATABASE %(dbname)s %(suffix)s)�execute)r �cursor�
parametersrC r r r �_execute_create_test_db� s z,BaseDatabaseCreation._execute_create_test_dbc C st | � � }| jj�|�| �� d�}| �� ��}z | �|||� W n� ty� } zu|r7|W Y d}~W d � S | �d| � |sFt d| �}|sL|dkr�z |dkr]| �d| �
||�f � |� d| � | �|||� W n( ty� } z| �d | � t �
d
� W Y d}~nd}~ww | �d � t �
d� W Y d}~n
d}~ww W d � |S W d � |S 1 s�w Y |S ) zF
Internal implementation - create the test db tables.
)�dbname�suffixNz+Got an error creating the test database: %szXType 'yes' if you would like to try deleting the test database '%s', or 'no' to cancel: �yesr z,Destroying old test database for alias %s...zDROP DATABASE %(dbname)sz-Got an error recreating the test database: %srp zTests cancelled.)r1 r �ops�
quote_name�sql_table_creation_suffixr rw � Exceptionr �inputr2 rt r �exit) r r) rA rC rD �test_db_paramsru �e�confirmr r r r3 � sV �
� ��
���
���
� �
� � z$BaseDatabaseCreation._create_test_dbc C sL | j jd }|dkr d}|rd}| �d|| �||�f � | �|||� dS )z(
Clone a test database.
r! r zCloning test databasezUsing existing clonez%s for alias %s...N)r r7 r r2 �_clone_test_db)r ry r) rA rC �source_database_namerE r r r �
clone_test_db� s
�z"BaseDatabaseCreation.clone_test_dbc C s$ | j j}i |�dd�|d |�i�S )zX
Return a modified connection settings dict for the n-th clone of a DB.
r! z{}_{})r r7 �format)r ry �orig_settings_dictr r r �get_test_db_clone_settings� s z/BaseDatabaseCreation.get_test_db_clone_settingsc C s t d��)zI
Internal implementation - duplicate the test db tables.
znThe database backend doesn't support cloning databases. Disable the option to run tests in parallel processes.)�NotImplementedError)r ry r) rC r r r r� � s �z#BaseDatabaseCreation._clone_test_dbNc C s� | j �� |du r| j jd }n| �|�d }|dkr.d}|r!d}| �d|| �||�f � |s6| �||� |durK|tj| j j d< || j jd< dS dS )zv
Destroy a test database, prompting the user for confirmation if the
database already exists.
Nr! r �
Destroying�
Preservingr )
r r4 r7 r� r r2 �_destroy_test_dbr r5 r6 )r �old_database_namer) rC ry rD rE r r r �destroy_test_db s$
� �z$BaseDatabaseCreation.destroy_test_dbc C sF | � � �}|�d| jj�|� � W d � dS 1 s w Y dS )zF
Internal implementation - remove the test db tables.
zDROP DATABASE %sN)r rt r r{ r| )r rD r) ru r r r r� ! s
�"�z%BaseDatabaseCreation._destroy_test_dbc
C s� | j jjD ]'}|�d�\}}}|�d�d }|tjv r,t|�}t||�}t ||t
|�� q| j jj � � D ]0\}} | D ])}|�d�\}}}|�d�d }|tjv rct|�}t||�}t ||t
|�|�� q:q4dS )z�
Mark tests in Django's test suite which are expected failures on this
database and test which should be skipped on this database.
�.r N)r �features� django_test_expected_failures�
rpartition�splitr �INSTALLED_APPSr
�getattr�setattrr �django_test_skips�itemsr )
r � test_name�test_case_name�_�test_method_name�test_app� test_case� test_method�reason�testsr r r r@ - s&
�
���z5BaseDatabaseCreation.mark_expected_failures_and_skipsc C s dS )zQ
SQL to append to the end of the test table creation statements.
rq r r r r r r} D s z.BaseDatabaseCreation.sql_table_creation_suffixc C s$ | j j}|d |d |d | �� fS )z�
Return a tuple with elements of self.connection.settings_dict (a
DATABASES setting value) that uniquely identify a database
accordingly to the RDBMS particularities.
�HOST�PORT�ENGINE)r r7 r1 )r r7 r r r �test_db_signatureJ s �z&BaseDatabaseCreation.test_db_signature)r FTF)F)r FF)Nr FN)�__name__�
__module__� __qualname__�__doc__r r r rG rI r; ro r2 r1 rw r3 r� r� r� r� r� r@ r} r� r r r r r s*
F
)
r )r r �ior �unittestr r � django.appsr � django.confr � django.corer � django.dbr �django.db.transactionr �django.utils.module_loadingr
rs r r r r r �<module> s