o [h=@szddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZd ZGd d d ZdS) N)StringIO)apps)settings) serializers)router)atomic) import_stringZtest_c@seZdZdZddZddZddZ d.d d Zd dZddZ ddZ ddZ ddZ d/ddZ d/ddZd0ddZddZd/dd Z !d1d"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd!S)2BaseDatabaseCreationzs Encapsulate backend-specific differences pertaining to creation and destruction of the test database. cCs ||_dSN) connection)selfr r /home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/django/db/backends/base/creation.py__init__ zBaseDatabaseCreation.__init__cCs |jSr )r _nodb_cursorr r r rrrz!BaseDatabaseCreation._nodb_cursorcCstj|tjdSr )sysstderrwriteoslinesep)r msgr r rlogszBaseDatabaseCreation.logFTc CsHddlm}|}|dkr!d}|rd}|d||||f|||||j|tj |jj d<||jj d<z6|jj dd d urTtj }d d t Dt_ |d t|ddd |jj ddW|jj dd d urr|t_ n|jj dd d ur|t_ w|r||j_|d|jj d|jtjddkr||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 Hsz7BaseDatabaseCreation.create_test_db..ZmigrateT) verbosity interactivedatabaseZ run_syncdbZcreatecachetable)r$ZRUNNING_DJANGOS_TEST_SUITEtrue)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_connectionrenvironget mark_expected_failures_and_skips) r r" autoclobber serializekeepdbrtest_database_nameactionZold_migration_modulesr r rcreate_test_db sT       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 rset_as_test_mirrorjsz'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]G}|jdurW|j|jvrW|jt j vrW| D]/}|j jrVtjj|rV|jjj|j jj}|jrKdnd}|j|dEdHq'qdS)Nr)MigrationLoaderi) chunk_size)Zdjango.db.migrations.loaderr:r rr-Z models_modulerZ migrated_appsnamerZTEST_NON_SERIALIZED_APPSZ get_models_metaZ can_migraterZallow_migrate_modelr+Z _base_managerusingZorder_bypkZ_prefetch_related_lookupsiterator)r:loaderZ app_configmodelZquerysetr;rr r get_objectsys.         z@BaseDatabaseCreation.serialize_db_to_string..get_objectsjsonN)indentstream)rrr4getvalue)r rCoutr rrr/qs z+BaseDatabaseCreation.serialize_db_to_stringc Cst|}t}t|jjd?|j"tjd||jjdD]}|| |j j j j qWdn1s9wY|jj|dWddS1sPwYdS)zw Reload the database with data from a string generated by the serialize_db_to_string() method. )r>rDN) table_names)rsetrr r+Zconstraint_checks_disabledrZ deserializesaveaddobject __class__r=Zdb_tableZcheck_constraints)r datarIobjr r rdeserialize_db_from_strings   "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 r"Z database_namer r rr's 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. rr)r r,TEST_DATABASE_PREFIXrr r rr&sz&BaseDatabaseCreation._get_test_db_namecCs|d|dS)Nz%CREATE DATABASE %(dbname)s %(suffix)s)execute)r cursor parametersr5r r r_execute_create_test_dbsz,BaseDatabaseCreation._execute_create_test_dbc Cst|}|jj||d}|}z ||||Wnty}zu|r7|WYd}~WdS|d||sFt d|}|sL|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: %srRzTests cancelled.)r&r ops quote_namesql_table_creation_suffixrrX Exceptionrinputr'rUrexit) r r"r3r5r6Ztest_db_paramsrVeZconfirmr r rr(s`       # ##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 rYr"r3r5Zsource_database_namer7r 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 rYZorig_settings_dictr r rget_test_db_clone_settingss z/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 rYr"r5r r rrbsz#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,rerr'_destroy_test_dbrr*r+)r Zold_database_namer"r5rYr6r7r r rdestroy_test_dbs*   z$BaseDatabaseCreation.destroy_test_dbcCsF|}|d|jj|WddS1swYdS)zF Internal implementation - remove the test db tables. zDROP DATABASE %sN)rrUr r[r\)r r6r"rVr r rrg>s "z%BaseDatabaseCreation._destroy_test_dbc Csddlm}m}|jjjD]'}|d\}}}|dd}|tj vr4t |}t ||} t |||| q |jjj D]0\} } | D])}|d\}}}|dd}|tj vrkt |}t ||} t |||| | qBqs