U
|
¸ý°dÇ ã @ sl d dl Z d dlZddlmZ ddlmZ ddlmZ ddlm Z ddlm
|
Z
|
ddlm Z dd lm Z dd
|
lm Z dd lmZ dd lmZ dd lmZ ddlmZ dddddhZdd Ze d¡dd Z e
|
d¡dd Ze d¡dd Ze d¡d d Ze d¡d d Zd!d" Ze d¡d#d$ Ze d¡d%d& Ze d¡d'd( Ze d¡dd)d*d+d,Z dS )-é Né )Úexc)Úurl)Ú create_db)Údrop_db)Úfollower_url_from_main)Úgenerate_driver_url)Úlog)Úpost_configure_engine)Ú run_reap_dbs)Ú stop_test_class_outside_fixtures)Útemp_table_keyword_args)ÚupsertZpysqliteZ aiosqliteÚ pysqlcipherZpysqlite_numericZpysqlite_dollarc C s t | ¡} |dkr| ¡ }| j}|dk}d}|rê|dkrêd|ksDtt d|¡}| }|D ]:}|tkrv|dkrZ|}qZqZ|dkrqZqZ|dkrZ| d¡}qZ|sªtd| j | d| }|rÊ|d| 7 }|d 7 }|rÞ|d
|
7 }| j |d } |rú| j d d } | j d|f d} | S )zUgiven a sqlite url + desired driver + ident, make a canonical
|
URL out of it
|
|
Nr ú:memory:Z test_schemaz[_\.])ÚdbÚencÚ_z#sqlite filename has no name token: ú.dbz.enc)ÚdatabaseÚtest)Úpasswordz sqlite+%s)Ú
|
drivername)
|
Úsa_urlÚmake_urlZget_driver_namer ÚAssertionErrorÚreÚsplitÚ _drivernamesÚstripÚset) r ÚdriverÚidentÚfilenameZ needs_encZ
|
name_tokenÚtokensZ new_filenameÚtoken© r& ú[d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/dialects/sqlite/provision.pyÚ _format_url s>
|
r( Zsqlitec C s: t | |d } z | ¡ W n tjk
|
r0 Y d S X | S d S ©N)r( Ú get_dialectr ZNoSuchModuleError)r r! Z query_strr& r& r' r O s r c C s t | d |S r) )r( )r r" r& r& r' Ú _sqlite_follower_url_from_main[ s r+ c sd ddl m} |r$| d|j d n |j d | |d¡ fdd}| |d¡ fd d
|
}d S ) Nr )Úeventr z_test_schema.dbÚconnectc s | d d¡ d S )NzATTACH DATABASE "z" AS test_schema)Úexecute)Zdbapi_connectionZconnection_record©Z attach_pathr& r' r- i s
|
|
ÿz._sqlite_post_configure_engine.<locals>.connectZengine_disposedc sD t j ¡rt ¡ | jj}|r@|dkr@t j |¡r@t |¡ dS )z¼most databases should be dropped using
|
stop_test_class_outside_fixtures
|
|
however a few tests like AttachedDBTest might not get triggered on
|
that main hook
|
|
r N)ÚosÚpathÚexistsÚremover r )Úenginer# r/ r& r' Údisposew s
|
|
|
z._sqlite_post_configure_engine.<locals>.dispose)Z
|
sqlalchemyr, r! Z listens_for)r r4 Zfollower_identr, r- r5 r& r/ r' Ú _sqlite_post_configure_engine` s
|
|
r6 c C s d S r) r& ©ÚcfgÚengr" r& r& r' Ú_sqlite_create_db s r: c C s t |jj|j| d S r) )Ú_drop_dbs_w_identr r r! r7 r& r& r' Ú_sqlite_drop_db s r< c C sJ t d¡D ]:}t j |¡\}}||kr
|
|dkr
|
t d|¡ t |¡ q
|
d S )NÚ.)r z.db.encz!deleting SQLite database file: %s)r0 Úlistdirr1 r r Úinfor3 )Z databasenamer! r" r1 ÚfnameÚextr& r& r' r; s
|
r; c C s | ¡ d S r) )r5 )Úconfigr Úclsr& r& r' r s r c C s
|
ddgiS )NÚprefixesÚ TEMPORARYr& )r8 r9 r& r& r' Ú _sqlite_temp_table_keyword_args¡ s rF c C sN t d| ¡ t dd |¡¡ t | ¡} |D ] }tD ]}t| j|| q4q,d S )Nzdb reaper connecting to %rzidentifiers in file: %sz, )r r? Újoinr r r r; r )r Zidentsr" r r& r& r' Ú_reap_sqlite_dbs¦ s
|
rH F)Ú
|
set_lambdaÚsort_by_parameter_orderc C sH ddl m} ||}|r,|j||jd}n| ¡ }|j|d|i}|S )Nr )Úinsert)Úset_rJ )Zsqlalchemy.dialects.sqliterK Zon_conflict_do_updateZexcludedZon_conflict_do_nothingÚ returning)r8 ÚtablerM rI rJ rK Zstmtr& r& r' Ú_upsert° s ÿÿrO ) r0 r Ú r r4 r r Ztesting.provisionr r r r r r
|
r r r r r r( Zfor_dbr+ r6 r: r< r; rF rH rO r& r& r& r' Ú<module> sR û 1
|
|
|
)
|
|
|
|
|
ÿ
|
