ih  UdZddlmZddlZddlZddlmZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlmZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddlm#Z#ddl$Z$ddl%m&Z&ddl%m'Z'ddl(m)Z)ddlm*Z*ddl%m+Z+e"d Z,e"d!d"#Z-e"d$ed%ef&Z.e"d'd(&Z/e"d)d*&Z0e"d+d,&Z1e"d-d.&Z2e'j3r/dd/l4m5Z5dd0l4m6Z6d1Z7e8j9d2j:Z;d3Z<dd:Z=d d;Z>ne'j?rd d<Z>nd d=Z>d!d@Z@GdAdBZAd"dFZBd#dHZC d$d%dLZDd&dRZEd'dVZFd(d[ZGd)daZHe"dbdP&ZId*deZJd+dkZKe"dlZLe"dmZMGdndoZNdpZOedd%dqd,dxZPeddydqd-dzZPddydqd-d{ZPd.d~ZQ d/d0dZR d1d2dZSd1dZT d3d4dZUdZVdZW d5d6dZXGddZYdZZdZ[ d7dZ\dZ]d8dZ^e"dd&Z_Gddee-Z`Gdde`e-ZaGdde`e-Zbe!r e`Zce`ZdeeZfeeZgnebxZcZfeaxZdZgd9dZhGddZie!reeZjneijkZjGddZld:dZmd;dZnddZpd?dZqd@dZrdAd„Zs d$dBdƄZtdCd˄Zud̄ZvGd̈́deeZwGdτdee,ZxGdфdee,ZyGdӄdee,ZzGdՄde{Z|Gdׄde8Z}Gdلde}ۦZ~e!r ddlmZeZne~Ze"dej&Z dDdEdZdadFdZdGdZdHdZGddeZd$dIdZdJdZiZded<dKdZe jdZ dLdMdZdNdZe jdZe jdZeefdOdZe8dZdZe jdZGdd ZGd d Zd Zd ZdPdZdQdZdRdZe jdZdSdZdTdZdDdZGddejZejZe#e,e*ejfZdS(UzRoutines to help with the creation, loading and introspection of modules, classes, hierarchies, attributes, functions, and methods. ) annotationsNupdate_wrapper)CodeType)Any)Callable)cast)Dict) FrozenSet)Generic)Iterator)List)Mapping)NoReturn)Optional)overload)Sequence)Set)Tuple)Type) TYPE_CHECKING)TypeVar)Union) _collections)compat)HAS_CYEXTENSION)Literal)exc_T_T_coT) covariant_F.)bound_MPzmemoized_property[Any]_MAz#HasMemoized.memoized_attribute[Any]_HPzhybridproperty[Any]_HMzhybridmethod[Any])call_annotate_function)Formatct|dd}|;t|||}t|tst |d|SdS)N __annotate__)ownerz!.__annotate__ returned a non-dict)getattrr* isinstancedict ValueError)objformatannotateanns c/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/sqlalchemy/util/langhelpers.py_get_and_call_annotater8Fsa355  (6EEECc4(( N C!L!L!LMMMJt__annotations__ct|tr# t|}n'#t$ricYSwxYwt |dd}|iSt|t st |dt |S)Nr:z+.__annotations__ is neither a dict nor None)r0type_BASE_GET_ANNOTATIONSAttributeErrorr/r1r2r3r6s r7_get_dunder_annotationsr@Rs c4  +C00!     #0$77C{ #t$$ EEE Cyys ' 66r3rr4r+returnMapping[str, Any]cB t|}|t|Sn#t$rYnwxYwt||}|t|}|8t |t st |riSt|d|siSt|S)z:A sparse implementation of annotationlib.get_annotations()Nz does not have annotations)r@r1 Exceptionr8r0r<callable TypeError)r3r4r6s r7_vendored_get_annotationsrGds  !)#..CCyy     D %S&11 ; *#..C ;#t$$    s@@@AA A ICyys # 00c8t|tjS)N)r4)rGr+ FORWARDREFr3s r7get_annotationsrKs )V5FGGGGr9c*tj|SN)inspectrKrJs r7rKrKs&s+++r9ct|tr|jdd}nt |dd}| t jStd|S)Nr:rB)r0r<__dict__getr/r EMPTY_DICTr r?s r7rKrKsa c4  8,""#4d;;CC#0$77C ;* *+S11 1r9xstrc|d}tj}||t t |S)Nzutf-8)encodermd5_not_for_securityupdater rT hexdigest)rSms r7md5_hexr[sH A#%%AHHQKKK Q[[]] # ##r9c2eZdZUdZdZded<ddZddZdS) safe_reraiseaReraise an exception after invoking some handler code. Stores the existing exception info before invoking so that it is maintained across a potential coroutine context switch. e.g.:: try: sess.commit() except: with safe_reraise(): sess.rollback() TODO: we should at some point evaluate current behaviors in this regard based on current greenlet, gevent/eventlet implementations in Python 3, and also see the degree to which our own asyncio (based on greenlet also) is impacted by this. .rollback() will cause IO / context switch to occur in all these scenarios; what happens to the exception context from an "except:" block if we don't explicitly store it? Original issue was #2703. ) _exc_infozdUnion[None, Tuple[Type[BaseException], BaseException, types.TracebackType], Tuple[None, None, None]]r^rANonec6tj|_dSrM)sysexc_infor^selfs r7 __enter__zsafe_reraise.__enter__sr9type_Optional[Type[BaseException]]valueOptional[BaseException] tracebackOptional[types.TracebackType]rc|jJ|+|j\}}}|Jd|_||d|_|J||rM)r^with_traceback)rdrfrhrjexc_type exc_valueexc_tbs r7__exit__zsafe_reraise.__exit__ss ~))) =*.. 'Hi(((!DN**622 2!DN$$$&&y11 1r9NrAr_)rfrgrhrirjrkrAr)__name__ __module__ __qualname____doc__ __slots__r:rerqr9r7r]r]sc0I((((222222r9r]clsType[_T]Iterator[Type[_T]]c#Kt}|g}|r]|}||vr|||||V|[dSdSrM)setpopaddextend__subclasses__)ryseenstacks r7walk_subclassesrsUUD EE iikk $;;  HHSMMM S''))***  r9elementczt|tr|S t|S#t$rd|zcYSwxYw)Nzunprintable element %r)r0rTrD)rs r7string_or_unprintablersV'36 6w<<  6 6 6+g5 5 5 5 6s (:: Type[Any]use_name Optional[str]cx|p|j}ddtjd|DS)N c3>K|]}|VdSrM)lower).0ns r7 z(clsname_as_plain_name..s*NN!AGGIINNNNNNr9z([A-Z][a-z]+|SQL))rsjoinrefindall)ryrnames r7clsname_as_plain_namers>  #s|D 88NNrz2F'M'MNNN N NNr9instance_or_clsUnion[Type[Any], object]against_methodCallable[..., Any]boolczt|ts|j}n|}|j}t ||}||kS)z1Return True if the two class methods don't match.)r0r< __class__rsr/)rr current_cls method_namecurrent_methods r7method_is_overriddenrsG ot , ,&%/ %  )K'.{K'H'HN ^ ++r9slcsliceTuple[Any, ...]cg}|j|j|jfD];}t|dr|}||z!_unique_symbols..+s$Q-r9iz&exhausted namespace for symbol base %s)r} itertoolschainmapranger NameError)rrused_setpoolsymrs @r7_unique_symbolsr&s4yyH M M G ''''t 5 5   M MC("" S!!! # DtKLL L M Mr9fnCallable[[int], Any]rint Iterator[Any]c#LK|r||dzz}||V||z}|dSdS)z6Call the given function given each nonzero bit from n.rNrx)rrbs r7map_bitsr6sW  !aLbee  Q r9_FntargetCallable[[_Fn], _Fn]c0dfd }t|S)z'A signature-matching decorator factory.rrrAcPtj|s#tj|stdt |dd}|0d|_ t j|}||_n #||_wxYwt j|}dt|j pdz}t |j pd}| i||}ttd|dtd|dd z|jfz}t#|d d \}}t|| }|t'|d |j|d<tj|r d|d<d|d<n d|d<d|d<dt+|dvrd|z} nd|z} | ||d|ji} tt.jt3| | |j} |j| _|j| _t9| |S)Nznot a decoratable functionr-rMrx)rdefaultskwonlydefaultszTuple[str, ...]rrrr)rrFgroupedrzasync prefixzawait target_prefix__zf%(prefix)sdef %(name)s%(grouped_args)s: return %(target_prefix)s%(target)s(%(fn)s, %(apply_pos)s) ze%(prefix)sdef %(name)s%(grouped_args)s: return %(target_prefix)s%(target)s(%(fn)s, %(apply_kw)s) rs)rN isfunctionismethodrDr/r-rinspect_getfullargspeclenrr1fromkeysr_replacerr rsrrXformat_argspec_plusiscoroutinefunctionreprrttypes FunctionType_exec_code_in_env __defaults____kwdefaults__r) rannofuncspecempty_defaultsempty_kwdefaultsnames targ_namefn_namemetadatacodeenv decoratedrs r7decoratezdecorator..decorateGst!"%% :g.>r.B.B :899 92~t44  "BO +4R88"*(****044D!3t}':#;#;;==)<)BCC}}#+   $($q'22 3 3$d1Q3i00 1{n   -UHdCC 7-1w-O-O-O+D%@@@AAA;  &r * * +!)HX (0H_ % %!#HX (*H_ % 4Q==   D   v R       dC 5 5  "$ #%#4  i,,,s A00 A9)rrrArr)rrs` r7 decoratorrDs:T-T-T-T-T-T-l (F + ++r9rUnion[str, types.CodeType]rDict[str, Any]rc2t||||SrM)exec)rrrs r7rrs sOOO w<r9_PF_TEc8eZdZ dddZdZdd ZddZddZdS) PluginLoaderNgrouprTauto_fnOptional[Callable[..., Any]]c0||_i|_||_dSrM)rimplsr)rdrrs r7__init__zPluginLoader.__init__s %'  r9c8|jdSrM)rclearrcs r7rzPluginLoader.clears r9rrArcx||jvr|j|S|jr+||}|r||j|<|Stj|jD]2}|j|kr%|j|j|<|cS3tjd|jd|)NzCan't load plugin: :) rrrimportlib_metadata_getrrloadr NoSuchModuleError)rdrloaderimpls r7rzPluginLoader.loads 4:  #4:d#%% % < \\$''F #) 4 vxx1$*== # #DyD  #'9 4 yy{{"""!##*.***dd ;   r9 modulepathobjnamer_c*fd}||j|<dS)Nct}dddD]}t||}t|S)N.r) __import__splitr/)modtokenr r s r7rz#PluginLoader.register..loadsWZ((C#))#..qrr2 * *c5))3(( (r9r)rdrr r rs `` r7registerzPluginLoader.registers6 ) ) ) ) ) )   4r9c|j|=dSrMr)rdrs r7 deregisterzPluginLoader.deregisters Jt   r9rM)rrTrr)rrTrAr)rrTr rTr rTrAr_)rrTrAr_)rsrtrurrrrrrxr9r7rrs{BF    &    r9rc  tj}|j}|j}t |jd|t |j|zfS#t$r4tj |}|dt |dfcYSwxYw)Nrr) rNCO_VARKEYWORDS__code__ co_argcountlist co_varnamesrco_flagsr>rr)rco_varkeywordsconargsrs r7_inspect_func_argsr s  /[ ' ( ( ~- . .   &&& ,R00AwT!W %%%% &s A;B  B )_setraiseerrr<r!Optional[Set[str]]r" Literal[True]Set[str]cdSrMrxryr!r"s r7get_cls_kwargsr(s sr9FcdSrMrxr's r7r(r(s r9c|du}|rt}|J|jdd}|o8t|tjot|jtj}|rCt|\}}| ||s|s|rtd|ddSnd}|r|r|j D]}t||n| d|S)aeReturn the full set of inherited kwargs for the given `cls`. Probes a class's __init__ method, collecting all named arguments. If the __init__ defines a \**kwargs catch-all, then the constructor is presumed to pass along unrecognized keywords to its base classes, and the collection process is repeated recursively on each of the bases. Uses a subset of inspect.getfullargspec() to cut down on method overhead, as this is used within the Core typing system to create copies of type objects which is a performance-sensitive operation. No anonymous tuple arguments please ! NrFz given cls z doesn't have an __init__ method)r!rd)r}rPrQr0rrrrr rXrF __bases__r(discard) ryr!r"toplevelctrhas_initrhas_kwcs r7r(r(s="t|Huu     ,  :u - -C  5 sE. / / 5 s|U^ 4 4  *3// v E h  FFFFt v  Aad+++34 LL Kr9func List[str]c6tj|dS)zReturn the set of legal kwargs for the given `func`. Uses getargspec so is safe to call for methods, functions, etc. r)rr)r2s r7get_func_kwargsr5.s  ( . .q 11r9no_self_is_initcompat.FullArgSpecc tj|rtd|ztj|rq|r[|rYt j|}t j|jdd|j|j |j |j |j |j St j|Stj|r|rg|s|jr^t j|j}t j|jdd|j|j |j |j |j |j St j|jStj|rt%|j|dSt)|drt j|jSt)|drAtj|jrt%|j|Std |ztd |z) zReturn the argument signature for any callable. All pure-Python callables are accepted, including functions, methods, classes, objects with __call__; builtins and other edge cases like functools.partial() objects raise a TypeError. zCan't inspect builtin: %srNT)r6r7__func____call__)r6zCan't inspect callable: %s)rN isbuiltinrFrrr FullArgSpecargsvarargsvarkwr kwonlyargsrrr__self__r:isclassget_callable_argspecrrr;)rr6r7rs r7rDrD9s*;3b8999  B  (;  5 5044D% !""    #  044 4  "  ;  > >BK >0==D% !""    #  0== =    ;# K4     Z ;,R[999 Z ;  BK ( ( ?' WEEE E82=>> >4r9:::r9-Union[Callable[..., Any], compat.FullArgSpec]rDict[str, Optional[str]]c t|rtj|}n|}tj|}tj|d|d|dd|d}|drG|dd}tj|ddd|d|dd|d}n|drd|dz}|}nd}|}d}|dr6|t t t t|dz }|dr|t |dz }|d|dz}|r|d|z d} nd} tj||d|d| d } |dr/tj|dd|d|d| d } n| } |rt|||| || St|||dd | dd |dd | dd  S)aReturns a dictionary of formatted, introspected function arguments. A enhanced variant of inspect.formatargspec to support code generation. fn An inspectable callable or tuple of inspect getargspec() results. grouped Defaults to True; include (parens, around, argument) lists Returns: args Full inspect.formatargspec for fn self_arg The name of the first positional argument, varargs[0], or None if the function defines no positional arguments. apply_pos args, re-written in calling rather than receiving syntax. Arguments are passed positionally. apply_kw Like apply_pos, except keyword-ish args are passed as keywords. apply_pos_proxied Like apply_pos but omits the self/cls argument Example:: >>> format_argspec_plus(lambda self, a, b, c=3, **d: 123) {'grouped_args': '(self, a, b, c=3, **d)', 'self_arg': 'self', 'apply_kw': '(self, a, b, c=c, **d)', 'apply_pos': '(self, a, b, c, **d)'} rrrNz%s[0]rrxc&dt|zSN=rrSs r7rz%format_argspec_plus..scCFFlr9) formatvaluec&dt|zSrJrrLs r7rz%format_argspec_plus..s#A,r9) grouped_argsself_arg apply_posapply_kwapply_pos_proxiedapply_kw_proxied) rErrinspect_formatargspecrr rrr1) rrrr>rQrPrS num_defaults name_argsdefaulted_valsrRrTs r7rrqswH||,R00  ' .D, Qa$q'4aI Aw&71:"8 GABBKa$q'4a   a&T!W$%%L Aw7DsT!W55666  Aw%DG $ Q$q'!I"1|#3#5#56+ Q Q** H Aw $!7 abbM G G ..    $ /-     "oad^/"5-ad3     r9c|tjurd}|rdnd}|rdnd}n0 t||S#t$rd}|r|nd}|rdnd }YnwxYwt d||||| S) a$format_argspec_plus with considerations for typical __init__ methods Wraps format_argspec_plus with error handling strategies for typical __init__ cases: .. sourcecode:: text object.__init__ -> (self) other unreflectable (usually C) -> (self, *args, **kwargs) z(self)rdz()rrz(self, *args, **kwargs)zself, *args, **kwargsz(*args, **kwargs)z*args, **kwargs)rPrOrQrRrSrT)objectrrrFr1)methodrrOr>proxieds r7format_argspec_initr^s   ".xx!)$$r L&vw??? ? L L L4L#*G<<0GD-4K)):KGGG L !!    s0A  A rx target_clstarget_cls_sphinx_nameproxy_cls_sphinx_name classmethodsmethods attributesuse_intermediate_variableCallable[[_T], _T]c d}|S)aSA class decorator indicating attributes should refer to a proxy class. This decorator is now a "marker" that does nothing at runtime. Instead, it is consumed by the tools/generate_proxy_methods.py script to statically generate proxy methods and attributes that are fully recognized by typing tools such as mypy. c|SrMrx)rys r7rz&create_proxy_methods..decorates r9rx)r_r`rarbrcrdrers r7create_proxy_methodsris& Or9c tj|S#t$r!|tjur dgdddfcYSdgdddfcYSwxYw)ainspect.getargspec with considerations for typical __init__ methods Wraps inspect.getargspec with error handling for typical __init__ cases: .. sourcecode:: text object.__init__ -> (self) other unreflectable (usually C) -> (self, *args, **kwargs) rdNr>kwargs)rrrFr[r)r\s r7getargspec_initrl sn6,V444 666 V_ $ $HdD$/ / / /Hfh5 5 5 5 6sAAAcVt|tjr|js|jS|S)zSAdjust the incoming callable such that a 'self' argument is not required. )r0r MethodTyperBr:) func_or_clss r7unbound_method_to_callablerp4s2 +u/009M##r9 additional_kwSequence[Tuple[str, Any]] to_inspect%Optional[Union[object, List[object]]] omit_kwargc h|g}ntj|}tg}tj}d}t |D]%\}} t j|j} | jrt| jnd} |dkrW| j r| j }| r$| | j d| nR| | j ddn/| fd| j d| D| rI| jsJ| dt| j | d| jD#t$rY#wxYwg} | fd|D|=t!|r-| dt#|D|D]P\} } | |vr t#| }|ur || kr| | d|A#t($rYMwxYw|rN|D]K\} } t#| }|ur || kr| | d|<#t($rYHwxYwjjd d | d S) z}Produce a __repr__() based on direct association of the __init__() specification vs. same-named attributes present. Nrrcg|]}|fSrxrx)rargmissings r7 z generic_repr..dsIIIc7^IIIr9cg|] \}}||f Srxrx)rrxdefaults r7rzz generic_repr..js0(Cgr9c3TK|]"}tt|dV#dSrM)rr/)rrxr3s r7rzgeneric_repr..ss7DDC$wsC..//DDDDDDr9c,g|]}t|Srxr)rvals r7rzz generic_repr..vs@@@StCyy@@@r9rK(, ))rto_listr[ OrderedDict enumeraterrrrrr?rr>rXziprFrr/itemsrrDrrsr)r3rqrsrupos_argskw_argsvargsrinspr default_lenoutputrxdefvalrrys` @r7 generic_reprr@siU !)*55 hhGH2>2J2L2LG EZ((4 0??D15 D#dm,,,1KAvv<) LE3OODIa n$=>>>>OODIabbM2222IIIItyK</HIII }$$$,/ I{lmm4dm--%    H 4F MMDDDD8DDDDDD WS%00 @@GC,?,?@@@AAA}} V *    #sG,,C'!!cVmm cc2333    D (  KC c300g%%#--MMSSS##"6777    }---tyy/@/@/@/@ AAs6E'' E54E595H// H<;H< 5I?? J  J c0eZdZdZdZdZdZd dZdZdS) portable_instancemethodz_Turn an instancemethod into a (parent, name) pair to produce a serializable callable. )rrrk __weakref__c,|j|j|jdS)Nrrrkrrcs r7 __getstate__z$portable_instancemethod.__getstate__s kIk   r9cp|d|_|d|_|dd|_dS)Nrrrkrx)rrrQrk)rdstates r7 __setstate__z$portable_instancemethod.__setstate__s1Ho &M ii"-- r9rxcD|j|_|j|_||_dSrM)rBrrsrrk)rdmethrks r7rz portable_instancemethod.__init__sm M  r9cv||jt|j|j|i|SrM)rXrkr/rr)rdrxkws r7r;z portable_instancemethod.__call__s9 $+.wt{DI..:r:::r9N)rx) rsrtrurvrwrrrr;rxr9r7rrsf :I   ...  ;;;;;r9rc*|ht|j}|r|}fd|jD}|D],}|||-|jdkst|dstfdt|ts| n| |DD],}|||-|tS)a'Return an unordered sequence of all classes related to cls. Traverses diamond hierarchies. Fibs slightly: subclasses of builtin types are not returned. Thus class_hierarchy(class A(object)) returns (A, object), not A plus every class systemwide that derives from object. c3$K|] }|v|V dSrMrxr_hiers r7rz"class_hierarchy..s'99q1D======99r9builtinsrcg|]}|v| Srxrxrs r7rzz#class_hierarchy..s.    }}  }}r9) r__mro__r~r+rrrtr issubclassr<r)ryprocessr1rrsrs @r7class_hierarchyrs? 5D3;G  KKMM9999AK999  A NN1    HHQKKKK <: % %WQ8H-I-I %     "!T**)  """%%a((     A NN1    HHQKKKK+ , ::r9c#~Kt|}|D](}|jD]}||jvr||j|fVn)dS)ziterate all the keys and attributes associated with a class, without using getattr(). Does not use getattr() so that class-sensitive descriptors (i.e. property.__get__()) are not called. N)dirrrP)rykeyskeyr1s r7iterate_attributesrsl s88D  Aaj  AJsO,,,,!r9 self.proxyc|r|}n dfdt|D}|D]-} t||}t|ds%t|d|}ttj|} n#t $rY]wxYw tj| } tj | d} tj | ddd} n#t$rd} d } YnwxYwd tz} |dur||ipi}t| | | j ||_ n#t $rYnwxYwt|||/dS) z9Automates delegation of __specials__ for a proxying type.N)rw__del____getattribute__ __metaclass__rrcg|]B}|dr+|drt|s|v@|CS)r) startswithendswithr)rrZinto_clsskips r7rzz0monkeypatch_proxied_specials..sl     T""  JJt$$   !,,  TMM  "MMr9r;r:rrz(self, *args, **kw)z (*args, **kw)z?def %(method)s%(fn_args)s: return %(name)s.%(method)s%(d_args)s)rr/rr rrr>rrrVrFlocalsrrsetattr)rfrom_clsronlyr from_instancedundersr\maybe_fnrrfn_argsd_argspyrs` ` r7monkeypatch_proxied_specialsrs   <D      ]]    // x00H8Z00 xX>>He((33BB    H  %044D247;;G1$q'!""+>>FF % % %+G$FFF %  35;XX >  % ?4*? E2  R  ')CK $ $    D &#f+....?//s< A;+A;; BB ACC.-C.D00 D=<D=cFt|d|t|d|uS)z;Return True if the two methods are the same implementation.r:)r/)meth1meth2s r7methods_equivalentr%s0 5*e , , z511 r9c J|s|stdt|trt||r|St|pdt |D}tt |}t j}t|tr|}n,|st}t j}nt|}||||r|St|ts<|t jurdpd}t|d|dd |Gdd }|rd |j z|_ t} t|D]r\} } | |vrtd | zt| st| d | d t|| t| | | s|| |r|Stdd || z z)aEnsure basic interface compliance for an instance or dict of callables. Checks that ``obj`` implements public methods of ``cls`` or has members listed in ``methods``. If ``required`` is not supplied, implementing at least one interface method is sufficient. Methods present on ``obj`` that are not in the interface are ignored. If ``obj`` is a dict and ``dict`` does not meet the interface requirements, the keys of the dictionary are inspected. Keys present in ``obj`` that are not in the interface will raise TypeErrors. Raises TypeError if ``obj`` does not meet the interface criteria. In all passing cases, an object with callable members is returned. In the simple case, ``obj`` is returned as-is; if dict processing kicks in then an anonymous class is returned. obj A type, instance, or dictionary of callables. cls Optional, a type. All public methods of cls are considered the interface. An ``obj`` instance of cls will always pass, ignoring ``required``.. methods Optional, a sequence of method names to consider as the interface. required Optional, a sequence of mandatory implementations. If omitted, an ``obj`` that provides at least one interface method is considered sufficient. As a convenience, required may be a type, in which case all public methods of the type are required. z2a class or collection of method names are requiredc<g|]}|d|S)r)r)rrZs r7rzz as_interface..Ts)MMMa1<<;L;LMMMMr9zany ofzall ofz does not implement z: rceZdZdZdS)(as_interface..AnonymousInterfacezA callable-holding shell.Nrsrtrurvrxr9r7AnonymousInterfacerks''''r9r Anonymousz%r: unknown in this interfacerKz is not callablez,dictionary does not contain required keys %s)rFr0r<r}roperatorgegt intersectionr1rrsdictlike_iteritemsrEr staticmethodr) r3ryrcrequired interface implementedcomplies qualifierrfoundr\rs r7 as_interfacer-sxB NwNLMMM#tC!5!5 GMMM3s88MMMNNIc#hh--K{H(D!!! !55;x==x ((33X>>  c4   +8DH ssIIItyy333 5   (((((((( A&1CL&@# EEE*3//   " ";fDEE E~~ FvvvtttDEE E"FL,>,>??? &xx  "!! 6 ))Hu$ % % &  r9_GFDzgeneric_fn_descriptor[Any]ceZdZUdZded<ded<ded<dd d Zed!dZed"dZd#dZerd$dZ d%dZ d&dZ e d'dZ dS)(generic_fn_descriptoragDescriptor which proxies a function when the attribute is not present in dict This superclass is organized in a particular way with "memoized" and "non-memoized" implementation classes that are hidden from type checkers, as Mypy seems to not be able to handle seeing multiple kinds of descriptor classes used for the same attribute. Callable[..., _T_co]fgetrrvrTrsNdoccH||_|p|j|_|j|_dSrMrrvrsrdrrs r7rzgeneric_fn_descriptor.__init__s$ *dl   r9rdrr3r_ryrrAcdSrMrxrdr3rys r7__get__zgeneric_fn_descriptor.__get__s:=#r9r[r"cdSrMrxrs r7rzgeneric_fn_descriptor.__get__s7:sr9Union[_GFD, _T_co]ctrMNotImplementedErrorrs r7rzgeneric_fn_descriptor.__get__!###r9instancerhcdSrMrxrdrrhs r7__set__zgeneric_fn_descriptor.__set__r9cdSrMrx)rdrs r7 __delete__z generic_fn_descriptor.__delete__rr9ctrMrrdr3s r7_resetzgeneric_fn_descriptor._resetrr9rctrMrryr3rs r7resetzgeneric_fn_descriptor.resets!###r9rM)rrrr)rdrr3r_ryrrAr)r3r[ryrrAr")rdrr3rryrrArrrrhrrAr_rrrAr_)r3rrAr_)r3rrrTrAr_)rsrtrurvr:rrrrrrr classmethodrrxr9r7rrsMMM&&&&& ===X= :::X:$$$$9AAAA8888$$$$$$$[$$$r9rc eZdZdZesdZdSdS)_non_memoized_propertyza plain descriptor that proxies a function. primary rationale is to provide a plain attribute that's compatible with memoized_property which is also recognized as equivalent by mypy. c4||S||SrMrrs r7rz_non_memoized_property.__get__s{ 99S>> !r9N)rsrtrurvrrrxr9r7rrsA " " " " " """r9rc8eZdZdZesdZdZedZdS)_memoized_propertyz2A read-only @property that is only evaluated once.cV||S||x|j|j<}|SrM)rrPrsrdr3ryresults r7rz_memoized_property.__get__s.{ 3799S>> ACL '&Mr9cFt||jdSrM)rrrsrs r7rz_memoized_property._resets   dm44444r9c<|j|ddSrM)rPr~rs r7rz_memoized_property.resets  t$$$$$r9N) rsrtrurvrrrrrrxr9r7rrs[<<     555%%[%%%r9rc.fd}t|S)zDecorate a method memoize its return value. Best applied to no-arg methods: memoization is not sensitive to argument values, and will always return the same value even when called with different arguments. cz|g|Ri|fd}j|_j|_||jj<S)NcSrMrxarrs r7memoz6memoized_instancemethod..oneshot..memosMr9)rsrvrPrdr>rrrrs @r7oneshotz(memoized_instancemethod..oneshotscD&4&&&2&&       z %) bk" r9r)rrs` r7memoized_instancemethodrs.      '2 & &&r9ceZdZUdZesdZeZded<ddZ ddZ dd Z Gdde e ZeddZdS) HasMemoizedzA mixin class that maintains the names of memoized elements in a collection for easy cache clearing, generative, etc. rxzFrozenSet[str]_memoized_keysrAr_cP|jD]}|j|ddSrM)rrPr~rdelems r7_reset_memoizationszHasMemoized._reset_memoizationss9' * *D M  dD ) ) ) ) * *r9c0|jD] }||jvsJdSrM)rrPrs r7_assert_no_memoizationsz#HasMemoized._assert_no_memoizationss4' - -Dt},,,,, - -r9rrTrhrc<||j|<|xj|hzc_dSrM)rPr)rdrrhs r7_set_memoized_attributez#HasMemoized._set_memoized_attributes*" c u$r9creZdZUdZded<ded<ded<ddd ZeddZeddZdZdS)HasMemoized.memoized_attributezTA read-only @property that is only evaluated once. :meta private: Callable[..., _T]rrrvrTrsNrcH||_|p|j|_|j|_dSrMrrs r7rz'HasMemoized.memoized_attribute.__init__,s$DI.$,DL MDMMMr9rdr'r3r_ryrrAcdSrMrxrs r7rz&HasMemoized.memoized_attribute.__get__1s> ACL '&   4=/ 1  Mr9rM)rrrr)rdr'r3r_ryrrAr')r3rryrrAr!)rsrtrurvr:rrrrxr9r7memoized_attributer!s    * * * * * ??? ? 888 8     r9r$rr$c0dfd }t|S)zMDecorate a method memoize its return value. :meta private: rdrr>rrAc|g|Ri|fd}j|_j|_||jj<|xjjhzc_S)NcSrMrxr s r7rzBHasMemoized.memoized_instancemethod..oneshot..memoIs r9)rsrvrPrrs @r7rz4HasMemoized.memoized_instancemethod..oneshotFs{R*t***r**F     KDM:DL)-DM"+ &   BK= 0  Mr9)rdrr>rrrrArr)ryrrs ` r7rz#HasMemoized.memoized_instancemethod>s3      gr***r9Nrr)rrTrhrrAr_rr$rAr$)rsrtrurvrrw frozensetrr:rrrmemoized_propertyr!r$rrrxr9r7rrs  %.Y[[N0000****----%%%%.r2:+++[+++r9rc$eZdZdZdZdZd dZd S) MemoizedSlotszApply memoized items to an object using a __getattr__ scheme. This allows the functionality of memoized_property and memoized_instancemethod to be available to a class using __slots__. rxc t|rM)r>)rdrs r7_fallback_getattrzMemoizedSlots._fallback_getattresS!!!r9rrTrArcdsdrttjdr.t d}t ||Stjdr(t dfd}j|_|SS)N_memoized_attr__memoized_method_cv|i|fd}j|_j|_t|S)NcSrMrxr s r7rz8MemoizedSlots.__getattr__..oneshot..memozs!Mr9)rsrvr)r>rrrrrrds @r7rz*MemoizedSlots.__getattr__..oneshotwsYTR"""""!#  !z c4((( r9)rr>rrr/rrvr.)rdrrhrrs`` @r7 __getattr__zMemoizedSlots.__getattr__hs >>+ , , / 1 1  /!%% %T^%%>%> ? ? /838899B       !jGON))#.. .r9N)rrTrAr)rsrtrurvrwr.r4rxr9r7r,r,[sHI"""//////r9r,ct|trD|}|dvrdS|dvrdSt d|zt |S)N)trueyesonyt1T)falsenooffrf0FzString is not true/false: %r)r0rTstriprr2rrJs r7asboolrBsp#sCiikk!! 6 6 64 9 9 95;cABB B 99r9text!Callable[[str], Union[str, bool]]cdfd }|S)zqReturn a callable that will evaluate a string as boolean, or one of a set of "alternate" string values. r3rTrAUnion[str, bool]c.|vr|St|SrM)rB)r3rCs r7 bool_or_valuez"bool_or_str..bool_or_values $;;J#;; r9)r3rTrArFrx)rCrHs` r7 bool_or_strrIs)  r9rh Optional[int]c(||St|S)zCoerce to integer.)r)rhs r7asintrLs } u::r9rrrf flexi_booldestOptional[Dict[str, Any]]r_c||}||vrnt|trt|||sG||=|tur|rt||||<dS|||||<dSdSdSdS)zIf 'key' is present in dict 'kw', coerce its value to type 'type\_' if necessary. If 'flexi_bool' is True, the string '0' is considered false when coercing to boolean. N)r0r<rrB)rrrfrMrNs r7coerce_kw_typerQs | r E4(( 0:2c7E0J0J  sG  D==Z=r#wDIIIbgDIII     r9c`t|}|ftfd|DzS)zbProduce a tuple structure that is cacheable using the __dict__ of obj to retrieve values c3HK|]}|jv |j|fVdSrMrPrkr3s r7rz"constructor_key..sC!"qCL/@/@CLO/@/@/@/@r9)r(r)r3ryrs` r7constructor_keyrWsS 3  E 6E&+ r9r>ct|}|fd||D||i|S)zInstantiate cls using the __dict__ of obj as constructor arguments. Uses inspect to match the named arguments of ``cls``. c3HK|]}|jv |j|fVdSrMrTrUs r7rz#constructor_copy..sC!"a3<>O>OCLO>O>O>O>Or9)r(rX difference)r3ryr>rrs` r7constructor_copyr[ss 3  EII&+&6&6r&:&: 3   r9Callable[[], int]cdtjtjdfd}|S)z%Return a threadsafe counter function.rcZ5tcdddS#1swxYwYdSrM)next)counterlocksr7_nextzcounter.._nextsv  ! !== ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! !s  $$) threadingLockrcount)rbr`ras @@r7r`r`sG >  Doa  G!!!!!! Lr9specimenr|Optional[Type[Any]]ct|dr/|j!t|jtrtS|jSt |t rtnt}||t rt S||trtS||trtSt|drt St|drtSt|drtS|S)zGiven an instance or class, guess if it is or is acting as one of the basic collection types: list, set and dict. If the __emulates__ property is present, return that preferentially. __emulates__Nrrr})rrirr}r0r<rr1)rfr|isas r7duck_type_collectionrksx(()  ,  !32 2 ,J( ("8T22 B** C s8T Xs   Xt   x"" 5 ! ! 5 ! ! r9rxargtype'Union[Tuple[Type[Any], ...], Type[Any]]rc 6t||r|St|trHtjd|ddd|Ddt |dtjd|d|dt |d) Nz Argument 'z ' is expected to be one of type z or c3 K|] }d|zV dS)z'%s'Nrx)rr s r7rz"assert_arg_type..s&$A$AAVaZ$A$A$A$A$A$Ar9z, got ''z' is expected to be of type 'z', got ')r0rr ArgumentErrorrr<)rxrlrs r7assert_arg_typerrs#w  gu % % ##44$A$A$A$A$AAAAA49999N  ##44$s))))- r9c tdr!tStdtddt dztdrfd}|Stdr-t fd DSt dz) z?Return a (key, value) iterator for almost any dict-like object.r __getitem__rQNzObject '%r' is not dict-likeiterkeysc3`KD]}J||fVdSrM)ru)rdictlikegetters r7iteratorz$dictlike_iteritems..iterator.sT((** ' ')))66#;;&&&&& ' 'r9rc32K|]}||fVdSrMrx)rrrxs r7rz%dictlike_iteritems..5s0BB3S&&++&BBBBBBr9)rrrr/rFiterr)rwryrxs` @r7rr"sx!!&HNN$$%%% X}ght.L.L M MF ~6ABBBx$$ C ' ' ' ' ' ' xzz 6 " "CBBBB(--//BBBBBB6ABBBr9c:eZdZUdZded<dfd Zddd ZxZS) classpropertyaAA decorator that behaves like @property except that operates on classes rather than instances. The decorator is currently special when using the declarative module, but note that the :class:`~.sqlalchemy.ext.declarative.declared_attr` decorator should be used for this purpose with declarative. Callable[[Any], Any]rrxrrcZtj|g|Ri||j|_dSrM)superrrv)rdrrxrrs r7rzclassproperty.__init__Gs7****r***| r9Nr3ryOptional[type]rAc,||SrMrrs r7rzclassproperty.__get__Ksyy~~r9)rr~rxrrrrM)r3rryrrAr)rsrtrurvr:rr __classcell__rs@r7r}r}:sp$$$$$$r9r}c&eZdZddZdd Zdd Zd S)hybridpropertyr2rc"||_||_dSrM)r2clslevelrdr2s r7rzhybridproperty.__init__Ps  r9rrr.rAr!c^|||}|S||SrMrr2rdrr.clsvals r7rzhybridproperty.__get__T0  ]]5))FM99X&& &r9rhybridproperty[_T]c||_|SrMrrs r7 classlevelzhybridproperty.classlevel[  r9Nr2rrrr.rrAr!)r2rrAr)rsrtrurrrrxr9r7rrOsP''''r9rc6eZdZddZdd Zdd ZddZddZdS)rw_hybridpropertyr2rc0||_||_d|_dSrM)r2rsetfnrs r7rzrw_hybridproperty.__init__as  37 r9rrr.rAr!c^|||}|S||SrMrrs r7rzrw_hybridproperty.__get__frr9rhr_cD|jJ|||dSrMrrs r7rzrw_hybridproperty.__set__ms*z%%% 8U#####r9rrw_hybridproperty[_T]c||_|SrMrrs r7setterzrw_hybridproperty.setterqs  r9c||_|SrMrrs r7rzrw_hybridproperty.classlevelurr9Nrrr)r2rrAr)rsrtrurrrrrrxr9r7rr`sx8888 ''''$$$$r9rc*eZdZdZddZdd Zdd Zd S) hybridmethodz/Decorate a function as cls- or instance- level.r2rc0|x|_|_||_dSrM)r2r:rrs r7rzhybridmethod.__init__}s$(( DM r9rrr.rAc|| |j||jS|j||SrM)rrrr2)rdrr.s r7rzhybridmethod.__get__s:  =((@@ @9$$Xu55 5r9rhybridmethod[_T]c||_|SrMrrs r7rzhybridmethod.classlevelrr9Nr)rrr.rrAr)r2rrAr)rsrtrurvrrrrxr9r7rrzsV996666 r9rcjeZdZUdZded<iZded<ejZ ddd Z d Z dZ dZ dS)symbolaIA constant symbol. >>> symbol("foo") is symbol("foo") True >>> symbol("foo") A slight refinement of the MAGICCOOKIE=object() pattern. The primary advantage of symbol() is its repr(). They are also singletons. Repeated calls of symbol('name') will all return the same instance. rTrzDict[str, symbol]symbolsNrr canonicalrJrAc|j5|j|}|ct|tsJ|t |}t t|}||_ |r||_ ||j|<n|r||krtd|d||cdddS#1swxYwYdS)Nz#Can't replace canonical symbol for z with new int value ) _lockrrQr0rThashr__new__rrrvrF)ryrrrrs r7rzsymbol.__new__s0 Y  +//$''C{!$,,,,,$ $T Ikk&)44&"%CK %( D!!c!1!1#:d::.7::-                  sBB33B7:B7c>t|jdt|ffS)NrS)rrrrcs r7 __reduce__zsymbol.__reduce__s 3D 222r9c t|SrMrrcs r7__str__zsymbol.__str__sDzzr9cd|jdS)Nzsymbol(rrrcs r7__repr__zsymbol.__repr__s'''''r9)NN)rrTrrrrJrAr) rsrtrurvr:rrcrdrrrrrrxr9r7rrs  III!#G#### IN  E "#' <333(((((r9rceZdZdd Zdd ZdS) _IntFlagMeta classnamerTrTuple[Type[Any], ...]dict_rrrrAr_c gx|_}|D]\}}tjd|rt |t rt ||}n%|dstdgt|||| |tj d|D|_ dS)Nz^__.*__$)rrz#Expected integer values for IntFlagci|] }|j| Srxr)rrs r7 z)_IntFlagMeta.__init__..s , , ,sSXs , , ,r9)_itemsrrmatchr0rrrrFrrr immutabledict __members__) ryrrrrrrVvrs r7rz_IntFlagMeta.__init__s  UKKMM  DAqx Q'' !S!! Q!,,,\\#&&  EFFF CC LL    &4 , ,e , , ,  r9Iterator[symbol]c td)Nziter not implemented to ensure compatibility with Python 3.11 IntFlag. Please use __members__. See https://github.com/python/cpython/issues/99304rrcs r7__iter__z_IntFlagMeta.__iter__s! =   r9N) rrTrrrrrrrAr_)rAr)rsrtrurrrxr9r7rrs<    2      r9rceZdZdZdS) _FastIntFlagzAn 'IntFlag' copycat that isn't slow when performing bitwise operations. the ``FastIntFlag`` class will return ``enum.IntFlag`` under TYPE_CHECKING and ``_FastIntFlag`` otherwise. Nrrxr9r7rrsr9r) metaclass)IntFlag_EchoicesDict[_E, List[Any]]resolve_symbol_names Optional[_E]c|D]&\}}||ur|cS|r||jkr|cS||vr|cS'|dStjd|d|)aOGiven a user parameter, parse the parameter into a chosen value from a list of choice objects, typically Enum values. The user argument can be a string name that matches the name of a symbol, or the symbol object itself, or any number of alternate choices such as True/False/ None etc. :param arg: the user argument. :param choices: dictionary of enum values to lists of possible entries for each. :param name: name of the argument. Used in an :class:`.ArgumentError` that is raised if the parameter doesn't match any available argument. NzInvalid value for 'z': )rrr rq)rxrrr enum_valuechoices r7parse_user_argument_for_enumrs(&mmoo F *      ! cZ_&<&<    F]]    {t B$BB3BB C CCr9rc2t|_tdz adS)zAssign a '_creation_order' sequence to the given instance. This allows multiple instances to be sorted in order of creation (typically within a single thread; the counter is not particularly threadsafe). rN)_creation_order)rs r7set_creation_orderr$s /HqOOOr9rkc ||i|S#t$r/tdtjddzYdSwxYw)zXexecutes the given function, catches all exceptions and converts to a warning. z%s('%s') ignoredrrN)rDwarnrarb)r2r>rks r7warn_exceptionr1sc 7tT$V$$$ 777 #,..1"5 56666667s 5AAcl t||kr d|d|zS|S#t$r|cYSwxYw)Nz%s...r)rrF)rhlen_s r7ellipses_stringr<sT u::  U1T6]* *L  s$$ 33c@eZdZUdZded<dfd Zdd ZddZxZS)_hash_limit_stringa/A string subclass that can only be hashed on a maximum amount of unique values. This is used for warnings so that we can send out parameterized warnings without the __warningregistry__ of the module, or the non-overridable "once" registry within warnings.py, overloading memory, r_hashrhrTnumr> Sequence[Any]rAc||zd|zz}t||}td|t||zfz|_|S)Nz6 (this warning may be suppressed after %d occurrences)z%s_%d)rrrr)ryrhrr> interpolatedrdrs r7rz_hash_limit_string.__new__Ss_ Ds J wwsL11'UD,>,>,D$EEFF  r9c|jSrM)rrcs r7__hash__z_hash_limit_string.__hash__]s zr9otherrrcBt|t|kSrM)r)rdrs r7__eq__z_hash_limit_string.__eq__`sDzzT%[[((r9)rhrTrrr>rrAr)rAr)rrrAr) rsrtrurvr:rrrrrs@r7rrFs|JJJ))))))))r9rmsgc|r%ttj||dSt|tjdS)zdIssue a warning. If msg is a string, :class:`.exc.SAWarning` is used as the category. )rN)_warnings_warnr SAWarning)rrs r7rrdsG +s}St44455555sCM*****r9rc`|rt|d|}t|tjdS)z\Issue a warning with a parameterized string, limiting the number of registrations. N)rrr r)rr>s r7 warn_limitedrqs5  0 b$//3 &&&&&r9z)Dict[CodeType, Tuple[str, Type[Warning]]] _warning_tagsmessagecategory Type[Warning]Callable[[_F], _F]cfd}|S)Nc*ft|j<|SrM)rr)rrrs r7goz#tag_method_for_warnings..gos&-x%8 bk" r9rx)rrrs`` r7tag_method_for_warningsr~s) Ir9z&^(?:sqlalchemy\.(?!testing)|alembic\.)Union[str, Warning]Optional[Type[Warning]] stacklevelc tj|}dx}}|~|s5tjt|jddsd}|jtvr#d}t|j\}}|p|}|d|d}|j }|s|dz }n|r|rn|~n#t$rd}Ynd}YnxYw|tj |||dz dStj ||dz dS) NFrsrTz (rrr)r) ra _getframerr_not_sa_pattern f_globalsrQf_coderf_backr2warningsr)rrrframestacklevel_foundwarning_tag_found_suffix _categorys r7rrs] % j))054,$ (BH!4!4Z!D!D-- ( $( |},,$(!'4U\'B$)#0y$22222LE# a ! &7 3   < gxJNCCCCCC g*q.999999sB B.)B.rretry_on_exceptionCallable[..., Optional[_T]]c"gdfd }|S)zNDecorate the given function to be a no-op after it is called exactly once.rxrrrA Optional[_T]c}r;} ||i|S#rd|xYwdS)Nr)r~insert)rxr strong_fnonce_fnroncers r7rzonly_once..gosf  hhjjG w*r*** %,KK7+++ts#>)rxrrrrAr rx)rrrrs`` @r7 only_oncers= 4D         Ir9z%sqlalchemy/([a-z_]+/){0,2}[a-z_]+\.pyzunit(?:2|test2?/)tbexclude_prefixre.Pattern[str]exclude_suffixc`d}t|dz }||krA|||r&|dz }||kr|||&||krA|||r&|dz}||kr|||&|||dzS)atChop extraneous lines off beginning and end of a traceback. :param tb: a list of traceback lines as returned by ``traceback.format_stack()`` :param exclude_prefix: a regular expression object matching lines to skip at beginning of ``tb`` :param exclude_suffix: a regular expression object matching lines to skip at end of ``tb`` rr)rsearch)rrrrends r7chop_tracebackrs" E b''A+C 3,,>00E;;,   3,,>00E;;, 3,,>00C99, q 3,,>00C99, ecAgo r9c|d|z}t}t|||dS)Nz&def set(obj, value): obj.%s = valuer})rcopyr)attrnamerrs r7 attrsetterrs4 3h >D ((--//CsOOO u:r9z^__.+__$c(eZdZdZdZdfd ZxZS) TypingOnlyzA mixin class that marks a class as 'typing only', meaning it has absolutely no methods, attributes, or runtime functionality whatsoever. rxrAr_ct|jvr)d|jD}|rtd|d|dt dS)NcFh|]}t||Srx)_dundersrrrs r7 z/TypingOnly.__init_subclass__.. s9X^^D5I5Ir9zClass z< directly inherits TypingOnly but has additional attributes r )rr+rPAssertionErrorr__init_subclass__)ry remainingrs r7r&zTypingOnly.__init_subclass__ s  & &!$I $:S::-6::: !!#####r9rr)rsrtrurvrwr&rrs@r7rrsN I $ $ $ $ $ $ $ $ $ $r9rcJeZdZUdZded< d fd Zed d ZxZS) EnsureKWArga6Apply translation of functions to accept \**kw arguments if they don't already. Used to ensure cross-compatibility with third party legacy code, for things like compiler visit methods that need to accept ``**kw`` arguments, but may have been copied from old code that didn't accept them. rT ensure_kwargrAr_c2|j}|j}|re|D]b}tj||}|rI||}t j|}|js&||}t|||ct dSrM) r*rPrrrrr@ _wrap_w_kwrrr&) ryfn_regclsdictrrZrrwrappedrs r7r&zEnsureKWArg.__init_subclass__+s!,  3 3 3HVS))3 B!8<.wrap;s2s8Or9)rxrrrrArr)ryrr2s ` r7r,zEnsureKWArg._wrap_w_kw9s3      dB'''r9rr)rrrAr) rsrtrurvr:r&rr,rrs@r7r)r)s| $ $ $ $ $ $((([(((((r9r)c0t|drt||S|}|jj|_t|dr |j|_t|jdr|jjr|jj|_n|jr |j|_|S)zAugment functools.update_wrapper() to work with objects with a ``__call__()`` method. :param fn: object with __call__ method rsrtrv)rrrrsrtr;rv)wrapperr_fs r7 wrap_callabler6Asr: gr*** l+ 2| $ $ *MBM 2; * * $r{/B $,BJJ Z $BJ r9cd|vr|dSd}gg}d}t|}||kr||}|dkrE|dkr9||dz kr0||dzdkr!|dd|dz }nC|dz}n=|dkr|dkr|gn|d||dz }||kd|DS)aParse a dotted identifier with accommodation for quoted names. Includes support for SQL-style double quotes as a literal character. E.g.:: >>> quoted_token_parser("name") ["name"] >>> quoted_token_parser("schema.name") ["schema", "name"] >>> quoted_token_parser('"Schema"."Name"') ['Schema', 'Name'] >>> quoted_token_parser('"Schema"."Name""Foo"') ['Schema', 'Name""Foo'] "r rrrUc8g|]}d|S)r)r)rrs r7rzz'quoted_token_parser..s" / / /uBGGENN / / /r9)rrr)rhrridxlvchars r7quoted_token_parserr=Ys$ %{{3 E!dF C UB ((Sz 3;;zzcBFlluS1W~/D/Dr !!#&&&q  S[[UaZZ MM"     2J  d # # # q (( 0 / / / //r9paramsc>tjfd}|S)Ncv|jdur|jpd}|rt|fdD}||_|S)Nrci|]}|Srxrx)rparamrCs r7rz8add_parameter_text..decorate..s)J)J)J%%)J)J)Jr9)rvinject_param_text)rrr>rCs r7rz$add_parameter_text..decoratesTj$39r  L#C)J)J)J)J6)J)J)JKKC  r9)rr)r>rCrs`` r7add_parameter_textrDs9  !& ) )F Or9c|dd}t|dkr|S|\}}|ds|dztj|zStj|S)N rr)rrrtextwrapdedent)rC split_text firstliner's r7_dedent_docstringrKsuD!$$J :! ) 9    $ $%4(/)"<"<<<t$$$r9 given_doctext injecttextposc,t|pd}|d}t|dkr|dt j|d}|dr|dddt|D}|dd|t|t|dz }|d||z||dz}d |S)NrrFrrc@g|]\}}||Srx)rA)rrlines r7rzz)inject_docstring_text..s) H H Hic44::<< Hc H H Hr9) rKrrrrGrHr rminr)rLrMrNdoctextlines injectlinesblanks inject_poss r7inject_docstring_textrXs%]%8b99G MM$  E 5zzQ R/*--33D99K1~"1b!!! H H9U#3#3 H H HF MM!QCVq112J !J, + -jkk0B BE 99U  r9z(\s+):param (.+?):rS inject_paramsDict[str, str]cztj|}g}d}|rz|}t|}||r|dd}||vrwdt|dzdz}|rBtjd|d} | r%dt| dz}|||z}n|r| d|dgd}no| s| ||dgd}n@| dr+| ||ge| ||zd|S) Nr*rrz(\s+)\SrrFz::) collectionsdeque splitlinespopleft _param_regrrlstriprrrrstriprrr) rSrYdoclinesrT to_injectrQrZrBindentm2s r7rCrCs !3!3!5!566H EI !!   T " "   > ))#..M)) 3qwwqzz??2S8F <Xj(1+>><%(3rxx{{+;+;%;F &u)= =I   LL$ 40 1 1 1II  LL$ 40 1 1 1II ]]4   LL$ 0 0 2 23 4 4 4  T? B 99U  r9rc*t|dkrdSt|dk}|r |ddn|dd|ddz}d|D}|rd|Sd|ddd|dS) zTrims a list of strings from the middle and return a string of up to four elements. Strings greater than 11 characters will be truncatedrNrHrrUcRg|]$}t|dkr d|ddzn|%S) z%s..N)rr#s r7rzz$repr_tuple_names..s8 L L LdTR6D"I  T L L Lr9rz, ..., )rr)rflagress r7repr_tuple_namesrms 5zzQt u::?D ;E!A#JJE!A#Jrss$;E L Le L L LC >yy~~ $ #ac( 3 3 3 3SWW==r9c:trdS|rtddS)NTzDcython extensions were expected to be installed, but are not presentF)r ImportError)raise_s r7has_compiled_extrqs5t  "   ur9c,eZdZejZdS)_MissingN)rsrtruenumautoMissingrxr9r7rsrssdikkGGGr9rs)r3rr4r+rArB)r3rrArB)rSrrArT)ryrzrAr{)rrrArTrM)ryrrrrArT)rrrrrAr)rrrAr)rrrrTrAr)rrrrrAr)rrrAr)rrrrrrTrAr)ryr<r!r#r"r$rAr%)ryr<r!r#r"rrAr#)r2rrAr3)FF)rrr6rr7rrAr8)T)rrErrrArF)rxrxrxrx)r_rr`rTrarTrbrrcrrdrrerrArf)rxNrx) r3rrqrrrsrtrurrArT)NNrN)NNNr()r3rrAr)rCrTrArD)rhrrArJ)TN) rrrrTrfrrMrrNrOrAr_)r3rryrrAr) r3r!ryrzr>rrrrAr!)rAr\)rfrr|rgrArg)rxrrlrmrrTrAr)F) rxrrrrrTrrrArr)r2rr>rrkrrAr)r)rrTrrrAr_)rrTr>rrAr_)rrTrrrAr)Nr)rrrrrrrAr_)rrrrrAr )rr3rrrrrAr3)r>rrCrTrAr)rCrTrArT)rLrrMrTrNrrArT)rSrTrYrZrArT)rr3rAr)rv __future__rr]rt functoolsrrNrrrrarGrcrrtypingrrr r r r r rrrrrrrrrrrrrrrr_has_cyrrr r!r"r$r&r'r(r)py314 annotationlibr*r+r8r<rPrr=r@rGrKpy310r[r]rrrrrrrrrrrrrr r(r5rDrr^rirlrprrrrrrrrrrrr*non_memoized_propertypropertyro_memoized_propertyro_non_memoized_propertyrr!HasMemoized_ro_memoized_attributer$r,rBrIrLrQrWr[r`rkrrrr}rrrrrrrr FastIntFlagEnumrrrrrrrTrrrrr:rcompilerrr_SQLA_RE _UNITTEST_RErNoneTyperr"rr)r6r=rDrKrXrarCrmrqrsrv MissingOrrxr9r7rs#""""" $$$$$$   $$$$$$ WT]]4((( WT#s(+,,, ge3444 ge@AAA ge0111 ge./// <c2544444$$$$$$!M*;<D$BHHHHH \2,,,,, 2222"$$$$8282828282828282v    6666/3OOOOO,,,,$     M M M M ge/000 Y,Y,Y,Y,xgenn genn((((((((V   (  $!   -1E  .2E222222j2222EJ5;5;5;5;5;rHLr r r r r jJ#% "/12666(   028< " KBKBKBKBKB\;;;;;;;;<###L   &   @/@/@/@/FSSSSlwv9:::($($($($($GEN($($($V"""""259""""%%%%%.u5%%%<N.1 $'0BA,7MM4''''.J+J+J+J+J+J+J+J+ZG(0%%(3(F%*/*/*/*/*/*/*/*/\        %) '''''4        37     F$CCC0H*WR[" 472;$9(9(9(9(9(S9(9(9(x     4   D\KKK WT###"' DDDDDD    7777))))))))< + + + + +''''<> ===="*FGG )-0:0:0:0:0:f4 2:> ? ?rz.// '3&.4 4:: 2:j ! !$$$$$$$$*&(&(&(&(&(&(&(&(R0)0)0)0X     % % % %(RZ- . . ((((V > > > >    ty   "gh.// 0 r9