Edit File: strategies.cpython-37.pyc
B ��4]& � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd lm Z dd l m Z i Z G d d � d e�ZG dd� de�ZG dd� de�Ze� G dd� de�Ze� G dd� de�Ze� dS )aq Strategies for creating new instances of Engine types. These are semi-private implementation classes which provide the underlying behavior for the "strategy" keyword argument available on :func:`~sqlalchemy.engine.create_engine`. Current available options are ``plain``, ``threadlocal``, and ``mock``. New strategies can be added via new ``EngineStrategy`` classes. � )� attrgetter� )�base)� threadlocal)�url� )�event)�pool)�util)�schemac @ s e Zd ZdZdd� Zdd� ZdS )�EngineStrategyz�An adaptor that processes input arguments and produces an Engine. Provides a ``create`` method that receives input arguments and produces an instance of base.Engine or a subclass. c C s | t | j< d S )N)� strategies�name)�self� r �O/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/engine/strategies.py�__init__( s zEngineStrategy.__init__c O s t � �dS )z/Given arguments, returns a new Engine instance.N)�NotImplementedError)r �args�kwargsr r r �create+ s zEngineStrategy.createN)�__name__� __module__� __qualname__�__doc__r r r r r r r s r c @ s e Zd ZdZdd� ZdS )�DefaultEngineStrategyz#Base class for built-in strategies.c sF t �|�}|���}|j�dd � ��dd � |�� }|�|����dd�r^d'��fdd� }n�j}i }x&t���D ]}|�krt||�||< qtW ��dd �} | d kr�i } x(t� �j �D ]}|�kr�||�| |< q�W �j f | �} | |d< x|D ]} | � �|� q�W �f |���� |�\� ��� |d i �� t� �� |d d �} | d k�r d(� ��fd d � } |d | �}|dd �}|d k�r~��|�}d�i}dddddddddd� }x6t�|�D ](}|�||�}|�k�r�||�||< �q�W x|D ]} | �||� �q�W ||f|�} n(t| tjj��r | j� ��} n| } �| _| j}i }x*t�|�D ] }|�k�r>||�||< �q>W ��dd�}��r�td d �d d � �D ���jj | jj |j f ��|| �|f|��|�r �� � ���r�fd d!�}t � | d"|� t � | d#|� ��fd$d%�}t j | d"|dd&� �� �� |�k �r(|� �� x|D ]} | � �� �q.W �S ))N�plugin�pluginsZ_coerce_configFc s( �� | |�}| � jkr$� j| |�}|S )N)�popZengine_config_types)�key�default�value)� dialect_clsr r r � pop_kwargB s z/DefaultEngineStrategy.create.<locals>.pop_kwarg�module�dbapiZ connect_argsr c s> �j r2x*�jjD ] }|�| � ��}|d k r|S qW �j� ��S )N)� _has_events�dispatchZ do_connect�connect)�connection_record�fn� connection)�cargs�cparams�dialectr r r( j s z-DefaultEngineStrategy.create.<locals>.connect�creator� poolclassr. Zpool_logging_nameZ echo_poolZ pool_timeoutZ pool_recycleZ pool_eventsZpool_threadlocalZpool_reset_on_returnZ pool_pre_pingZ pool_use_lifo) Z logging_nameZecho�timeoutZrecycle�eventsZuse_threadlocalZreset_on_returnZpre_pingZuse_lifo� _initializeTz�Invalid argument(s) %s sent to create_engine(), using configuration %s/%s/%s. Please check that the keyword arguments are appropriate for this combination of components.�,c s s | ]}d | V qdS )z'%s'Nr )�.0�kr r r � <genexpr>� s z/DefaultEngineStrategy.create.<locals>.<genexpr>c s$ t | d| �}|d krd S � |� d S )NZ _sqla_unwrap)�getattr)�dbapi_connectionr) Zconn)� do_on_connectr r � on_connect� s z0DefaultEngineStrategy.create.<locals>.on_connect� first_connectr( c s4 t j�| dd�}t�� |_� �|� � �|j� d S )NF)r+ r& )r Z Connectionr Z immutabledictZ_execution_options� initializeZ do_rollbackr+ )r9 r) �c)r. �enginer r r< � s z3DefaultEngineStrategy.create.<locals>.first_connect)�once)N)N)!r �make_urlZ_instantiate_plugins�queryr Z_get_entrypointZget_dialect_clsr �get_cls_kwargsZget_func_kwargsr% Zhandle_dialect_kwargsZcreate_connect_args�update�listZget_pool_class�getZhandle_pool_kwargs� isinstance�poollibZ dbapi_proxyZ_DBProxyZget_pool�_dialect� engine_cls� TypeError�join� __class__r r; r �listenZengine_created)r � name_or_urlr �ur Z entrypointr# � dialect_argsr6 r% Z dbapi_argsr r r( r/ r0 Z pool_args� translateZtkZ engineclassZ engine_argsr3 r; r< r )r, r- r. r" r: r? r r r 4 s� z DefaultEngineStrategy.createN)r r r r r r r r r r 1 s r c @ s e Zd ZdZdZejZdS )�PlainEngineStrategyz*Strategy for configuring a regular Engine.�plainN)r r r r r r ZEnginerJ r r r r rS � s rS c @ s e Zd ZdZdZejZdS )�ThreadLocalEngineStrategyz=Strategy for configuring an Engine with threadlocal behavior.r N)r r r r r r ZTLEnginerJ r r r r rU � s rU c @ s. e Zd ZdZdZdd� ZG dd� dej�ZdS )�MockEngineStrategyz�Strategy for configuring an Engine-like object with mocked execution. Produces a single mock Connectable object which dispatches statement execution to a passed-in function. �mockc K sV t �|�}|�� }i }x(t�|�D ]}||kr"|�|�||< q"W |f |�}t�||�S )N)r rA Z get_dialectr rC r rV �MockConnection) r rO �executorr rP r" rQ r6 r. r r r r � s zMockEngineStrategy.createc @ s� e Zd Zdd� Zedd� �Zeed��Zedd� �Ze � d�Z dd � Z d d � Z d d � Zdd� Zdd� Zdd� Zddd�Zdd� ZdS )z!MockEngineStrategy.MockConnectionc C s || _ || _d S )N)rI �execute)r r. rZ r r r r s z*MockEngineStrategy.MockConnection.__init__c C s | S )Nr )�sr r r �<lambda> � z*MockEngineStrategy.MockConnection.<lambda>rI c C s | j jS )N)rI r )r[ r r r r\ r] Nc K s | S )Nr )r r r r r �contextual_connect s z4MockEngineStrategy.MockConnection.contextual_connectc K s | S )Nr )r r r r r r( s z)MockEngineStrategy.MockConnection.connectc K s | S )Nr )r �kwr r r �execution_options s z3MockEngineStrategy.MockConnection.execution_optionsc K s | j j||fd| i|��S )Nr? )rI �compiler)r � statement� parametersr r r r ra s z*MockEngineStrategy.MockConnection.compilerc K s0 d|d< ddl m} |j| j| f|��|� d S )NF� checkfirstr )�ddl)�sqlalchemy.enginere ZSchemaGeneratorr. �traverse_single)r �entityr re r r r r s z(MockEngineStrategy.MockConnection.createc K s0 d|d< ddl m} |j| j| f|��|� d S )NFrd r )re )rf re Z SchemaDropperr. rg )r rh r re r r r �drop& s z&MockEngineStrategy.MockConnection.dropc K s" d|d< || j | f|��|� d S )NFrd )r. rg )r Zvisitorcallable�elementr+ r r r r � _run_visitor. s z.MockEngineStrategy.MockConnection._run_visitorc O s t � �d S )N)r )r Zobject_Z multiparams�paramsr r r rZ 6 s z)MockEngineStrategy.MockConnection.execute)N)r r r r �propertyr? r r. r r Z_schema_getterZschema_for_objectr^ r( r` ra r ri rk rZ r r r r rX s rX N) r r r r r r r Z ConnectablerX r r r r rV � s rV N)r �operatorr � r r r r r rH r Zsqlr r �objectr r rS rU rV r r r r �<module> s$ &P