B 0H[.@sZddlZddlmZddlmZddlmZddlmZddl m Z dZ Gdd d Z dS) N)StringIO)apps)settings) serializers)routerZtest_c@seZdZdZddZeddZd(d d Zd d Zd dZ ddZ ddZ ddZ d)ddZ d*ddZd+ddZddZd,ddZd-d d!Zd"d#Zd$d%Zd&d'ZdS).BaseDatabaseCreationzs Encapsulate backend-specific differences pertaining to creation and destruction of the test database. cCs ||_dS)N) connection)selfrr dC:\Users\HIRONO~1\AppData\Local\Temp\pip-install-6bm3nxem\django\django\db\backends\base\creation.py__init__szBaseDatabaseCreation.__init__cCs|jjS)zH Used to be defined here, now moved to DatabaseWrapper. )r_nodb_connection)r r r r r sz%BaseDatabaseCreation._nodb_connectionFTcCsddlm}|}|dkr@d}|r(d}td||||f|||||j|tj |jj d<||jj d<|dt |ddd |jj d d |r| |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 existingz %s test database for alias %s...NAMEZmigrateFT) verbosity interactivedatabaseZ run_syncdbZcreatecachetable)r)Zdjango.core.managementr_get_test_db_nameprint_get_database_display_str_create_test_dbrcloser DATABASESalias settings_dictmaxserialize_db_to_stringZ_test_serialized_contentsZensure_connection)r r autoclobber serializekeepdbrtest_database_nameactionr r r create_test_dbs0      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)rr)r Zprimary_settings_dictr r r set_as_test_mirrorUsz'BaseDatabaseCreation.set_as_test_mirrorcsddlm}|j}gx@tD]4}|jdk r$|j|jkr$|jt j kr$ |dfq$Wfdd}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`xZtD]L}|jjr tjj|r |j jj |jj j }| EdHq WdS)N)rZsort_dependenciesZ_metaZ can_migraterrZallow_migrate_modelrZ_default_managerusingZorder_bypknameiterator)modelZqueryset)app_listr r r get_objectsos z@BaseDatabaseCreation.serialize_db_to_string..get_objectsjson)indentstream)Zdjango.db.migrations.loaderr%rrZget_app_configsZ models_modulelabelZ migrated_appsr(rZTEST_NON_SERIALIZED_APPSappendrrrgetvalue)r r%loaderZ app_configr,outr )r+r r r\s     z+BaseDatabaseCreation.serialize_db_to_stringcCs2t|}x$tjd||jjdD] }|qWdS)zw Reload the database with data from a string generated by the serialize_db_to_string() method. r-)r&N)rrZ deserializerrsave)r dataobjr r r deserialize_db_from_stringzsz/BaseDatabaseCreation.deserialize_db_from_stringcCs d|jj|dkrd|ndfS)zR Return display string for a database for use in various actions. z'%s'%sz ('%s'))rr)r rZ database_namer r r rsz.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. ZTESTr)rrTEST_DATABASE_PREFIX)r r 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 Cs<|}|jj||d}|j}y||||Wntk r,}z|r\|St j d||s|t d|}|s|dkr y>|dkrt d|||f|d|||||Wn>tk r}zt j d|t d Wd d }~XYnXnt d t dWd d }~XYnXWd QRX|S) zF Internal implementation - create the test db tables. )Zdbnamesuffixz,Got an error creating the test database: %s zXType '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: %s r9NzTests cancelled.)rrops quote_namesql_table_creation_suffixr r=r? Exceptionsysstderrwriteinputrrr<exit) r rrr r!Ztest_db_paramsr=eZconfirmr r r rs:    &z$BaseDatabaseCreation._create_test_dbcCsJ|jjd}|dkr8d}|r d}td||||f||||dS)z( Clone a test database. rrzCloning test databasezUsing existing clonez%s for alias %s...N)rrrr_clone_test_db)r r@rrr Zsource_database_namer"r r r clone_test_dbs z"BaseDatabaseCreation.clone_test_dbcCs(|jj}|}d|d||d<|S)zX Return a modified connection settings dict for the n-th clone of a DB. z{}_{}r)rrcopyformat)r r@Zorig_settings_dictZnew_settings_dictr r r get_test_db_clone_settingssz/BaseDatabaseCreation.get_test_db_clone_settingscCs tddS)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 r@rr r r r rLsz#BaseDatabaseCreation._clone_test_dbNcCs|j|dkr |jjd}n||d}|dkrZd}|rBd}td||||f|sj||||dk r|tj|jj d<||jjd<dS)zv Destroy a test database, prompting the user for confirmation if the database already exists. NrrZ DestroyingZ Preservingz %s test database for alias %s...) rrrrPrr_destroy_test_dbrrr)r Zold_database_namerr r@r!r"r r r destroy_test_dbs   z$BaseDatabaseCreation.destroy_test_dbc Cs4|jj}|d|jj|WdQRXdS)zF Internal implementation - remove the test db tables. zDROP DATABASE %sN)rr r=r<rBrC)r r!rr=r r r rR sz%BaseDatabaseCreation._destroy_test_dbcCsdS)zQ SQL to append to the end of the test table creation statements. r:r )r r r r rDsz.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)rrr)r rr r r test_db_signatures z&BaseDatabaseCreation.test_db_signature)rFTF)F)F)rFF)F)NrFN)__name__ __module__ __qualname____doc__r propertyr r#r$rr8rrr?rrMrPrLrSrRrDrTr r r r rs$  8    +    r) rFiorZ django.appsrZ django.confrZ django.corerZ django.dbrr;rr r r r s