U
|
¸ý°d. ã @ sf d dl Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd lm
|
Z
|
dd
|
lm Z dd lm Z dd lm Z dd lmZ ddlmZ ddlmZ ddlmZ e d¡dd Ze d¡dd Ze d¡dd Ze d¡dd Ze
|
d¡dd Z
|
e d¡d d Z e d¡d d Ze d¡dd d!d"d#Zd$d%gZe d¡d&d' ZdS )(é Né )Úexc)Úinspect)Útext)Úwarn_test_suite)Ú create_db)Ú#drop_all_schema_objects_post_tables)Ú"drop_all_schema_objects_pre_tables)Údrop_db)Úlog)Úpost_configure_engine)Úprepare_for_drop_tables)Ú set_default_schema_on_connection)Útemp_table_keyword_args)ÚupsertÚ
|
postgresqlc C sÄ | j j}|jdd ¡ ¢}|s,| d¡ ¡ }d}z| d||f ¡ W q¶ tjk
|
r¤ } z>|d7 }|dkrl dt|krt
|
d ||j |¡ t d
|
¡ W 5 d }~X Y q0 Y q0X q¶q0W 5 Q R X d S ) NÚ
|
AUTOCOMMIT©Zisolation_levelzselect current_database()r z CREATE DATABASE %s TEMPLATE %sé r zaccessed by other userszFWaiting to create %s, URI %r, template DB %s is in use sleeping for .5g à?)ÚoptionsZpostgresql_templatedbÚexecution_optionsÚbeginÚexec_driver_sqlZscalarr ZOperationalErrorÚstrr ÚinfoÚurlÚtimeÚsleep)ÚcfgÚengÚidentZ template_dbÚconnÚattemptÚerr© r$ ú_d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/dialects/postgresql/provision.pyÚ _pg_create_db s4 ÿ
|
ÿ ûr& c
|
C sX | ¡ jdd>}| ¡ * | tdt|d¡ | d| ¡ W 5 Q R X W 5 Q R X d S )Nr r zselect pg_terminate_backend(pid) from pg_stat_activity where usename=current_user and pid != pg_backend_pid() and datname=:dname)ZdnamezDROP DATABASE %s)Úconnectr r Úexecuter Údictr )r r r r! r$ r$ r% Ú _pg_drop_db8 s
|
ÿúr* c C s
|
ddgiS )NÚprefixesÚ TEMPORARYr$ )r r r$ r$ r% Ú#_postgresql_temp_table_keyword_argsG s r- c C s4 |j }d|_ | ¡ }| d| ¡ | ¡ ||_ d S )NTz SET SESSION search_path='%s')Z
|
autocommitÚcursorr( Úclose)r Zdbapi_connectionZ schema_nameZexisting_autocommitr. r$ r$ r% Ú,_postgresql_set_default_schema_on_connectionL s r0 c C sB | ¡ jdd(}| d¡ ¡ D ]}| d| ¡ q W 5 Q R X d S )Nr r z!select gid from pg_prepared_xactszROLLBACK PREPARED '%s')r' r r Zscalarsr( )r r r! Úxidr$ r$ r% r X s
|
ÿ r c
|
C s^ ddl m} t|}| ¡ 8}| d¡D ]&}| | |j|d |d d¡¡ q(W 5 Q R X d S )Nr )r Ú*ÚnameÚschema)r3 r4 )Zsqlalchemy.dialectsr r r Z get_enumsr( Z DropEnumTypeZENUM)r r r Z inspectorr! Úenumr$ r$ r% r a s
|
ÿÿr c C s6 | d¡}| ¡ }|r2tdd dd |D ¡ dS )z;Ensure there are no locks on the current username/database.zµselect pid, state, wait_event_type, query from pg_stat_activity where usename=current_user and datname=current_database() and state='idle in transaction' and pid != pg_backend_pid()zHPostgreSQL may not be able to DROP tables due to idle in transaction: %sz; c s s | ]}|j d V qdS )ÚqueryN)Ú_mapping)Ú.0Úrowr$ r$ r% Ú <genexpr> s z*prepare_for_drop_tables.<locals>.<genexpr>N)r Úallr Újoin)ÚconfigÚ
|
connectionÚresultÚrowsr$ r$ r% r o s ÿþÿr F)Ú
|
set_lambdaÚsort_by_parameter_orderc C sV ddl m} ||}t|j}|r:|j|j||jd}n| ¡ }|j|d|i}|S )Nr )Úinsert)Zindex_elementsÚset_rB ) Z sqlalchemy.dialects.postgresqlrC r Z
|
selectableZon_conflict_do_updateZ primary_keyZexcludedZon_conflict_do_nothingÚ returning)r ÚtablerE rA rB rC ZstmtZtable_pkr$ r$ r% Ú_upsert s
|
ÿÿÿrG )Zcitext©é )ZhstorerH c C sN | ¡ <}tD ]0\}}|jj|kr| td| ¡ | ¡ qW 5 Q R X d S )Nz CREATE EXTENSION IF NOT EXISTS )r' Ú _extensionsÚdialectZserver_version_infor( r Úcommit)r ZengineZfollower_identr! Ú extensionÚ min_versionr$ r$ r% Ú_create_citext_extension¡ s
|
ÿrO )r Ú r r r Ztestingr Ztesting.provisionr r r r
|
r r r r r r Zfor_dbr& r* r- r0 rG rJ rO r$ r$ r$ r% Ú<module> sJ
|
"
|
|
|
|
|
|
ÿþ
|
