U
|
¸ý°d ã @ st d dl mZ d dl mZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd lm
|
Z
|
dd
|
lm Z dd l m Z dd l mZ dd l mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ ddl mZ e d¡dd Ze d¡dd Ze d¡dd Zdd Ze d¡d d Ze d¡d d Ze d¡d!d" Z e d¡d#d$ Ze d¡d%d& Zd'S )(é )Úinspect)ÚIntegeré )Ú create_engine)Úexc)ÚColumn)ÚDropConstraint)ÚForeignKeyConstraint)ÚMetaData)ÚTable)Ú create_db)Ú"drop_all_schema_objects_pre_tables)Údrop_db)Úgenerate_driver_url)Úget_temp_table_name)Úlog)Únormalize_sequence)Ú run_reap_dbs)Útemp_table_keyword_argsZmssqlc C sl | ¡ }| jd||f d}|dkr0|jdd}|r>| |¡}z | ¡ W n tjk
|
rb Y d S X |S d S )Nz%s+%s)Z
|
drivernameZpyodbcÚ )Úquery)Zget_backend_nameÚsetZupdate_query_stringÚ get_dialectr ZNoSuchModuleError)ÚurlZdriverZ query_strÚbackendÚnew_url© r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/dialects/mssql/provision.pyr s
|
r c C sl | ¡ jddR}| d| ¡ | d| ¡ | d| ¡ | d| ¡ | d¡ | d¡ W 5 Q R X d S ) NÚ
|
AUTOCOMMIT©Zisolation_levelzcreate database %sz1ALTER DATABASE %s SET ALLOW_SNAPSHOT_ISOLATION ONz0ALTER DATABASE %s SET READ_COMMITTED_SNAPSHOT ONzuse %szcreate schema test_schemazcreate schema test_schema_2)ÚconnectÚexecution_optionsÚexec_driver_sql©ÚcfgÚengÚidentÚconnr r r Ú_mssql_create_db, s ÿÿ
|
r( c C s* | ¡ jdd}t|| W 5 Q R X d S )Nr r )r r! Ú_mssql_drop_ignorer# r r r Ú_mssql_drop_db; s r* c
|
C s\ z | d| ¡ t d|¡ W dS tjk
|
rV } zt d|¡ W Y ¢dS d }~X Y nX d S )Nzdrop database %sz Reaped db: %sTzcouldn't drop db: %sF)r" r Úinfor Z DatabaseErrorÚwarning)r' r& Úerrr r r r) A s r) c C s¼ t d| ¡ t| }| ¡ jdd}t dd |¡¡ | d¡}dd |D }t }|D ]}||krZ| |¡ qZd }} t |d
|
D ]\} }
|
t
|
||
|
r|d
|
7 }qt d || ¡ W 5 Q R X d S ) Nzdb reaper connecting to %rr r zidentifiers in file: %sz, zselect d.name from sys.databases as d where name like 'TEST_%' and not exists (select session_id from sys.dm_exec_sessions where database_id=d.database_id)c S s h | ]\}| ¡ qS r )Úlower)Ú.0Údbnamer r r Ú <setcomp>` s z"_reap_mssql_dbs.<locals>.<setcomp>r é z-Dropped %d out of %d stale databases detected) r r+ r r r! Újoinr" r ÚaddÚ enumerater) ) r Zidentsr% r' Zto_reapZ all_namesZto_dropÚnameZdroppedÚtotalr0 r r r Ú_reap_mssql_dbsR s* ÿ
|
|
ÿr8 c C s i S )Nr )r$ r% r r r Ú _mssql_temp_table_keyword_argso s r9 c C s d| S )Nz##r )r$ r% Ú base_namer r r Ú_mssql_get_temp_table_namet s r; c C s° | ¡ jdd}t|}d d| j| jfD ]v}|j|dD ]d}t|t tdt tdt |d}t|j
|
||dD ],}| t t |jjg|jjg|d d¡ qpq:q*W 5 Q R X d S ) Nr r Zdbo)ÚschemaÚxÚyr6 )r6 )r r! r Z test_schemaZ test_schema_2Zget_table_namesr r
|
r r Zget_foreign_keysÚexecuter r Úcr= r> )r$ r% r' Z inspectorr< ZtnameÚtbZfkr r r r y s* û ÿÿÿr c C s |j d krd|_ |S )Nr2 )Ústart)r$ Úsequencer r r r s
|
r N) Z
|
sqlalchemyr r r r r r< r r r r
|
r Ztesting.provisionr r r r r r r r r Zfor_dbr( r* r) r8 r9 r; r r r r Ú<module> sD
|
|
|
|
|
|
|
|
