h 3tddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dZGd d Zy) N)StringIO)apps)settings) serializers)router)atomictest_ceZdZdZdZdZdZddZdZdZ dZ d Z d Z dd Z dd Zdd ZdZddZddZdZdZdZy)BaseDatabaseCreationzs Encapsulate backend-specific differences pertaining to creation and destruction of the test database. c||_yN) connection)selfrs c/var/www/html/ranktracker/api/venv/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)*rc ddlm}|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'|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*)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_connection) rr( autoclobber serializekeepdbrtest_database_nameactionold_migration_modulesapps rcreate_test_dbz#BaseDatabaseCreation.create_test_dbs 8!335 >F) HH..y:LM  Y V< G:GG,Hc<|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)rr4)rprimary_settings_dicts rset_as_test_mirrorz'BaseDatabaseCreation.set_as_test_mirroras 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] }|j |j |jvs*|jtjvsG|jD]}|jjjs)tjjj |sT|j"j%jj j'|jj(j}|j+Ed{#y7 w)Nr)MigrationLoader)django.db.migrations.loaderrIrrr6 models_moduler7 migrated_appsnamerTEST_NON_SERIALIZED_APPS get_models_meta can_migraterallow_migrate_modelr3_default_managerusingorder_bypkiterator)rIloader app_configmodelquerysetrs r get_objectsz@BaseDatabaseCreation.serialize_db_to_string..get_objectsos C$T__5F"224 ; ,,8$$(<(<<OO8+L+LL!+!6!6!8;!KK33DOOD"66t7L7LeT','='='C'C $ 5 5(&hu{{~~':':;%(0'8'8':::;  ;;s0=EEE78E0*EA/E E EjsonN)indentstream)rrr=getvalue)rr\outs` rr9z+BaseDatabaseCreation.serialize_db_to_stringhs3 ;&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. )rTr]N) table_names)rsetrrr3constraint_checks_disabledr deserializesaveaddobject __class__rPdb_tablecheck_constraints)rdatarcobjs rdeserialize_db_from_stringz/BaseDatabaseCreation.deserialize_db_from_strings ~e $//// 0 G;;= I&2264tG\G\]ICHHJOOCJJ$8$8$>$>$G$GHI 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'))rr3)rr( database_names rr/z.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$)rr4TEST_DATABASE_PREFIXrs rr.z&BaseDatabaseCreation._get_test_db_namesO ?? ( ( 0 8??008@ @#doo&C&CF&KKKrc,|jd|zy)Nz%CREATE DATABASE %(dbname)s %(suffix)s)execute)rcursor parametersr>s 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: %srrzTests cancelled.)r.rops quote_namesql_table_creation_suffixrr{ Exceptionrinputr/rxrexit) rr(r<r>r?test_db_paramsryeconfirms rr0z$BaseDatabaseCreation._create_test_dbs"335oo))445GH446      F ,,V^VL :"!5 --  FJK"#=?QRSG'U"2 $$> HH $ > >yJ\ ]&'AN'RS44V^VT$$!PST!TU $HH/0HHQK1  :"!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)rr4rr/_clone_test_db)rr~r(r<r>source_database_namer@s r clone_test_dbz"BaseDatabaseCreation.clone_test_dbsf $<,F/ HH..y:NO  FIv6rcd|jj}i|ddj|d|iS)zX Return a modified connection settings dict for the n-th clone of a DB. r$z{}_{})rr4format)rr~orig_settings_dicts rget_test_db_clone_settingsz/BaseDatabaseCreation.get_test_db_clone_settingss:"__::a$afgnn=OPV=WY_.`aarctd)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)rr~r(r>s rrz#BaseDatabaseCreation._clone_test_dbs" EF FrNc|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#) rr1r4rrr/_destroy_test_dbrr2r3)rold_database_namer(r>r~r?r@s rdestroy_test_dbz$BaseDatabaseCreation.destroy_test_dbs  >!%!>!>v!F !%!@!@!H!P  >!F% HH..y:LM   ! !"4i @  (@QH  t44 5f =4EDOO ) )& 1 )rc|j5}|jd|jjj |zdddy#1swYyxYw)zF Internal implementation - remove the test db tables. zDROP DATABASE %sN)rrxrrr)rr?r(rys rrz%BaseDatabaseCreation._destroy_test_dbsZ    QF NN-!__00;;rs1  #(j j r