Xw/@sddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dZGd d d eZdS) N)apps)settings) serializers)router)StringIO)inputZtest_c@seZdZdZddZeddZddddd d Zd d Zd dZ ddZ ddZ ddZ dddZ dddddZddZdddZddddddZd 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 N/home/ubuntu/projects/ifolica/build/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_connectionFTc Csddlm}|j}|dkr`d}|r=d}ntd||j||fn|j||||jj|tj |jj d<||jj d<|dd t |ddd d d |jj d d|r|j |j_n|dd |jj |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...NAMEZmigrate verbosity interactiveFZdatabaseZ run_syncdbTZcreatecachetable)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.get_objectsjsonindentstream)Zdjango.db.migrations.loaderr%r rZget_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_stringcCsCt|}x0tjd|d|jjD]}|jq+WdS)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}s %z/BaseDatabaseCreation.deserialize_db_from_stringcCs*d|jj|dkr"d|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 rs z.BaseDatabaseCreation._get_database_display_strcCs8|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 #Cs|j}|j}|jjj}|jj`}y!|jd|||fWn5tk r}z|ry|St j j d||st d|} n|s| dkrhyd|dkrt d|j||fn|jd|||jd|||fWqtk rd}z&t j j d|t jd WYd d }~XqXnt d t jdWYd d }~XnXWd QX|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 r9NzTests cancelled.)sql_table_creation_suffixrr ops quote_namercursorexecute Exceptionsysstderrwriterrrexit) r rrr suffixr!qnr@eZconfirmr r r rsB         # &z$BaseDatabaseCreation._create_test_dbcCsk|jjd}|dkrTd}|r1d}ntd||j||fn|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_dbcCs6|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 rKZorig_settings_dictZnew_settings_dictr r r get_test_db_clone_settingss  z/BaseDatabaseCreation.get_test_db_clone_settingscCstddS)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 rKrr r r r rJsz#BaseDatabaseCreation._clone_test_dbNcCs|jj|dkr,|jjd}n|j|d}|dkrd}|r`d}ntd||j||fn|s|j||n|dk r|tj|jj d<||jjdr?)r r!rr@r r r rQs  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)__name__ __module__ __qualname____doc__r propertyrr#r$rr8rrrrLrOrJrRrQr=rUr r r r rs"  8   0   r)rCrSZ django.appsrZ django.confrZ django.corerZ django.dbrZdjango.utils.sixrZdjango.utils.six.movesrr;objectrr r r r s