+h&/ ddlZddlZddlmZddlmZddlmZddlmZddlmZdd lm Z dd l m Z dd l m Z dd lmZejeZdZGd deZdZdZdZdZdZej4dZdZej4dZej4dZdZej4dZ ej4dZ!ej4dZ"ej4dZ#ej4dZ$ej4dZ%ej4dZ&d Z'ej4d!Z(ej4d"Z)ej4d#Z*ej4d$Z+ej4d%Z,y)&N)config)engines)util)exc)inspect)url)ddlschema)compatc.eZdZdZedZdZdZy)registerci|_yNfns)selfs O/var/www/html/venv/lib/python3.12/site-packages/sqlalchemy/testing/provision.py__init__zregister.__init__s c@tjd|SN*)rfor_db)clsfns rinitz register.inits%xz  %b))rcfd}|S)Nc4D]}|j|<Srr)rdbnamedbnamesrs rdecoratez!register.for_db..decorates%! &#%  &Kr)rr#r$s`` rrzregister.for_dbs  rc\t|tjrtj|}n3t|tj r|}n|j j}|j}||jvr|j||g|S|jd|g|Sr) isinstancer string_typessa_urlmake_urlURLdbr get_backend_namer)rcfgargr backends r__call__zregister.__call__$s c6.. ///#&C VZZ (C&&**C&&( dhh $488G$S/3/ / 488C=+s+ +rN)__name__ __module__ __qualname__r classmethodrrr1r%rrrrs%** ,rrctD]D}tjd||jjt ||j|Fy)NzCREATE database %s, URI %r)_configs_for_db_operationloginfor,r create_dbfollower_identr.s rcreate_follower_dbr=2s=(*/ -~svvzzJ#svv~./rctj|j}|j|r t ||}i}t ||d|d<t j||}t||||jjtjj||||}|r |t_|r t|||S)Nglobalscope)r)r* get_dialectload_provisioningfollower_url_from_mainupdate_db_optsrtesting_enginepost_configure_engineconnectcloserConfigridentconfigure_follower)db_urloptions file_configr<dialectdb_optsengr.s r setup_configrR8soof%113G '?G67#GG   1C&#~6KKM -- gw DC% 3/ JrctD]D}tjd||jjt ||j|Fy)NzDROP database %s, URI %r)r7r8r9r,r drop_dbr;s rdrop_follower_dbrUTs=(*- +^SVVZZHSVV^,-rc#HKt}tjt}|Dcgc]}tj|c}Dcgc]}||j f}}|D]-\}}||j j|j/i}|D]x\}}|j } |j| |vr#t|j|| x|| <} n|| } t|| D]} | |vr|j| | zycc}wcc}ww)a&Generate a set of URLs to test given configured URLs plus additional driver names. Given:: --dburi postgresql://db1 --dburi postgresql://db2 --dburi postgresql://db2 --dbdriver=psycopg2 --dbdriver=asyncpg?async_fallback=true Noting that the default postgresql driver is psycopg2, the output would be:: postgresql+psycopg2://db1 postgresql+asyncpg://db1 postgresql+psycopg2://db2 postgresql+psycopg2://db3 That is, for the driver in a --dburi, we want to keep that and use that driver for each URL it's part of . For a driver that is only in --dbdrivers, we want to use it just once for one of the URLs. for a driver that is both coming from --dburi as well as --dbdrivers, we want to keep it in that dburi. Driver specific query options can be specified by added them to the driver name. For example, to enable the async fallback option for asyncpg:: --dburi postgresql://db1 --dbdriver=asyncpg?async_fallback=true N) set collections defaultdictr)r*rAnameadddriverrB difference_generate_driver_urls) db_urls extra_driversurls!backend_to_driver_we_already_haverLurl_objurls_plus_dialectsrObackend_to_driver_we_needr0extra_per_backend driver_urls rgenerate_db_urlsrhZsJB 5D(3(?(?(D%?FFF/F  '%%'( /L)',,7;;GNNKL!#.,,!!# 3 3EHFj:7CD E %g .1B!:' B /9JK JT! HHZ    Gs(D"DD" D$B>D"c#:K|j}|j|t||d}t|t |D]N}d|vr|j dd\}}n|}d}t|||}|s1|j |t|Pyw)N?r)get_driver_namediscardgenerate_driver_urlstrlistsplitremove)r r` main_driverdrv driver_only query_strnew_urls rr^r^s%%'K+& c; 3C c(NM" #:%(YYsA%6 "KKI%c; B    %g, s A7B:!Bc|j}|j|d|}|r|j|} |j|S#tj $rYywxYw)N+) drivername)r-rWupdate_query_stringrArNoSuchModuleError)r r\rvr0rws rrnrnsr""$Ggg%v.G--i8  sAA$#A$c#"Kt}tjjD]}|jj tjjD]f}|jj }|j}||j|j|jf}||vsR||j|htjjD]}|jj ywr) rWrrI all_configsr,disposer r-usernamehostdatabaser[)hostsr.r r0 host_confs rr7r7s EE}}((* }}((*!ffjj&&(cllCHHcllC E !I IIi !}}((* s B7D:ADcyrr%r.rQs r"drop_all_schema_objects_pre_tablesrrcyrr%rs r#drop_all_schema_objects_post_tablesrrrc pt||t|} |j}|j5}|D]L}|j t j tj|tjN dddtjjj|r |jd}|j5}|D]N}|j t j tj|tjdP dddtj ||tjjj|rDtj |||j"tj |||j$t'||tjj(j|r|j5}|j+D]9}|j t j,tj.|;tjjj|rm|j"|j$fD]R}|j+|D];}|j t j,tj.||=Tdddyy#1swYhxYw#t$rYwwxYw#1swYxYw#t$rYwxYw#1swYyxYw)N test_schemar )rr get_view_namesbeginexecuter _DropViewr TableMetaDataNotImplementedErrorr requirementsschemasenabled_for_configrdrop_all_tablesr test_schema_2r sequencesget_sequence_names DropSequenceSequence)r.rQ inspector view_namesconnvnameseq schema_names rdrop_all_schema_objectsrs&sC0 I --/ YY[ D#  MM&,,ufoo6G"HI   ""55c: "111GJ ' ELL "LL % & 1'4   i( ""55c: S)COOD S)C4E4EF'S1 $$77< YY[ D 335 E S--fooc.BCD E""**==cB$'OOS5F5F#GK(;;* <  ,, &K H  =?        #   .  sIK?AK2>L ALC#L,2K<? L  L L L)(L),L5c2td|j)zDynamically create a database for testing. Used when a test run will employ multiple processes, e.g., when run via `tox` or `pytest -n4`. z no DB creation routine for cfg: rr r.rQrJs rr:r:s 039 rc2td|j)z8Drop a database that we dynamically created for testing.zno DB drop routine for cfg: rrs rrTrT's #''K LLrcy)zCSet database options (db_opts) for a test database that we created.Nr%)rLrPs rrDrD- rcy)zPerform extra steps after configuring an engine for testing. (For the internal dialects, currently only used by sqlite, oracle) Nr%)r enginer<s rrFrF3s  rcPtj|}|j|S)zCreate a connection URL for a dynamically-created test database. :param url: the connection URL specified when the test run was invoked :param ident: the pytest-xdist "worker identifier" to be used as the database name )r)r)r*rWr rJs rrCrC<s# //# C 77E7 ""rcy)z@Create dialect-specific config settings for a follower database.Nr%)r.rJs rrKrKHrrcy)a[Remove databases that were created during the test process, after the process has ended. This is an optional step that is invoked for certain backends that do not reliably release locks on the database as long as a process is still in use. For the internal dialects, this is currently only necessary for mssql and oracle. Nr%rs r run_reap_dbsrNs rctjdtjt}tjt}i}t |5}|D]}|j }|jd\}}tj|}||vr&|j||<||j|j|jf} || j||| j| ddd|D]$} t|| d} || } t!| | &y#1swY3xYw)NzReaping databases... r)r8r9rXrYrWopenstriprqr)r*rArBr-rr[rpr) idents_fileraidentsdialectsfile_linedb_namerLrcurl_keyr rJs rreap_dbsr[s*HH #$  " "3 'D  $ $S )FH k  )e )D::rs"g!,v,</ 8- >B0 "&      6r  MM        ##        !2             r