h=ddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZd ZGd d Zy) N)StringIO)apps)settings) serializers)router)atomic) import_stringtest_ceZdZdZdZdZdZ ddZdZdZ dZ d Z d Z dd Z dd Zdd ZdZddZ ddZdZdZdZdZdZy)BaseDatabaseCreationzs Encapsulate backend-specific differences pertaining to creation and destruction of the test database. c||_yN) connection)selfrs ]/var/www/django_project/virt/lib/python3.12/site-packages/django/db/backends/base/creation.py__init__zBaseDatabaseCreation.__init__s $c6|jjSr)r _nodb_cursorrs rrz!BaseDatabaseCreation._nodb_cursors++--rcdtjj|tjzyr)sysstderrwriteoslinesep)rmsgs rlogzBaseDatabaseCreation.logs rzz)*rcnddlm}|j}|dk\r-d}|rd}|j|d|j ||d|j ||||j j|tj|j jd<||j jd< |j jd d d urFtj}tjD cic]} | jd c} t_ |d t!|dz dd |j jd|j jd d d ur t_ |r|j#|j _|d|j j|j j't(j*j-ddk(r|j/|Scc} w#|j jd d d ur t_ wwxYw)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_commandCreatingzUsing existing test database for alias ...NAMETESTMIGRATEFNmigrateT) verbosity interactivedatabase run_syncdbcreatecachetable)r+RUNNING_DJANGOS_TEST_SUITEtrue)django.core.managementr _get_test_db_namer_get_database_display_str_create_test_dbrcloser DATABASESalias settings_dictMIGRATION_MODULESrget_app_configslabelmaxserialize_db_to_string_test_serialized_contentsensure_connectionrenvironget mark_expected_failures_and_skips) rr) autoclobber serializekeepdbr test_database_nameactionold_migration_modulesapps rcreate_test_dbz#BaseDatabaseCreation.create_test_db s 8!335 >F) HH229>PQ  Y V< >6 76 A  1 1 3!!A.,,V4Y?5H-B*Is:AH>H:HH,H4c<|d|jjd<y)z Set this database up to be used in testing as a mirror of a primary database whose settings are given. r%N)rr7)rprimary_settings_dicts rset_as_test_mirrorz'BaseDatabaseCreation.set_as_test_mirrorjs 1Ff0M%%f-rc~fd}t}tjd|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. c3Kddlm}|j}tjD]2}|j |j |jvs*|jtjvsG|jD]}|jjjs)tjjj |sT|j"j%jj j'|jj(j}|j*rdnd}|j-|Ed{5y7 w)Nr)MigrationLoaderi) chunk_size)django.db.migrations.loaderrOrrr9 models_moduler: migrated_appsnamerTEST_NON_SERIALIZED_APPS get_models_meta can_migraterallow_migrate_modelr6 _base_managerusingorder_bypk_prefetch_related_lookupsiterator)rOloader app_configmodelquerysetrPrs r get_objectsz@BaseDatabaseCreation.serialize_db_to_string..get_objectsys$ C$T__5F"224 P ,,8"((F,@,@@"x/P/PP!+!6!6!8 P ;;22 OO$889N9NPUV',':':'@'@ $ 5 5(&hu{{~~':':;%)1(J(JPT'(0'8'8J'8'OOO P  P Ps0=E)E)E)78E)0*E)BE)E' E)jsonN)indentstream)rrrCgetvalue)rrdouts` rr<z+BaseDatabaseCreation.serialize_db_to_stringqs4 P.jfkmDM||~rc$t|}t}t|jj5|jj 5t jd||jjD]K}|j|j|jjjjM ddd|jj|dddy#1swY.xYw#1swYyxYw)zw Reload the database with data from a string generated by the serialize_db_to_string() method. )r[reN) table_names)rsetrrr6constraint_checks_disabledr deserializesaveaddobject __class__rWdb_tablecheck_constraints)rdatarkobjs rdeserialize_db_from_stringz/BaseDatabaseCreation.deserialize_db_from_strings ~e $//// 0 G;;= I&22D(=(=ICHHJOOCJJ$8$8$>$>$G$GH I I OO - -+ - F G G I I G Gs$DA;C: $D:D ?DDcRd|jjd|dk\rd|zSdS)zR Return display string for a database for use in various actions. 'z ('%s'))rr6)rr) database_names rr2z.BaseDatabaseCreation._get_database_display_strs< OO ! !+4>Y & A  ?A A  rc|jjddr|jjddSt|jjdzS)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. r&r%)rr7TEST_DATABASE_PREFIXrs rr1z&BaseDatabaseCreation._get_test_db_namesO ?? ( ( 0 8??008@ @#doo&C&CF&KKKrc,|jd|zy)Nz%CREATE DATABASE %(dbname)s %(suffix)s)execute)rcursor parametersrDs r_execute_create_test_dbz,BaseDatabaseCreation._execute_create_test_dbs>KLrc  |j}|jjj||j d}|j 5} |j |||ddd|S#t$r}|r|cYd}~cdddS|jd|z|std|z}|sdk(r |dk\r%|jd|j||d|jd |z|j |||ne#t$r3}|jd |ztjd Yd}~n.d}~wwxYw|jd tjdYd}~d}~wwxYw#1swY|SxYw) zF Internal implementation - create the test db tables. )dbnamesuffixNz+Got an error creating the test database: %szXType 'yes' if you would like to try deleting the test database '%s', or 'no' to cancel: yesr!z'Destroying old test database for alias r$zDROP DATABASE %(dbname)sz-Got an error recreating the test database: %srzzTests cancelled.)r1rops quote_namesql_table_creation_suffixrr Exceptionrinputr2rrexit) rr)rBrDrEtest_db_paramsreconfirms rr3z$BaseDatabaseCreation._create_test_dbs"335oo))445GH446     ! F ,,V^VL! F"!A --! ! FJK"#=?QRG'U"2$$> HH%)$B$B(13E%&!"'AN'RS44V^VT$$!PST!TU $HH/0HHQK= ! F"!smFA66 FE;FF+E;>ADE; E )EE;E  )E;5F;FFF c|jjd}|dk\r-d}|rd}|j|d|j||d|j |||y)z( Clone a test database. r%r!zCloning test databasezUsing existing clonez for alias r$N)rr7rr2_clone_test_db)rrr)rBrDsource_database_namerFs r clone_test_dbz"BaseDatabaseCreation.clone_test_dbsh $<,F/ HH229>RS  FIv6rcd|jj}i|ddj|d|iS)zX Return a modified connection settings dict for the n-th clone of a DB. r%z{}_{})rr7format)rrorig_settings_dicts rget_test_db_clone_settingsz/BaseDatabaseCreation.get_test_db_clone_settingssA"__::   GNN#5f#=vF  rctd)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)rrr)rDs rrz#BaseDatabaseCreation._clone_test_dbs" E  rNc|jj||jjd}n|j|d}|dk\r-d}|rd}|j |d|j ||d|s|j |||D|tj|jjd<||jjd<yy)zv Destroy a test database, prompting the user for confirmation if the database already exists. Nr%r! Destroying Preservingr#r$) rr4r7rrr2_destroy_test_dbrr5r6)rold_database_namer)rDrrErFs rdestroy_test_dbz$BaseDatabaseCreation.destroy_test_dbs  >!%!>!>v!F !%!@!@!H!P  >!F% HH229>PQ   ! !"4i @  (@QH  t44 5f =4EDOO ) )& 1 )rc|j5}|jd|jjj |zdddy#1swYyxYw)zF Internal implementation - remove the test db tables. zDROP DATABASE %sN)rrrrr)rrEr)rs rrz%BaseDatabaseCreation._destroy_test_db>sQ    F NN"T__%8%8%C%CDV%WW    s 8AAc |ddlm}m}|jjj D]h}|j d\}}}|jdd}|tjvs?t|}t||} t|||| j|jjjjD]x\} } | D]n}|j d\}}}|jdd}|tjvs?t|}t||} t|||| | pzy)z Mark tests in Django's test suite which are expected failures on this database and test which should be skipped on this database. r)expectedFailureskip.N)unittestrrrfeaturesdjango_test_expected_failures rpartitionsplitrINSTALLED_APPSr getattrsetattrdjango_test_skipsitems) rrr test_nametest_case_name_test_method_nametest_app test_case test_methodreasontestss rrAz5BaseDatabaseCreation.mark_expected_failures_and_skipsKs3 311OO SI2;2F2Fs2K /NA/ s+A.H8222).9 %i1AB  #3_[5QR S"__55GGMMO TMFE" T 6?6J6J36O3#3$??3/2x666 -n =I"))5E"FKI'7fk9RS T Trcy)zQ SQL to append to the end of the test table creation statements. r{rs rrz.BaseDatabaseCreation.sql_table_creation_suffixesrch|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. HOSTPORTENGINE)rr7r1)rr7s rtest_db_signaturez&BaseDatabaseCreation.test_db_signatureksA 55 & ! & ! ( #  " " $   rc|jt|}|jjj ||jj yr)rstrrr7updater4)r _worker_idr7s rsetup_worker_connectionz,BaseDatabaseCreation.setup_worker_connectionys@77JH %%,,]; r)r!FTF)F)r!FF)Nr!FN)__name__ __module__ __qualname____doc__rrrrIrLr<rwr2r1rr3rrrrrrArrrrrrr r s %.+FKH"TN!FG*  LM-"^7,   IM!FF T4   rr )rrior django.appsr django.confr django.corer django.dbrdjango.db.transactionrdjango.utils.module_loadingr r~r rrrrs4  #(5o o r