관리-도구
편집 파일: provision.cpython-38.pyc
U -?�f � @ 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|ksDt�t�d|�}|� }|D ]:}|tkrv|dkrZ|}qZqZ|dkr�qZqZ|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_enc� name_token�tokensZ new_filename�token� r' �T/opt/hc_python/lib64/python3.8/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 U s r c C s t | d |�S r* )r) )r r" r' r' r( � _sqlite_follower_url_from_maina 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. o 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# r0 r' r( �dispose} s z._sqlite_post_configure_engine.<locals>.dispose)Z sqlalchemyr- r! Z listens_for)r r5 Zfollower_identr- r. r6 r' r0 r( � _sqlite_post_configure_enginef s r7 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! r8 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)r1 �listdirr2 r r �infor4 )Z databasenamer! r" r2 �fname�extr' r' r( r< � s r< c C s |� � d S r* )r6 )�configr �clsr' r' r( r � s r c C s ddgiS )N�prefixes� TEMPORARYr' )r9 r: r' r' r( � _sqlite_temp_table_keyword_args� s rG 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 rI F)� set_lambda�sort_by_parameter_orderc C sH ddl m} ||�}|r,|j||j�d�}n|�� }|j|d|i�}|S )Nr )�insert)�set_rK )Zsqlalchemy.dialects.sqliterL Zon_conflict_do_update�excludedZon_conflict_do_nothing� returning)r9 �tablerO rJ rK rL �stmtr' r' r( �_upsert� s ��rR ) r1 r � r r5 r r Ztesting.provisionr r r r r r r r r r r r) Zfor_dbr, r7 r; r= r< rG rI rR r' r' r' r( �<module> sR � 1 ) �