1jw& 6 d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lm
Z
dd l m Z i Z
G d d
e Z G d de Z G d de Z e G d de Z e G d de Z e 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 e Zd ZdZd Zd ZdS )EngineStrategyzAn 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 $ | t | j < d S N)
strategiesname)selfs S/opt/cloudlinux/venv/lib64/python3.11/site-packages/sqlalchemy/engine/strategies.py__init__zEngineStrategy.__init__( s $
49 c t )z/Given arguments, returns a new Engine instance.NotImplementedError)r argskwargss r createzEngineStrategy.create+ s "###r N)__name__
__module__ __qualname____doc__r r r r r r s< % % %$ $ $ $ $r r c e Zd ZdZd ZdS )DefaultEngineStrategyz#Base class for built-in strategies.c t j | }| }|j dd dd | }| | dd rd%fd }nj }i }t j D ]}|v r || ||< dd } | =i }
t j j
D ]}|v r || |
|< j
d&i |
} | |d< |D ]} | | d&i | | \
|di t |d d } | d%fd
}
|d |
} |d d }| | }d
i}dddddddddd }t j | D ]*}| || }|v r || ||< +|D ]} | || ||fi |} n6t% | t& j j r | j i } n| } | _ | j }i }t j | D ]}|v r || ||< dd }rRt3 dd d D d j j d | j j d |j d || |fi | |re r1 fd }t= j | d!| t= j | d"| fd#}t= j | d!|d$ |ur| |D ]} | S )'Npluginplugins_coerce_configFc r | | }| j v r j | | }|S r )popengine_config_types)keydefaultvalue dialect_clsr s r pop_kwargz/DefaultEngineStrategy.create..pop_kwargB s@
300+999 @K ;C @ G GE r moduledbapi connect_argsr
c r j r#j j D ]} || }||c S j i S r ) _has_eventsdispatch
do_connectconnect)connection_recordfn
connectioncargscparamsdialects r r7 z-DefaultEngineStrategy.create..connectj sn & . % . 9 . .%'R #%6w& &
&1#- - - - 2&w9999r creator poolclassr= pool_logging_name echo_pool pool_timeout pool_recycle pool_eventspool_threadlocalpool_reset_on_return
pool_pre_ping
pool_use_lifo) logging_nameechotimeoutrecycleeventsuse_threadlocalreset_on_returnpre_pinguse_lifo _initializeTzInvalid argument(s) ,c 3 K | ] }d |z V
dS )z'%s'Nr! ).0ks r z/DefaultEngineStrategy.create.. s& 8 8AVaZ 8 8 8 8 8 8r z. sent to create_engine(), using configuration /z^. Please check that the keyword arguments are appropriate for this combination of components.c H t | d| }|d S | d S )N _sqla_unwrap)getattr)dbapi_connectionr8 conn
do_on_connects r
on_connectz0DefaultEngineStrategy.create..on_connect s< "(.:J D | !M$'''''r
first_connectr7 c t j | d }t j |_ | |j d S )NF)r: r4 )r
Connectionr
immutabledict_execution_options
initialize do_rollbackr: )r\ r8 cr= engines r r` z3DefaultEngineStrategy.create..first_connect sd O'7U (,'9';';$""1%%%##AL11111r )_once_unless_exceptionr r! )!r make_url_instantiate_pluginsqueryr) _get_entrypointget_dialect_clsr get_cls_kwargsget_func_kwargsr1 handle_dialect_kwargscreate_connect_argsupdatelistget_pool_classgethandle_pool_kwargs
isinstancepoollib dbapi_proxy_DBProxyget_pool_dialect
engine_cls TypeErrorjoin __class__r r_ r listenengine_created) r name_or_urlr ur&
entrypointr/ dialect_argsrV r1
dbapi_argsr% r
r7 r> r? pool_args translatetk engineclass engine_argsrR r_ r` r; r< r= r. r^ rh s ` @@@@@@r r z DefaultEngineStrategy.create4 s] L % %((00 Hd###
9d###&&((
0033 ::& . . #
I $[11 / /AF{{"+)A,, Q
8T** =J)+*;<<
1
1;;$-IaLLJqM%K%33
33E % W D DF ( (l C C C C +-- -- #66q99yy 44555U y&& <
:
:
:
:
:
:
:
: i 733G! +t44I ' 6 6q 9 9 "G,I
!4#))'#5#9 + +
I (33
1
1 ]]1a((<<#,9R==IaL!
@
@)))Y???? 9W22 22DD$ 3 <==
$t}e7w77 #DM o $[11 . .AF{{!*1 A jj 55 ) HH 8 8 8 8 88888%...N+++ (((
T7A==== # . . 0 0M
:( ( ( ( ( T?J??? T9j999
2
2
2
2
2
2
L '+
""6*** [ ( ( % %f - - - * *F ! !& ) ) ) )
r N)r r r r r r! r r r# r# 1 s. --d d d d dr r# c $ e Zd ZdZdZej ZdS )PlainEngineStrategyz*Strategy for configuring a regular Engine.plainN)r r r r r r Enginer~ r! r r r r s 44 DJJJr r c $ e Zd ZdZdZej ZdS )ThreadLocalEngineStrategyz=Strategy for configuring an Engine with threadlocal behavior.r N)r r r r r r TLEnginer~ r! r r r r s! GG D %JJJr r c B e Zd ZdZdZd Z G d dej ZdS )MockEngineStrategyzStrategy 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 t j | }| }i }t j | D ] }||v r| | ||< |di |}t || S )Nr! )r rj get_dialectr ro r) r MockConnection) r r executorr r r. r rV r= s r r zMockEngineStrategy.create s L % %mmoo $[11 0 0AF{{"(**Q-- Q +-- --!00(CCCr c e Zd Zd Z ed Z e ed Z ed Z e j
d Z d Z d Z
d Zd Zd
Zd Z dd Zd
ZdS )!MockEngineStrategy.MockConnectionc " || _ || _ d S r )r} execute)r r= r s r r z*MockEngineStrategy.MockConnection.__init__ s #DM"DLLLr c | S r r! ss r z*MockEngineStrategy.MockConnection. s A r r} c | j j S r )r} r r s r r z*MockEngineStrategy.MockConnection. s
!*/ r Nc | S r r! r r s r contextual_connectz4MockEngineStrategy.MockConnection.contextual_connect Kr c | S r r! r s r r7 z)MockEngineStrategy.MockConnection.connect r r c | S r r! )r kws r execution_optionsz3MockEngineStrategy.MockConnection.execution_options r r c . | j j ||fd| i|S )Nrh )r} compiler)r statement
parametersr s r r z*MockEngineStrategy.MockConnection.compiler s5 )4=): .26<
r c j d|d< ddl m} |j | j | fi | | d S NF
checkfirstr )ddl)sqlalchemy.enginer SchemaGeneratorr= traverse_singler entityr r s r r z(MockEngineStrategy.MockConnection.create# s\ #(F< - - - - - - C d = =f = = M M
r c j d|d< ddl m} |j | j | fi | | d S r )r r
SchemaDropperr= r r s r dropz&MockEngineStrategy.MockConnection.drop+ s\ #(F< - - - - - - C dlD ; ;F ; ; K K
r c T d|d< || j | fi | | d S )NFr )r= r )r visitorcallableelementr: r s r _run_visitorz.MockEngineStrategy.MockConnection._run_visitor3 sI $)F< ODL$ 9 9& 9 9 I I
r c t r r )r object_ multiparamsparamss r r z)MockEngineStrategy.MockConnection.execute; s %'' 'r r )r r r r propertyrh r r= r r _schema_getterschema_for_objectr r7 r r r r r r r! r r r r
s # # # ++&&(::j1122x1122 1F 1$ 7 7
8< ( ( ( ( (r r N) r r r r r r r Connectabler r! r r r r sd DD D D"2( 2( 2( 2( 2() 2( 2( 2( 2( 2(r r N)r operatorr r r r r r
ry r sqlr r objectr r# r r r r! r r r s
$ $ $ $ $V $ $ $"g g g g gN g g gT / & & & & & 5 & & & M( M( M( M( M( M( M( M(` r