OwlCyberSecurity - MANAGER
Edit File: creation.cpython-39.pyc
a kH�f4= � @ sz d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlm Z d dl m Z d dl m Z d dlmZ d ZG d d � d �ZdS ) � N)�StringIO)�apps)�settings)� serializers)�router)�atomic)� import_stringZtest_c @ s� e Zd ZdZdd� Zdd� Zdd� Zd.d d �Zd d� Zdd� Z dd� Z dd� Z dd� Z d/dd�Z d0dd�Zd1dd �Zd d � Zd2d d �Zd3d"d#�Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd!S )4�BaseDatabaseCreationzs Encapsulate backend-specific differences pertaining to creation and destruction of the test database. c C s || _ d S �N)� connection)�selfr � r �b/home/gouroczh/virtualenv/test/3.9/lib/python3.9/site-packages/django/db/backends/base/creation.py�__init__ s z BaseDatabaseCreation.__init__c C s | j �� S r )r � _nodb_cursor�r r r r r s 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 sN ddl m} | �� }|dkrBd}|r(d}| �d|| �||�f � | �|||� | j�� |tj | jj d< || jj d< zl| jj d d d u r�tj }d d � t �� D �t_ |d t|d d�d | jj dd� W | jj d d d u r�|t_ n | jj d d d u �r|t_ 0 |�r| �� | j_|d| jj d� | j�� tj�d�dk�rJ| �� |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 ZCreatingzUsing existing� %s test database for alias %s...�NAME�TESTZMIGRATEFc S s i | ] }|j d �qS r )�label)�.0Zappr r r � <dictcomp>H s z7BaseDatabaseCreation.create_test_db.<locals>.<dictcomp>ZmigrateT)� verbosity� interactive�databaseZ run_syncdbZcreatecachetable)r# ZRUNNING_DJANGOS_TEST_SUITE�true)Zdjango.core.managementr �_get_test_db_namer �_get_database_display_str�_create_test_dbr �closer � DATABASES�alias� settings_dictZMIGRATION_MODULESr �get_app_configs�max�serialize_db_to_stringZ_test_serialized_contentsZensure_connectionr �environ�get� mark_expected_failures_and_skips) r r! � autoclobber� serialize�keepdbr �test_database_name�actionZold_migration_modulesr r r �create_test_db sR ��� � �� 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 r+ )r Zprimary_settings_dictr r r �set_as_test_mirrorj 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 ur |j|jv r |jt j vr |� � D ]L}|j � � j�rLt�� jj|�rL|j�� jj��|j jj�}|�� E d H qLq d S )Nr )�MigrationLoader)Zdjango.db.migrations.loaderr9 r r r, Z models_moduler Z migrated_apps�namer ZTEST_NON_SERIALIZED_APPSZ get_models�_metaZ can_migrater Zallow_migrate_modelr* Z _base_manager�usingZorder_by�pk�iterator)r9 �loaderZ app_config�modelZquerysetr r r � get_objectsy s* � � � ����z@BaseDatabaseCreation.serialize_db_to_string.<locals>.get_objects�jsonN)�indent�stream)r r r3 �getvalue)r rA �outr r r r. q s z+BaseDatabaseCreation.serialize_db_to_stringc C s� t |�}t� }t| jjd��| | j�� �D tjd|| jjd�D ] }|�� |� |j j j j � q>W d � n1 sr0 Y | jj|d� W d � n1 s�0 Y dS )zw Reload the database with data from a string generated by the serialize_db_to_string() method. )r< rB N)� table_names)r �setr r r* Zconstraint_checks_disabledr Z deserialize�save�add�object� __class__r; Zdb_tableZcheck_constraints)r �datarG �objr r r �deserialize_db_from_string� s � 2z/BaseDatabaseCreation.deserialize_db_from_stringc C s d| j j|dkrd| ndf S )zR Return display string for a database for use in various actions. z'%s'%s� z ('%s')� )r r* )r r! Z database_namer r r r& � 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 r+ �TEST_DATABASE_PREFIXr r r r r% � s z&BaseDatabaseCreation._get_test_db_namec C s |� d| � d S )Nz%CREATE DATABASE %(dbname)s %(suffix)s)�execute)r �cursor� parametersr4 r r r �_execute_create_test_db� s z,BaseDatabaseCreation._execute_create_test_dbc C sp | � � }| jj�|�| �� d�}| �� ��0}z| �|||� W �n t�yJ } z�|rt|W Y d}~W d � S | �d| � |s�t d| �}|s�|dk�r"z@|dkr�| �d| � ||�f � |� d| � | �|||� W n> t�y } z$| �d | � t � d � W Y d}~n d}~0 0 n| �d � t � d� W Y d}~n d}~0 0 W d � n1 �sb0 Y |S ) zF Internal implementation - create the test db tables. )Zdbname�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.)r% r �ops� quote_name�sql_table_creation_suffixr rV � Exceptionr �inputr& rS r �exit) r r! r2 r4 r5 Ztest_db_paramsrT �eZconfirmr r r r' � sH � ������" @z$BaseDatabaseCreation._create_test_dbc C sL | j jd }|dkr:d}|r d}| �d|| �||�f � | �|||� dS )z( Clone a test database. r r zCloning test databasezUsing existing clonez%s for alias %s...N)r r+ r r& �_clone_test_db)r rW r! r2 r4 Zsource_database_namer6 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 r+ �format)r rW Zorig_settings_dictr r r �get_test_db_clone_settings s ��z/BaseDatabaseCreation.get_test_db_clone_settingsc C s t d��dS )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.N)�NotImplementedError)r rW r! r4 r r r r` s �z#BaseDatabaseCreation._clone_test_dbNc C s� | j �� |du r | j jd }n| �|�d }|dkr\d}|rBd}| �d|| �||�f � |sl| �||� |dur�|tj| j j d< || j jd< dS )zv Destroy a test database, prompting the user for confirmation if the database already exists. Nr r Z DestroyingZ Preservingr ) r r( r+ rc r r& �_destroy_test_dbr r) r* )r Zold_database_namer! r4 rW r5 r6 r r r �destroy_test_db s( ��� z$BaseDatabaseCreation.destroy_test_dbc C sD | � � �(}|�d| jj�|� � W d � n1 s60 Y dS )zF Internal implementation - remove the test db tables. zDROP DATABASE %sN)r rS r rY rZ )r r5 r! rT r r r re ; s �z%BaseDatabaseCreation._destroy_test_dbc C s� ddl m}m} | jjjD ]N}|�d�\}}}|�d�d }|tj v rt |�}t ||�} t |||| �� q| jjj �� D ]`\} } | D ]R}|�d�\}}}|�d�d }|tj v r�t |�}t ||�} t |||| �| �� q�qxdS )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 )�expectedFailure�skip�.N)Zunittestrg rh r �featuresZ django_test_expected_failures� rpartition�splitr ZINSTALLED_APPSr �getattr�setattrZdjango_test_skips�items) r rg rh Z test_nameZtest_case_name�_Ztest_method_nameZtest_appZ test_caseZ test_method�reasonZtestsr r r r1 H 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[ b 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. ZHOSTZPORTZENGINE)r r+ r% )r r+ r r r �test_db_signatureh s �z&BaseDatabaseCreation.test_db_signaturec C s* | � t|��}| jj�|� | j�� d S r )rc �strr r+ �updater( )r Z _worker_idr+ r r r �setup_worker_connectionv s z,BaseDatabaseCreation.setup_worker_connection)r FTF)F)F)r FF)F)Nr FN)�__name__� __module__� __qualname__�__doc__r r r r7 r8 r. rO r& r% rV r' ra rc r` rf re r1 r[ rr ru r r r r r s. � J / � # r )r r �ior Z django.appsr Z django.confr Z django.corer Z django.dbr Zdjango.db.transactionr Zdjango.utils.module_loadingr rR r r r r r �<module> s