3 (hw/@spddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dZGd d d eZdS) N)apps)settings) serializers)router)StringIO)inputZtest_c@seZdZdZddZeddZd&d d Zd d Zd dZ ddZ ddZ ddZ d'ddZ d(ddZddZd)ddZd*ddZd d!Zd"d#Zd$d%ZdS)+BaseDatabaseCreationz This class encapsulates all backend-specific differences that pertain to creation and destruction of the test database. cCs ||_dS)N) connection)selfr r D/tmp/pip-install-q3hcpn_q/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 rsz%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 Creates a test database, prompting the user for confirmation if the database already exists. Returns 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_dbr closer 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)r r)r Zprimary_settings_dictr r r set_as_test_mirrorWsz'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 Serializes all data in the database into a JSON string. Designed only for test runner usage; will not handle large amounts of data. r)MigrationLoaderNc3sjxdtjD]V}|jjjr tjjj|r |jj jjj |jj j }x|j D] }|VqTWq WdS)N)rZsort_dependenciesZ_metaZ can_migrater rZallow_migrate_modelrZ_default_managerusingZorder_bypknameiterator)modelZquerysetobj)app_listr r r get_objectsqs z@BaseDatabaseCreation.serialize_db_to_string..get_objectsjson)indentstream)Zdjango.db.migrations.loaderr&r rZget_app_configsZ models_modulelabelZ migrated_appsr)rZTEST_NON_SERIALIZED_APPSappendrrr getvalue)r r&loaderZ app_configr.outr )r-r r r^s     z+BaseDatabaseCreation.serialize_db_to_stringcCs2t|}x$tjd||jjdD] }|jqWdS)zv Reloads the database with data from a string generated by the serialize_db_to_string method. r/)r'N)rrZ deserializer rsave)r datar,r r r deserialize_db_from_string}sz/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'))r r)r rZ database_namer r r rsz.BaseDatabaseCreation._get_database_display_strcCs0|jjddr |jjddSt|jjdS)z Internal implementation - returns 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)r rTEST_DATABASE_PREFIX)r r r r rsz&BaseDatabaseCreation._get_test_db_namec #CsJ|j}|j}|jjj}|jj}y|jd|||fWntk r:}z|r^|St j j d||s~t d|} |s| dkryJ|dkrt d|j||f|jd|||jd|||fWn>tk r}z t j j d|t jd WYd d }~XnXnt d t jdWYd d }~XnXWd QRX|S) zG Internal implementation - creates the test db tables. zCREATE DATABASE %s %sz,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 %sz.Got an error recreating the test database: %s r:NzTests cancelled.)sql_table_creation_suffixrr ops quote_namercursorexecute Exceptionsysstderrwriterrrexit) r rrr!suffixr"qnrAeZconfirmr r r rsB   &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)r rrr_clone_test_db)r numberrrr!Zsource_database_namer#r r r clone_test_dbs z"BaseDatabaseCreation.clone_test_dbcCs(|jj}|j}dj|d||d<|S)zX Return a modified connection settings dict for the n-th clone of a DB. z{}_{}r)r rcopyformat)r rLZorig_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 rLrr!r r r rKsz#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...) r rrrPrr_destroy_test_dbrrr)r Zold_database_namerr!rLr"r#r r r destroy_test_dbs   z$BaseDatabaseCreation.destroy_test_dbc Cs>|jjj(}tjd|jd|jjj|WdQRXdS)zF Internal implementation - remove the test db tables. rzDROP DATABASE %sN)r rrAtimesleeprBr?r@)r r"rrAr r r rRs z%BaseDatabaseCreation._destroy_test_dbcCsdS)zQ SQL to append to the end of the test table creation statements. r;r )r r r r r>sz.BaseDatabaseCreation.sql_table_creation_suffixcCs$|jj}|d|d|d|jfS)z Returns 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 rr)r rr r r test_db_signature#s z&BaseDatabaseCreation.test_db_signature)rFTF)F)rFF)F)NrFN)__name__ __module__ __qualname____doc__r propertyrr$r%rr9rrrrMrPrKrSrRr>rVr r r r rs"  8   0   r)rDrTZ django.appsrZ django.confrZ django.corerZ django.dbrZdjango.utils.sixrZdjango.utils.six.movesrr<objectrr r r r s