관리-도구
편집 파일: provision.cpython-38.pyc
U -?�f9 � @ s� d dl mZ d dlZd dlZddlmZ ddlmZ ddlmZ ddlmZ dd lm Z dd l m Z dd l mZ dd l mZ e�e�ZdZG d d� d�Zdd� Zdd� Zdd� Zdd� Zdd� Zejdd� �Zdd � Z ejd d � �Z ejd d � �Z d!d"� Z d#d$� Z d%d&� Z!ejd'd(� �Z"ejd)d*� �Z#d+d,� Z$e�%e$�d-d.� �Z&ejd/d0� �Z'ejd1d2� �Z(ejd3d4� �Z)ejd5d6� �Z*d7d8� Z+ejd9d:� �Z,ejd;d<� �Z-ejd=d>� �Z.ejd?d@� �Z/ejdAdB� �Z0ejddCdD�dEdF��Z1ejdGdH� �Z2dS )I� )� annotationsN� )�config)�engines)�util� )�exc)�inspect)�url)�ddl��schemac @ s> e Zd Zd dd�Zedd� �Zedd� �Zdd � Zd d � ZdS ) �registerNc C s i | _ || _d S �N)�fns� decorator)�selfr � r �L/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/testing/provision.py�__init__ s zregister.__init__c C s t � �d�|�S �N�*�r �for_db)�cls�fnr r r �init" s z register.initc C s t |��d�S r r )r r r r r �init_decorator& s zregister.init_decoratorc s � �fdd�}|S )Nc s( �j r�� | �} � D ]}| �j|< q�S r )r r )r Zdbname��dbnamesr r r �decorate+ s z!register.for_db.<locals>.decorater )r r r r r r r * s zregister.for_dbc O sr t |t�rt�|�}nt |tj�r(|}n|jj}|�� }|| jkrX| j| |f|�|�S | jd |f|�|�S d S r ) � isinstance�str�sa_url�make_url�URL�dbr �get_backend_namer )r �cfg�arg�kwr �backendr r r �__call__4 s zregister.__call__)N) �__name__� __module__� __qualname__r � classmethodr r r r, r r r r r s r c C s0 t � D ]$}t�d| |jj� t||j| � qd S )NzCREATE database %s, URI %r)�_configs_for_db_operation�log�infor& r � create_db��follower_identr( r r r �create_follower_dbB s r7 c C s� t �| ��� }|�� |r$t| |�} i }t| ||� d|d< t�| |�}t| ||� |� � � � t j � ||||�}|r||t _|r�t||� |S )N�global�scope)r# r$ � get_dialect�load_provisioning�follower_url_from_main�update_db_optsr Ztesting_engine�post_configure_engine�connect�closer �Configr �ident�configure_follower)�db_url�optionsZ file_configr6 �dialect�db_opts�engr( r r r � setup_configH s rI c C s0 t � D ]$}t�d| |jj� t||j| � qd S )NzDROP database %s, URI %r)r1 r2 r3 r& r �drop_dbr5 r r r �drop_follower_dbe s rK c c s� t � }t�t �}dd� dd� | D �D �}|D ] \}}|�� }||j �|� q,i }|D ]l\}}|j} |�� | |kr�t |��|| � || < } n|| } t|| �D ] } | |kr�q�|�| � | V q�qVdS )a& Generate a set of URLs to test given configured URLs plus additional driver names. Given:: --dburi postgresql://db1 --dburi postgresql://db2 --dburi postgresql://db2 --dbdriver=psycopg2 --dbdriver=asyncpg?async_fallback=true Noting that the default postgresql driver is psycopg2, the output would be:: postgresql+psycopg2://db1 postgresql+asyncpg://db1 postgresql+psycopg2://db2 postgresql+psycopg2://db3 That is, for the driver in a --dburi, we want to keep that and use that driver for each URL it's part of . For a driver that is only in --dbdrivers, we want to use it just once for one of the URLs. for a driver that is both coming from --dburi as well as --dbdrivers, we want to keep it in that dburi. Driver specific query options can be specified by added them to the driver name. For example, to enable the async fallback option for asyncpg:: --dburi postgresql://db1 --dbdriver=asyncpg?async_fallback=true c S s g | ]}||� � f�qS r )r: )�.0�url_objr r r � <listcomp>� s �z$generate_db_urls.<locals>.<listcomp>c S s g | ]}t �|��qS r )r# r$ )rL rD r r r rN � s N) �set� collections� defaultdict�get_driver_name�name�addr; � difference�_generate_driver_urls) Zdb_urls� extra_drivers�urlsZ!backend_to_driver_we_already_haveZurls_plus_dialectsrM rF Z driver_nameZbackend_to_driver_we_needr+ Zextra_per_backendZ driver_urlr r r �generate_db_urlsk s0 ! � �� rY c c sx | � � }|�|� t| |d�} | V t|�D ]F}d|krJ|�dd�\}}n|}d }t| ||�}|r,|�|� |V q,d S )N� �?r )rR �discard�generate_driver_url�list�split�remove)r rW Z main_driver�drvZ driver_only� query_str�new_urlr r r rV � s rV c C sX | � � }| jd||f d�}|r*|�|�}z |�� W n tjk rN Y d S X |S d S )Nz%s+%s)Z drivername)r' rO Zupdate_query_stringr: r ZNoSuchModuleError)r Zdriverrb r+ rc r r r r] � s � r] c c s� t � } tj�� D ]}|j�� qtj�� D ]>}|jj}|�� }||j|j |j f}|| kr*|V | � |� q*tj�� D ]}|j�� qtd S r ) rO r rA Z all_configsr& Zdisposer r' �username�host�databaserT )�hostsr( r r+ Z host_confr r r r1 � s r1 c C s d S r r �r( rH r r r �"drop_all_schema_objects_pre_tables� s ri c C s d S r r rh r r r �#drop_all_schema_objects_post_tables� s rj c C s t | |� t| |� tjjjr(t| |� t|�}d}tjj� | �rR|| j | j f7 }t j |||d� t| |� tjj� | �r�|�� �v}|�� D ]}|�t�t�|��� q�tjj� | �r�| j | j fD ]0}|j|d�D ] }|�t�tj||d��� q�q�W 5 Q R X d S )Nr )�consider_schemasr )ri � drop_viewsr � requirementsZmaterialized_views�enabled�drop_materialized_viewsr �schemas�enabled_for_config� test_schemaZ test_schema_2r Zdrop_all_tablesrj � sequences�beginZget_sequence_names�executer Z DropSequencer �Sequence)r( rH � inspectorrk �conn�seq� schema_namer r r �drop_all_schema_objects� s0 � ��r{ c C s� t |�}z |�� }W n tk r( Y n<X |�� �,}|D ] }|�t�t�|t� � ��� q8W 5 Q R X t j j � | �r�z|j| jd�}W n tk r� Y nBX |�� �2}|D ]&}|�t�tj|t� � | jd��� q�W 5 Q R X d S )Nr )r Zget_view_names�NotImplementedErrorrt ru r Z _DropViewr �TableZMetaDatar rm rp rq rr )r( rH rw Z view_namesrx �vnamer r r rl s4 � ���rl c C s� t |�}|�� }|�� � }|D ]}|�d|� �� q W 5 Q R X tjj�| �r�|j| jd�}|�� �(}|D ] }|�d| j� d|� �� qhW 5 Q R X d S )NzDROP MATERIALIZED VIEW r �.) r Zget_materialized_view_namesrt Zexec_driver_sqlr rm rp rq rr )r( rH rw Z mview_namesrx r~ r r r ro 6 s � �ro c C s t d|jf ��dS )z�Dynamically create a database for testing. Used when a test run will employ multiple processes, e.g., when run via `tox` or `pytest -n4`. z"no DB creation routine for cfg: %sN�r| r �r( rH rB r r r r4 J s �r4 c C s t d|jf ��dS )z8Drop a database that we dynamically created for testing.z no DB drop routine for cfg: %sNr� r� r r r rJ V s rJ c s, t �� �}t|j�dkr � S � fdd�S d S )N� c s � | |�S r r )rD rG �_options�r r r �<lambda>a � z'_adapt_update_db_opts.<locals>.<lambda>)r Zinspect_getfullargspec�len�args)r Zinspr r� r �_adapt_update_db_opts\ s r� c C s dS )zCSet database options (db_opts) for a test database that we created.Nr )rD rG rE r r r r= d s r= c C s dS )z�Perform extra steps after configuring an engine for testing. (For the internal dialects, currently only used by sqlite, oracle) Nr )r �enginer6 r r r r> i s r> c C s t �| �} | j|d�S )z�Create a connection URL for a dynamically-created test database. :param url: the connection URL specified when the test run was invoked :param ident: the pytest-xdist "worker identifier" to be used as the database name )rf )r# r$ rO �r rB r r r r< q s r< c C s dS )z@Create dialect-specific config settings for a follower database.Nr )r( rB r r r rC } s rC c C s dS )a[ Remove databases that were created during the test process, after the process has ended. This is an optional step that is invoked for certain backends that do not reliably release locks on the database as long as a process is still in use. For the internal dialects, this is currently only necessary for mssql and oracle. Nr r� r r r � run_reap_dbs� s r� c C s� t �d� t�t�}t�t�}i }t| ��z}|D ]n}|�� }|�d�\}}t� |�}||krt|� � ||< || � � |� � |j f} || �|� || �|� q0W 5 Q R X |D ]&} t|| �d } || } t| | � q�d S )NzReaping databases...� r )r2 r3 rP rQ rO �open�stripr_ r# r$ r: r; r' re rT r^ r� ) Z idents_filerX ZidentsZdialects�file_�lineZdb_namerD rM Zurl_keyr rB r r r �reap_dbs� s& r� c C s t d|jf ��dS )aO Specify keyword arguments for creating a temporary Table. Dialect-specific implementations of this method will return the kwargs that are passed to the Table method when creating a temporary table for testing, e.g., in the define_temp_tables method of the ComponentReflectionTest class in suite/test_reflection.py z.no temp table keyword args routine for cfg: %sNr� rh r r r �temp_table_keyword_args� s �r� c C s d S r r )r � connectionr r r �prepare_for_drop_tables� s r� c C s d S r r )r r& Ztestclsr r r � stop_test_class_outside_fixtures� s r� c C s |S )a� Specify table name for creating a temporary Table. Dialect-specific implementations of this method will return the name to use when creating a temporary table for testing, e.g., in the define_temp_tables method of the ComponentReflectionTest class in suite/test_reflection.py Default to just the base name since that's what most dialects will use. The mssql dialect's implementation will need a "#" prepended. r )r( rH � base_namer r r �get_temp_table_name� s r� c C s t d| jjf ��d S )Nz9backend does not implement a schema name set function: %s�r| r& r )r( Zdbapi_connectionrz r r r � set_default_schema_on_connection� s ��r� F)� set_lambda�sort_by_parameter_orderc C s t d| jj� ���dS )a� return the backends insert..on conflict / on dupe etc. construct. while we should add a backend-neutral upsert construct as well, such as insert().upsert(), it's important that we continue to test the backend-specific insert() constructs since if we do implement insert().upsert(), that would be using a different codepath for the things we need to test like insertmanyvalues, etc. z3backend does not include an upsert implementation: Nr� )r( �tableZ returningr� r� r r r �upsert� s �r� c C s |S )z�Normalize sequence parameters for dialect that don't start with 1 by default. The default implementation does nothing r )r( �sequencer r r �normalize_sequence� s r� )3� __future__r rP �loggingrZ r r r r r r� r r# �sqlr r � getLoggerr- r2 ZFOLLOWER_IDENTr r7 rI rK rY rV r r] r1 ri rj r{ rl ro r4 rJ r� r r= r> r< rC r� r� r� r� r� r� r� r� r� r r r r �<module> sx % D !! �