o بf.@sbddlZddlZddlmZddlmZddlmZddlm Z ddl m Z dZ Gdd d Z dS) N)StringIO)apps)settings) serializers)routerZtest_c@seZdZdZddZeddZddZd*d d Zd dZ ddZ ddZ ddZ ddZ d+ddZd+ddZd,ddZddZd+dd Zd-d"d#Zd$d%Zd&d'Zd(d)Zd!S).BaseDatabaseCreationzs Encapsulate backend-specific differences pertaining to creation and destruction of the test database. cCs ||_dSN) connection)selfr r X/var/www/dscantool/venv/lib/python3.10/site-packages/django/db/backends/base/creation.py__init__s zBaseDatabaseCreation.__init__cCs|jjS)zH Used to be defined here, now moved to DatabaseWrapper. )r _nodb_connectionr r r r rsz%BaseDatabaseCreation._nodb_connectioncCstj|tjdSr)sysstderrwriteoslinesep)r msgr r r logszBaseDatabaseCreation.logFTcCsddlm}|}|dkr!d}|rd}|d||||f|||||j|tj |jj d<||jj d<|dt |ddd |jj d d |rT| |j_|d |jj d |j|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_commandrZCreatingzUsing existing %s test database for alias %s...NAMEZmigrateFT) verbosity interactivedatabaseZ run_syncdbZcreatecachetable)r)Zdjango.core.managementr_get_test_db_namer_get_database_display_str_create_test_dbr closer DATABASESalias settings_dictmaxserialize_db_to_stringZ_test_serialized_contentsZensure_connection)r r autoclobber serializekeepdbrtest_database_nameactionr r r create_test_db!s4       z#BaseDatabaseCreation.create_test_dbcCs|d|jjd<dS)z Set this database up to be used in testing as a mirror of a primary database whose settings are given. rN)r r$)r Zprimary_settings_dictr r r set_as_test_mirrorYsz'BaseDatabaseCreation.set_as_test_mirrorcsddlm}|j}gtD]}|jdur+|j|jvr+|jt j vr+ |dfqfdd}t }t jd|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. r)MigrationLoaderNc3s^tD]&}|jjr,tjj|r,|j jj |jj j }| EdHqdSr)rZsort_dependencies_metaZ can_migrater rZallow_migrate_modelr#Z_default_managerusingZorder_bypknameiterator)modelZquerysetZapp_listr r r get_objectsssz@BaseDatabaseCreation.serialize_db_to_string..get_objectsjson)indentstream)Zdjango.db.migrations.loaderr.r rZget_app_configsZ models_modulelabelZ migrated_appsr2rZTEST_NON_SERIALIZED_APPSappendrrr(getvalue)r r.loaderZ app_configr6outr r5r r&`s      z+BaseDatabaseCreation.serialize_db_to_stringcCs.t|}tjd||jjdD]}|qdS)zw Reload the database with data from a string generated by the serialize_db_to_string() method. r7)r0N)rrZ deserializer r#save)r dataobjr r r deserialize_db_from_string~s z/BaseDatabaseCreation.deserialize_db_from_stringcCs$d|jj|dkrd|fSdfS)zR Return display string for a database for use in various actions. z'%s'%sz ('%s'))r r#)r rZ database_namer r r rs z.BaseDatabaseCreation._get_database_display_strcCs0|jjddr|jjddSt|jjdS)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. TESTr)r r$TEST_DATABASE_PREFIXrr r r rsz&BaseDatabaseCreation._get_test_db_namecCs|d|dS)Nz%CREATE DATABASE %(dbname)s %(suffix)s)execute)r cursor parametersr)r r r _execute_create_test_dbsz,BaseDatabaseCreation._execute_create_test_dbc Csv|}|jj||d}|j}z ||||Wnty}zu|r8|WYd}~WdS| d||sGt d|}|sM|dkrz |dkr^| d| ||f| d|||||Wn(ty}z| d |t d WYd}~nd}~ww| d t dWYd}~n d}~wwWd|SWd|S1swY|S) zF Internal implementation - create the test db tables. )ZdbnamesuffixNz+Got an error creating the test database: %szXType 'yes' if you would like to try deleting the test database '%s', or 'no' to cancel: yesrz,Destroying old test database for alias %s...zDROP DATABASE %(dbname)sz-Got an error recreating the test database: %srCzTests cancelled.)rr ops quote_namesql_table_creation_suffixrrHrJ ExceptionrinputrrGrexit) r rr'r)r*Ztest_db_paramsrHeZconfirmr r r r sV         z$BaseDatabaseCreation._create_test_dbcCsL|jjd}|dkrd}|rd}|d||||f||||dS)z( Clone a test database. rrzCloning test databasezUsing existing clonez%s for alias %s...N)r r$rr_clone_test_db)r rKrr'r)Zsource_database_namer+r r r clone_test_dbs  z"BaseDatabaseCreation.clone_test_dbcCs$|jj}i|dd|d|iS)zX Return a modified connection settings dict for the n-th clone of a DB. rz{}_{})r r$format)r rKZorig_settings_dictr r r get_test_db_clone_settingssz/BaseDatabaseCreation.get_test_db_clone_settingscCstd)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 rKrr)r r r rTsz#BaseDatabaseCreation._clone_test_dbNcCs|j|dur|jjd}n||d}|dkr.d}|r!d}|d||||f|s6||||durK|tj|jj d<||jjd<dSdS)zv Destroy a test database, prompting the user for confirmation if the database already exists. NrrZ DestroyingZ Preservingr) r r!r$rWrr_destroy_test_dbrr"r#)r Zold_database_namerr)rKr*r+r r r destroy_test_dbs$   z$BaseDatabaseCreation.destroy_test_dbcCsJ|jj}|d|jj|WddS1swYdS)zF Internal implementation - remove the test db tables. zDROP DATABASE %sN)r rrHrGrMrN)r r*rrHr r r rY s  "z%BaseDatabaseCreation._destroy_test_dbcCsdS)zQ SQL to append to the end of the test table creation statements. rDr rr r r rOsz.BaseDatabaseCreation.sql_table_creation_suffixcCs$|jj}|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_signatures z&BaseDatabaseCreation.test_db_signature)rFTF)F)rFF)NrFN)__name__ __module__ __qualname____doc__r propertyrrr,r-r&rBrrrJr rUrWrTrZrYrOr[r r r r rs*  8   )  r)rriorZ django.appsrZ django.confrZ django.corerZ django.dbrrFrr r r r s