3 [x.@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 D/tmp/pip-install-j7i95hgp/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}|j}|dkr@d}|r(d}td||j||f|j||||jj|tj |jj d<||jj d<|dt |ddd |jj d d |r|j |j_|d |jj d |jj|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@tjD]4}|jdk r$|j|jkr$|jt j kr$j |dfq$Wfdd}t }t jd|d|d|jS)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`xZtjD]L}|jjjr tjjj|r |jj jjj |jj j }|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] }|jqWdS)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|jd|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<|j}|jjj||jd}|jj}y|j|||Wntk r,}z|r\|St j j d||s|t d|}|s|dkr y>|dkrt d|j||f|jd||j|||Wn>tk r}z t j j d|t jd WYd d }~XnXnt d t jdWYd d }~XnXWd 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||j||f|j|||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}|ddj|d|iS)zX Return a modified connection settings dict for the n-th clone of a DB. rz{}_{})rrformat)r r@Zorig_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|jj|dkr |jjd}n|j|d}|dkrZd}|rBd}td||j||f|sj|j|||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...) rrrrOrr_destroy_test_dbrrr)r Zold_database_namerr r@r!r"r r r destroy_test_dbs   z$BaseDatabaseCreation.destroy_test_dbc Cs4|jjj}|jd|jjj|WdQRXdS)zF Internal implementation - remove the test db tables. zDROP DATABASE %sN)rr r=r<rBrC)r r!rr=r r r rQsz%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|jfS)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?rrMrOrLrRrQrDrSr r r r rs$  8    +    r) rFiorZ django.appsrZ django.confrZ django.corerZ django.dbrr;rr r r r s