a xd 3@snddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dZGd d d ZdS) N)StringIO)apps)settings) serializers)router)atomicZtest_c@seZdZdZddZddZddZd*d d Zd dZddZ ddZ ddZ ddZ d+ddZ d,ddZd-ddZddZd.dd Zd/d"d#Zd$d%Zd&d'Zd(d)Zd!S)0BaseDatabaseCreationzs Encapsulate backend-specific differences pertaining to creation and destruction of the test database. cCs ||_dSN) connection)selfr r X/var/www/html/Ranjet/env/lib/python3.9/site-packages/django/db/backends/base/creation.py__init__szBaseDatabaseCreation.__init__cCs |jSr )r _nodb_cursorr r r r rsz!BaseDatabaseCreation._nodb_cursorcCstj|tjdSr )sysstderrwriteoslinesep)r msgr r r logszBaseDatabaseCreation.logFTc Cs4ddlm}|}|dkrBd}|r(d}|d||||f|||||j|tj |jj d<||jj d<zl|jj dd d urtj }d d t Dt_ |d t|ddd |jj ddW|jj dd d ur|t_ n|jj dd d ur|t_ 0|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 existing %s test database for alias %s...NAMETESTZMIGRATEFcSsi|] }|jdqSr )label).0Zappr r r Asz7BaseDatabaseCreation.create_test_db..ZmigrateT) verbosity interactivedatabaseZ run_syncdbZcreatecachetable)r")Zdjango.core.managementr_get_test_db_namer_get_database_display_str_create_test_dbr closer DATABASESalias settings_dictZMIGRATION_MODULESrget_app_configsmaxserialize_db_to_stringZ_test_serialized_contentsZensure_connection) r r autoclobber serializekeepdbrtest_database_nameactionZold_migration_modulesr r r create_test_dbsH        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_mirrorasz'BaseDatabaseCreation.set_as_test_mirrorcs.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. c3sddlm}|j}tD]|}|jdur|j|jvr|jt j vr| D]L}|j jrLtjj|rL|jjj|j jj}|EdHqLqdS)Nr)MigrationLoader)Zdjango.db.migrations.loaderr4r rr*Z models_modulerZ migrated_appsnamerZTEST_NON_SERIALIZED_APPSZ get_models_metaZ can_migraterZallow_migrate_modelr(Z_default_managerusingZorder_bypkiterator)r4loaderZ app_configmodelZquerysetrr r get_objectsos(       z@BaseDatabaseCreation.serialize_db_to_string..get_objectsjsonN)indentstream)rrr.getvalue)r r<outr rr r,hs z+BaseDatabaseCreation.serialize_db_to_stringc Cst|}t}t|jjd||jDtjd||jjdD]}|| |j j j j q>Wdn1sr0Y|jj|dWdn1s0YdS)zw Reload the database with data from a string generated by the serialize_db_to_string() method. )r7r=N) table_names)rsetrr r(Zconstraint_checks_disabledrZ deserializesaveaddobject __class__r6Zdb_tableZcheck_constraints)r datarBobjr r r deserialize_db_from_strings 2z/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 r Z database_namer r r r$sz.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. rr)r r)TEST_DATABASE_PREFIXrr r r r#sz&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 Csp|}|jj||d}|0}z||||Wn tyJ}z|rt|WYd}~WdS|d||st d|}|s|dkr"z@|dkr|d| ||f| d|||||Wn>ty}z$|d |t d WYd}~n d}~00n|d t dWYd}~n d}~00Wdn1sb0Y|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: %srKzTests cancelled.)r#r ops quote_namesql_table_creation_suffixrrQ Exceptionrinputr$rNrexit) r r r-r/r0Ztest_db_paramsrOeZconfirmr r r r%s>   " @z$BaseDatabaseCreation._create_test_dbcCsL|jjd}|dkr:d}|r d}|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 rRr r-r/Zsource_database_namer1r 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 rRZorig_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 rRr r/r r r r[sz#BaseDatabaseCreation._clone_test_dbNcCs|j|dur |jjd}n||d}|dkr\d}|rBd}|d||||f|sl||||dur|tj|jj d<||jjd<dS)zv Destroy a test database, prompting the user for confirmation if the database already exists. NrrZ DestroyingZ Preservingr) r r&r)r^rr$_destroy_test_dbrr'r()r Zold_database_namer r/rRr0r1r r r destroy_test_dbs"   z$BaseDatabaseCreation.destroy_test_dbcCsD|(}|d|jj|Wdn1s60YdS)zF Internal implementation - remove the test db tables. zDROP DATABASE %sN)rrNr rTrU)r r0r rOr r r r`s  z%BaseDatabaseCreation._destroy_test_dbcCsdS)zQ SQL to append to the end of the test table creation statements. rLr rr r r rV(sz.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_signature.s z&BaseDatabaseCreation.test_db_signature)rFTF)F)F)rFF)F)NrFN)__name__ __module__ __qualname____doc__rrrr2r3r,rJr$r#rQr%r\r^r[rar`rVrbr r r r rs& C   )    r)rriorZ django.appsrZ django.confrZ django.corerZ django.dbrZdjango.db.transactionrrMrr r r r s