U
|
¸ý°dÿ ã @ sÖ d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ e dd ¡d
|
d Ze dd ¡d d Z
|
e dd ¡dd Z e dd ¡dd Z e dd ¡dd Z e d ¡dddddZdS )é )Úexc)Úconfigure_follower)Ú create_db)Údrop_db)Úgenerate_driver_url)Útemp_table_keyword_args)ÚupsertÚmysqlÚmariadbc C sn | ¡ }|dkr&| ¡ }| | ¡r&d}| jd||f d |¡}z | ¡ W n tjk
|
rd Y d S X |S d S )Nr r
|
z%s+%s)Z
|
drivername)Zget_backend_nameÚ get_dialectZ_is_mariadb_from_urlÚsetZupdate_query_stringr ZNoSuchModuleError)ÚurlZdriverZ query_strÚbackendZ dialect_clsÚnew_url© r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/dialects/mysql/provision.pyr s
|
|
ÿþ r c C s| | ¡ ,}zt| || W n tk
|
r. Y nX W 5 Q R X | ¡ 0}| d| ¡ | d| ¡ | d| ¡ W 5 Q R X d S )Nz(CREATE DATABASE %s CHARACTER SET utf8mb4z4CREATE DATABASE %s_test_schema CHARACTER SET utf8mb4z6CREATE DATABASE %s_test_schema_2 CHARACTER SET utf8mb4)ÚbeginÚ_mysql_drop_dbÚ ExceptionÚexec_driver_sql©ÚcfgÚengÚidentÚconnr r r Ú_mysql_create_db- s
|
|
ÿÿÿr c C s d| | _ d| | _d S )Nz%s_test_schemaz%s_test_schema_2)Z test_schemaZ test_schema_2)Úconfigr r r r Ú_mysql_configure_followerA s
|
r c C sB | ¡ 0}| d| ¡ | d| ¡ | d| ¡ W 5 Q R X d S )Nz DROP DATABASE %s_test_schemaz DROP DATABASE %s_test_schema_2zDROP DATABASE %s)r r r r r r r G s
|
r c C s
|
ddgiS )NÚprefixesÚ TEMPORARYr )r r r r r Ú _mysql_temp_table_keyword_argsO s r NF)Ú
|
set_lambdaÚsort_by_parameter_orderc C s\ ddl m} ||}|r,|jf ||j}n |jjd }| |j|i¡}|j|d|i}|S )Né )Úinsertr" )Zsqlalchemy.dialects.mysqlr$ Zon_duplicate_key_updateZinsertedZ primary_keyÚcÚkeyÚ returning)r Útabler' r! r" r$ ZstmtZpk1r r r Ú_upsertT s ÿÿr) )Ú r Ztesting.provisionr r r r r r Zfor_dbr r r r r) r r r r Ú<module> s(
|
|
|
|
|
|
|
|
|
|
ÿ
|
