U h  @s 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"d#Z-e"d$ed%efd&Z.e"d'd(d&Z/e"d)d*d&Z0e"d+d,d&Z1e"d-d.d&Z2e'j3rZdd/l4m5Z5dd0l4m6Z6d1d2Z7e8j9d3j:Z;d4d5ZZ>n*e'j?rtd6d8d<d?d>Z>nd6d8d<d@d>Z>d6dAdBdCdDZ@GdEdFdFZAdGdHdIdJdKZBd6dAdLdMdNZCd@dOdPdAdQdRdSZDdTdUdVdWdXdYZEdZd[d\d]d^ZFd_dAd`dadbdcZGdddedfdgdhdiZHe"djdUd&ZIdUdkdldmdnZJdodpdAdUdqdrdsZKe"dtZLe"duZMGdvdwdwZNdxdyZOedd%dzd{d|d}d~dddZPedddzd{d|dVd|dddZPdddzd{d|dVd|dddZPdUddddZQdAdUdVdVddddZRdBddVddddZSdCddZTdDdOdAdAd_d_d_d_ddddZUddZVddZWdEd6ddd_dAdddZXGdddZYddZZddZ[dFddZ\ddZ]dGddZ^e"ddd&Z_Gdddee-Z`Gddde`e-ZaGddde`e-Zbe!re`Zce`ZdeeZfeeZgnebZcZfeaZdZgd$d$dddZhGdddZie!rd?d?ejZejZe#e,e*ejfZdS(OzRoutines 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)FormatcCsBt|dd}|dk r>t|||d}t|ts:t|d|SdS)N __annotate__)ownerz!.__annotate__ returned a non-dict)getattrr) isinstancedict ValueError)objformatZannotateannr4?/tmp/pip-unpacked-wheel-y9ln43lq/sqlalchemy/util/langhelpers.py_get_and_call_annotateFs  r6__annotations__cCsjt|tr2z t|}WqJtk r.iYSXnt|dd}|dkrJiSt|tsbt|dt|S)Nr7z+.__annotations__ is neither a dict nor None)r.type_BASE_GET_ANNOTATIONSAttributeErrorr-r/r0r1r3r4r4r5_get_dunder_annotationsRs     r<rr*Mapping[str, Any])r1r2returncCsz t|}Wntk r YnX|dk r2t|St||}|dkrLt|}|dkrxt|tsft|rjiSt|d|siSt|S)z:A sparse implementation of annotationlib.get_annotations()Nz does not have annotations)r< Exceptionr/r6r.r8callable TypeError)r1r2r3r4r4r5_vendored_get_annotationsds   rBr1r>cCst|tjdS)N)r2)rBr*Z FORWARDREFr1r4r4r5get_annotationssrEcCs t|SN)inspectrErDr4r4r5rEscCsBt|tr|jdd}n t|dd}|dkr4tjStd|SdS)Nr7r=)r.r8__dict__getr-r EMPTY_DICTrr;r4r4r5rEs   str)xr>cCs*|d}t}||tt|S)Nzutf-8)encoderZmd5_not_for_securityupdaterrK hexdigest)rLmr4r4r5md5_hexs  rQc@s@eZdZUdZdZded<ddddZd d d d d ddZdS) 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_infozfUnion[None, Tuple[Type[BaseException], BaseException, types.TracebackType], Tuple[(None, None, None)]]rSNoner>cCst|_dSrF)sysexc_inforSselfr4r4r5 __enter__szsafe_reraise.__enter__zOptional[Type[BaseException]]zOptional[BaseException]zOptional[types.TracebackType]r)type_value tracebackr>cCs`|jdk st|dkr@|j\}}}|dk s.td|_||nd|_|dk sRt||dSrF)rSAssertionErrorwith_traceback)rYr[r\r]exc_type exc_valueexc_tbr4r4r5__exit__s    zsafe_reraise.__exit__N)__name__ __module__ __qualname____doc__ __slots__r7rZrcr4r4r4r5rRs  rRzType[_T]zIterator[Type[_T]])clsr>ccsHt}|g}|rD|}||kr$q n |||||Vq dSrF)setpopaddextend__subclasses__)riseenstackr4r4r5walk_subclassess rq)elementr>cCs:t|tr|Sz t|WStk r4d|YSXdS)Nzunprintable element %r)r.rKr?)rrr4r4r5string_or_unprintables   rsz Type[Any] Optional[str])riuse_namer>cCs&|p|j}dddtd|DS)N css|]}|VqdSrF)lower).0nr4r4r5 sz(clsname_as_plain_name..z([A-Z][a-z]+|SQL))rdjoinrefindall)rirunamer4r4r5clsname_as_plain_names rzUnion[Type[Any], object]Callable[..., Any]bool)instance_or_clsagainst_methodr>cCs.t|ts|j}n|}|j}t||}||kS)z1Return True if the two class methods don't match.)r.r8 __class__rdr-)rrZ current_cls method_nameZcurrent_methodr4r4r5method_is_overriddens   rslicezTuple[Any, ...])slcr>cCs>g}|j|j|jfD] }t|dr*|}||qt|S)zrdecode a slice object as sent to __getitem__. takes into account the 2.5 __index__() method, basically. __index__)startstopstephasattrrappendtuple)rretrLr4r4r5 decode_slices   rz Sequence[str]z Iterator[str])usedbasesr>c'sht|}|D]Vtftfddtd}|D] }||kr4|||Vq q4tdq dS)Ncs t|SrFrK)ibaser4r5+z!_unique_symbols..iz&exhausted namespace for symbol base %s)rj itertoolschainmaprangerl NameError)rrZused_setpoolsymr4rr5_unique_symbols&s rzCallable[[int], Any]intz Iterator[Any])fnryr>ccs*|r&||d@}||V||N}qdS)z6Call the given function given each nonzero bit from n.rNr4)rrybr4r4r5map_bits6s r_FnzCallable[[_Fn], _Fn])targetr>csdddfdd }t|S)z'A signature-matching decorator factory.rrr>c st|st|stdt|dd}|dk rPd|_zt|}W5||_Xn t|}dt|j pfd}t |j pxd}|j i||d}ttd|dtd|dd |jf}t|d d \}}t ||d }|t|d d|j|d<t|rd|d<d|d<nd|d<d|d<dt|dkr:d|} nd|} |||d|ji} ttjt| | |j} |j| _|j| _t| |S)Nznot a decoratable functionr+rFr4)rdefaultskwonlydefaultszTuple[str, ...]rrrr)rrFgroupedr~zasync prefixzawait Z 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) rd)rG isfunctionismethodr?r-r+rinspect_getfullargspeclenrr/fromkeysr_replacerrrdrrNformat_argspec_plusiscoroutinefunctionreprretypes FunctionType_exec_code_in_env __defaults____kwdefaults__r) rZannofuncspecZempty_defaultsZempty_kwdefaultsnamesZ targ_namefn_namemetadatacodeenvZ decoratedrr4r5decorateGsl           zdecorator..decorater)rrr4rr5 decoratorDsVrzUnion[str, types.CodeType]Dict[str, Any])rrrr>cCst||||SrF)exec)rrrr4r4r5rs r_PF_TEc@sZeZdZddddddZddZdd d d d Zdddd dddZdd d ddZdS) PluginLoaderNrKzOptional[Callable[..., Any]])groupauto_fncCs||_i|_||_dSrF)rimplsr)rYrrr4r4r5__init__szPluginLoader.__init__cCs|jdSrF)rclearrXr4r4r5rszPluginLoader.clearr)r~r>cCs||jkr|j|S|jr:||}|r:||j|<|St|jD]&}|j|krF|j|j|<|SqFtd|j|fdS)NzCan't load plugin: %s:%s) rrrZimportlib_metadata_getrr~loadrZNoSuchModuleError)rYr~loaderimplr4r4r5rs       zPluginLoader.loadrT)r~ modulepathobjnamer>csfdd}||j|<dS)Ncs4t}dddD]}t||}qt|S)N.r) __import__splitr-)modtokenrrr4r5rs z#PluginLoader.register..loadr)rYr~rrrr4rr5registerszPluginLoader.registercCs |j|=dSrFr)rYr~r4r4r5 deregisterszPluginLoader.deregister)N)rdrerfrrrrrr4r4r4r5rs  rcCsnz tj}Wn2tk r<t|}|dt|dfYSX|j}|j}t|j d|t|j |@fSdS)Nrr) rGCO_VARKEYWORDSr:rrr__code__ co_argcountlist co_varnamesco_flags)rZco_varkeywordsrconargsr4r4r5_inspect_func_argss   r)_setraiseerrr8zOptional[Set[str]]z Literal[True]zSet[str])rirrr>cCsdSrFr4rirrr4r4r5get_cls_kwargssrFcCsdSrFr4rr4r4r5rsc Cs|dk}|rt}|dk st|jdd}|oHt|tjoHt|jtj}|rt |\}}| ||s|s|rt d|dqdSnd}|r|r|j D]}t ||ddkrqq|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)rrY)rjr^rHrIr.rrrrrrNrA __bases__rdiscard) rirrtoplevelZctrZhas_initrZhas_kwcr4r4r5rs4        z List[str]funcr>cCst|dS)zReturn the set of legal kwargs for the given `func`. Uses getargspec so is safe to call for methods, functions, etc. r)rrrr4r4r5get_func_kwargs.srzcompat.FullArgSpec)rno_self_is_initr>c CsDt|rtd|n&t|rn|rb|rbt|}t|jdd|j|j |j |j |j |j St|Snt|r|r|s|jrt|j}t|jdd|j|j |j |j |j |j St|jSntt|rt|j|ddSt|drt|jSt|dr4t|jr&t|j|dStd |n td |dS) 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)rr__func____call__)rzCan't inspect callable: %s)rG isbuiltinrArrr FullArgSpecargsvarargsvarkwr kwonlyargsrrr__self__risclassget_callable_argspecrrr)rrrrr4r4r5r9sR             rz-Union[Callable[..., Any], compat.FullArgSpec]zDict[str, Optional[str]])rrr>c Cst|rt|}n|}tj|}t|d|d|dd|d}|dr|dd}t|ddd|d|dd|d}n"|drd|d}|}nd}|}d}|dr|tttt|d7}|dr|t|d7}|d|d}|r |d|d} nd} tj||d|d| d d d } |drdtj|dd|d|d| d d d } n| } |rt|||| || d St|||dd| dd|dd| ddd SdS)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]rr4cSs dt|SN=rrLr4r4r5rrz%format_argspec_plus..) formatvaluecSs dt|Srrrr4r4r5rr) grouped_argsself_arg apply_posapply_kwapply_pos_proxiedapply_kw_proxied) r@rrinspect_formatargspecrrrrr/) rrrrr rr  num_defaultsZ name_argsZdefaulted_valsr r r4r4r5rqs$           rcCs||tjkr(d}|rdnd}|r"dnd}n@zt||dWStk rfd}|rR|nd}|r^dnd }YnXtd|||||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)rYz()rrz(self, *args, **kwargs)zself, *args, **kwargsz(*args, **kwargs)z*args, **kwargs)rrr r r r )objectrrrAr/)methodrrrZproxiedr4r4r5format_argspec_inits$   rr4zCallable[[_T], _T]) target_clstarget_cls_sphinx_nameproxy_cls_sphinx_name classmethodsmethods attributesuse_intermediate_variabler>cCs dd}|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. cSs|SrFr4)rir4r4r5rsz&create_proxy_methods..decorater4)rrrrrrrrr4r4r5create_proxy_methodssrcCsTz t|WStk rN|tjkr8dgdddfYSdgdddfYSYnXdS)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) rYNrkwargs)rrrArr)rr4r4r5getargspec_init s   rcCs t|tjr|js|jS|SdS)zSAdjust the incoming callable such that a 'self' argument is not required. N)r.r MethodTyperr)Z func_or_clsr4r4r5unbound_method_to_callable4srzSequence[Tuple[str, Any]]z%Optional[Union[object, List[object]]])r1 additional_kw to_inspect omit_kwargr>c sV|dkrg}n t|}tg}t}d}t|D]\}}zt|j} Wntk rhYq8Yq8X| j rzt | j nd} |dkr| j r| j }| r| | j d| q| | j ddn$|fdd| j d| D| r8| j st|ddt| j | d| j Dq8g} | fdd|D|dk rlt|rl| d dt|D|D]f\} } | |krqtz6t| }|k r|| kr| d | |fWntk rYnXqt|r@|D]X\} } z6t| }|k r"|| kr"| d | |fWntk r:YnXqd jjd | fS) z}Produce a __repr__() based on direct association of the __init__() specification vs. same-named attributes present. Nrrcsg|] }|fqSr4r4rxarg)missingr4r5 dsz generic_repr..cSsg|]\}}||fqSr4r4)rxr#defaultr4r4r5r%jsc3s|]}tt|dVqdSrF)rr-r"rDr4r5rzsszgeneric_repr..cSsg|] }t|qSr4r)rxvalr4r4r5r%vsz%s=%rz%s(%s), )rto_listr OrderedDict enumeraterrrrArrrrmrrNr^ziprr-itemsrr?rrdr{)r1rr r!Zpos_argsZkw_argsZvargsrZinsprZ default_lenoutputr#Zdefvalr(r4)r$r1r5 generic_repr@sl          r0c@s6eZdZdZdZddZddZd dd Zd d Zd S)portable_instancemethodz_Turn an instancemethod into a (parent, name) pair to produce a serializable callable. )rr~r __weakref__cCs|j|j|jdS)Nrr~rr3rXr4r4r5 __getstate__sz$portable_instancemethod.__getstate__cCs&|d|_|d|_|dd|_dS)Nrr~rr4)rr~rIr)rYstater4r4r5 __setstate__s  z$portable_instancemethod.__setstate__r4cCs|j|_|j|_||_dSrF)rrrdr~r)rYmethrr4r4r5rsz portable_instancemethod.__init__cOs ||jt|j|j||SrF)rNrr-rr~)rYr#kwr4r4r5rs z portable_instancemethod.__call__N)r4) rdrerfrgrhr4r6rrr4r4r4r5r1s  r1cs|ht|j}|r|}fdd|jD}|D]}|||q4|jdkst|dsdqfddt|t s| n| |DD]}|||qqtS)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. c3s|]}|kr|VqdSrFr4rx_Zhierr4r5rzsz"class_hierarchy..builtinsrncsg|]}|kr|qSr4r4r9r;r4r5r%sz#class_hierarchy..) r__mro__rkrrrlrer issubclassr8rn)riprocessrrrsr4r;r5class_hierarchys$       rAccs@t|}|D].}|jD]"}||jkr||j|fVq qq 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)dirr=rH)rikeyskeyrr4r4r5iterate_attributess   rE self.proxyc s<|r |}n$dkrdfddt|D}|D]}z4t||}t|dsRWq2t|d|}ttj|} Wntk rYq2YnXz2t| } t | d} t | ddd} Wnt k rd } d } YnXd t } |dk r||ipi}t | |z| j ||_ Wntk r$YnXt|||q2dS) z9Automates delegation of __specials__ for a proxying type.N)rh__del____getattribute__ __metaclass__r4r6cs6g|].}|dr|drt|s|kr|qS)r) startswithendswithrrxrPinto_clsskipr4r5r%s    z0monkeypatch_proxied_specials..rrrrz(self, *args, **kw)z (*args, **kw)z?def %(method)s%(fn_args)s: return %(name)s.%(method)s%(d_args)s)rBr-rrrrr:rrrrAlocalsrrsetattr)rNZfrom_clsrOonlyr~Z from_instanceZdundersrZmaybe_fnrrZfn_argsZd_argspyrr4rMr5monkeypatch_proxied_specialssF          rTcCst|d|t|d|kS)z;Return True if the two methods are the same implementation.r)r-)Zmeth1Zmeth2r4r4r5methods_equivalent%s  rUc Csp|s|stdt|tr(t||r(|St|p>ddt|D}tt|}tj}t|trd|}n|svt}tj}nt|}||||r|St|t s|tjkrdpd}td||d |fGdd d }|rd |j |_ t} t |D]R\} } | |krtd | t | s*td | | ft|| t| | | q|| |rV|Std d || dS)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 requiredcSsg|]}|ds|qS)r:)rJrLr4r4r5r%Ts z as_interface..zany ofzall ofz%r does not implement %s: %sr)c@seZdZdZdS)z(as_interface..AnonymousInterfacezA callable-holding shell.Nrdrerfrgr4r4r4r5AnonymousInterfaceksrWZ Anonymousz%r: unknown in this interfacez%r=%r is not callablez,dictionary does not contain required keys %sN)rAr.r8rjrBoperatorgegt intersectionr/r{rddictlike_iteritemsr@rQ staticmethodrl) r1rirrequiredZ interfaceZ implementedZcompliesZ qualifierrWfoundrrr4r4r5 as_interface-sP!          r`_GFDzgeneric_fn_descriptor[Any]c@seZdZUdZded<ded<ded<d$ddd d d Zed d dd dddZeddddddZd ddddddZerddd dddZ dd dddZ dd ddd Z e ddd d!d"d#Z 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. zCallable[..., _T_co]fgetrtrgrKrdNrcdoccCs||_|p|j|_|j|_dSrFrcrgrdrYrcrer4r4r5rs zgeneric_fn_descriptor.__init__rarTrrYr1rir>cCsdSrFr4rYr1rir4r4r5__get__szgeneric_fn_descriptor.__get__rr!r1rir>cCsdSrFr4rir4r4r5rjszUnion[_GFD, _T_co]cCs tdSrFNotImplementedErrorrir4r4r5rjsinstancer\r>cCsdSrFr4rYror\r4r4r5__set__rzgeneric_fn_descriptor.__set__ror>cCsdSrFr4)rYror4r4r5 __delete__rz generic_fn_descriptor.__delete__rCcCs tdSrFrlrYr1r4r4r5_resetszgeneric_fn_descriptor._reset)r1r~r>cCs tdSrFrlrir1r~r4r4r5resetszgeneric_fn_descriptor.reset)N)rdrerfrgr7rrrjrrqrsru classmethodrwr4r4r4r5rbs  rbc@seZdZdZesddZdS)_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. cCs|dkr |S||SrFrcrir4r4r5rjsz_non_memoized_property.__get__N)rdrerfrgrrjr4r4r4r5rysryc@s0eZdZdZesddZddZeddZdS) _memoized_propertyz2A read-only @property that is only evaluated once.cCs&|dkr |S|||j|j<}|SrF)rcrHrdrYr1riresultr4r4r5rjsz_memoized_property.__get__cCst||jdSrF)r{rwrdrtr4r4r5rusz_memoized_property._resetcCs|j|ddSrF)rHrkrvr4r4r5rwsz_memoized_property.resetN) rdrerfrgrrjrurxrwr4r4r4r5r{s r{rcsfdd}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. cs<|f||fdd}j|_j|_||jj<S)NcsSrFr4ar8r}r4r5memosz6memoized_instancemethod..oneshot..memo)rdrgrHrYrr8rrrr5oneshots   z(memoized_instancemethod..oneshotr)rrr4rr5memoized_instancemethods rc@s~eZdZUdZesdZeZded<ddddZ ddd d Z d d dd ddZ Gddde e ZedddddZdS) HasMemoizedzA mixin class that maintains the names of memoized elements in a collection for easy cache clearing, generative, etc. r4zFrozenSet[str]_memoized_keysrTrUcCs|jD]}|j|dqdSrF)rrHrkrYelemr4r4r5_reset_memoizationss zHasMemoized._reset_memoizationscCs|jD]}||jkstqdSrF)rrHr^rr4r4r5_assert_no_memoizationss z#HasMemoized._assert_no_memoizationsrKr)rDr\r>cCs||j|<|j|hO_dSrF)rHr)rYrDr\r4r4r5_set_memoized_attributes z#HasMemoized._set_memoized_attributec@sreZdZUdZded<ded<ded<dddd d d Zed d dd dddZeddddddZddZdS)zHasMemoized.memoized_attributezTA read-only @property that is only evaluated once. :meta private: Callable[..., _T]rcrtrgrKrdNrdcCs||_|p|j|_|j|_dSrFrfrgr4r4r5r,s z'HasMemoized.memoized_attribute.__init__r&rTrrhcCsdSrFr4rir4r4r5rj1sz&HasMemoized.memoized_attribute.__get__r rkcCsdSrFr4rir4r4r5rj4scCs8|dkr |S|||j|j<}|j|jhO_|SrF)rcrHrdrr|r4r4r5rj7s )N)rdrerfrgr7rrrjr4r4r4r5memoized_attribute!s rr#rcs"dddddfdd }t|S)zMDecorate a method memoize its return value. :meta private: r)rYrr8r>csN|f||fdd}j|_j|_||jj<|jjhO_S)NcsSrFr4r~rr4r5rIszBHasMemoized.memoized_instancemethod..oneshot..memo)rdrgrHrrrrr5rFs  z4HasMemoized.memoized_instancemethod..oneshotr)rirrr4rr5r>s z#HasMemoized.memoized_instancemethodN)rdrerfrgrrh frozensetrr7rrrmemoized_propertyr rrxrr4r4r4r5rs rc@s,eZdZdZdZddZddddd Zd 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__. r4cCs t|dSrF)r:)rYrDr4r4r5_fallback_getattreszMemoizedSlots._fallback_getattrrKr)rDr>csdsdrtn|tjdrRtd}t||Stjdrtdfdd}j|_|SSdS)NZ_memoized_attr_Z_memoized_method_cs6||fdd}j|_j|_t|S)NcsSrFr4r~rr4r5rzsz8MemoizedSlots.__getattr__..oneshot..memo)rdrgrQ)rr8rrrDrYrr5rws    z*MemoizedSlots.__getattr__..oneshot)rJr:rrr-rQrgr)rYrDr\rr4rr5 __getattr__hs   zMemoizedSlots.__getattr__N)rdrerfrgrhrrr4r4r4r5r[srcCsBt|tr:|}|dkr"dS|dkr.dStd|t|S)N)trueyesonyt1T)falsenooffryf0FzString is not true/false: %r)r.rKstriprwr0rrDr4r4r5asbools   rz!Callable[[str], Union[str, bool]])textr>csdddfdd }|S)zqReturn a callable that will evaluate a string as boolean, or one of a set of "alternate" string values. rKzUnion[str, bool]rCcs|kr |St|SdSrF)rrDrr4r5 bool_or_valuesz"bool_or_str..bool_or_valuer4)rrr4rr5 bool_or_strsr Optional[int])r\r>cCs|dkr |St|S)zCoerce to integer.N)r)r\r4r4r5asintsrzOptional[Dict[str, Any]]rT)r8rDr[ flexi_booldestr>cCsj|dkr |}||krft|tr,t|||sf||dk rf|tkrV|rVt||||<n|||||<dS)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)r.r8rr)r8rDr[rrr4r4r5coerce_kw_types    rrkcs$t|}|ftfdd|DS)zbProduce a tuple structure that is cacheable using the __dict__ of obj to retrieve values c3s&|]}|jkr|j|fVqdSrFrHrxkrDr4r5rzs z"constructor_key..)rr)r1rirr4rDr5constructor_keysr)r1rirr8r>cs0t|}|fdd||D|||S)zInstantiate cls using the __dict__ of obj as constructor arguments. Uses inspect to match the named arguments of ``cls``. c3s&|]}|jkr|j|fVqdSrFrrrDr4r5rzs z#constructor_copy..)rrN difference)r1rirr8rr4rDr5constructor_copys rzCallable[[], int]rUcs$ttdfdd}|S)z%Return a threadsafe counter function.rc s$tW5QRSQRXdSrF)nextr4counterlockr4r5_nextszcounter.._next) threadingLockrcount)rr4rr5rs rzOptional[Type[Any]])specimenr&r>cCst|dr*|jdk r$t|jtr$tS|jSt|tr8tnt}||trJtS||trXtS||trftSt|drttSt|drtSt|drtS|SdS)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__Nrrlrj)rrr>rjr.r8rr/)rr&Zisar4r4r5duck_type_collections*        rz'Union[Tuple[Type[Any], ...], Type[Any]])r#argtyper~r>cCs^t||r|St|trBtd|ddd|Dt|fntd||t|fdS)Nz8Argument '%s' is expected to be one of type %s, got '%s'z or css|]}d|VqdS)z'%s'Nr4)rxrr4r4r5rzsz"assert_arg_type..z6Argument '%s' is expected to be of type '%s', got '%s')r.rr ArgumentErrorr{r8)r#rr~r4r4r5assert_arg_types   rcstdrtStdtdddkr>tdtdr\fdd}|Std rtfd d DStddS) z?Return a (key, value) iterator for almost any dict-like object.r. __getitem__rINzObject '%r' is not dict-likeiterkeysc3s,D]}dk st||fVqdSrF)rr^)rDdictlikegetterr4r5iterator.s  z$dictlike_iteritems..iteratorrCc3s|]}||fVqdSrFr4)rxrD)rr4r5rz5sz%dictlike_iteritems..)rrr.r-rAiterrC)rrr4rr5r\"s     r\csHeZdZUdZded<ddddfdd Zd dd dd d d ZZS) 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. zCallable[[Any], Any]rcr)rcr#r8cs tj|f|||j|_dSrF)superrrg)rYrcr#r8rr4r5rGszclassproperty.__init__NzOptional[type]rkcCs ||SrFrzrir4r4r5rjKszclassproperty.__get__)N)rdrerfrgr7rrj __classcell__r4r4rr5r:s  rc@s<eZdZddddZdddddd Zd d d d dZdS)hybridpropertyrrcCs||_||_dSrF)rclslevelrYrr4r4r5rPszhybridproperty.__init__rr ror,r>cCs$|dkr||}|S||SdSrFrrrYror,Zclsvalr4r4r5rjTs zhybridproperty.__get__rzhybridproperty[_T]rcCs ||_|SrFrrr4r4r5 classlevel[szhybridproperty.classlevelN)rdrerfrrjrr4r4r4r5rOsrc@s^eZdZddddZdddddd Zddd d d d ZdddddZdddddZdS)rw_hybridpropertyrrcCs||_||_d|_dSrF)rrsetfnrr4r4r5raszrw_hybridproperty.__init__rr rcCs$|dkr||}|S||SdSrFrrr4r4r5rjfs zrw_hybridproperty.__get__rTrncCs|jdk st|||dSrF)rr^rpr4r4r5rqmszrw_hybridproperty.__set__rzrw_hybridproperty[_T]rcCs ||_|SrF)rrr4r4r5setterqszrw_hybridproperty.settercCs ||_|SrFrrr4r4r5ruszrw_hybridproperty.classlevelN)rdrerfrrjrqrrr4r4r4r5r`s rc@s@eZdZdZddddZdddddd Zd d d d dZdS) hybridmethodz/Decorate a function as cls- or instance- level.rrcCs||_|_||_dSrF)rrrrr4r4r5r}s zhybridmethod.__init__rrcCs*|dkr|j||jS|j||SdSrF)rrjrr)rYror,r4r4r5rjszhybridmethod.__get__rzhybridmethod[_T]rcCs ||_|SrFrrr4r4r5rszhybridmethod.classlevelN)rdrerfrgrrjrr4r4r4r5rzsrc@s\eZdZUdZded<iZded<eZdddddd d d Z d d Z ddZ ddZ 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. rKr~zDict[str, symbol]symbolsNrtr)r~re canonicalr>c Cs|j|j|}|dkrbt|ts*t|dkr:t|}tt |}||_ |rV||_ ||j|<n |r||krt d|d||W5QRSQRXdS)Nz#Can't replace canonical symbol for z with new int value ) _lockrrIr.rKr^hashr__new__rr~rgrA)rir~rerrr4r4r5rs     zsymbol.__new__cCst|jdt|ffS)NrL)rr~rrXr4r4r5 __reduce__szsymbol.__reduce__cCst|SrFr'rXr4r4r5__str__szsymbol.__str__cCsd|jdS)Nzsymbol()r~rXr4r4r5__repr__szsymbol.__repr__)NN) rdrerfrgr7rrrrrrrrr4r4r4r5rs  rc@s0eZdZddddddddZd d d d Zd S) _IntFlagMetarKzTuple[Type[Any], ...]rrrT) classnamerdict_r8r>c Ksg|_}|D]Z\}}td|r(qt|tr@t||d}n|dstdnqt |||| |qt dd|D|_ dS)Nz^__.*__$)rr:z#Expected integer values for IntFlagcSsi|] }|j|qSr4r)rxrr4r4r5 sz)_IntFlagMeta.__init__..)Z_itemsr.r|matchr.rrrJrArQrrZ immutabledict __members__) rirrrr8r.rvrr4r4r5rs        z_IntFlagMeta.__init__zIterator[symbol]rUcCs tddS)Nziter not implemented to ensure compatibility with Python 3.11 IntFlag. Please use __members__. See https://github.com/python/cpython/issues/99304rlrXr4r4r5__iter__sz_IntFlagMeta.__iter__N)rdrerfrrr4r4r4r5rsrc@seZdZdZdS) _FastIntFlagzAn 'IntFlag' copycat that isn't slow when performing bitwise operations. the ``FastIntFlag`` class will return ``enum.IntFlag`` under TYPE_CHECKING and ``_FastIntFlag`` otherwise. NrVr4r4r4r5rsr) metaclass)IntFlag_EzDict[_E, List[Any]]z Optional[_E])r#choicesr~resolve_symbol_namesr>cCsn|D]>\}}||kr |S|r6||jkr6|S||kr|Sq|dkrTdStd|d|dS)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': )r.r~rr)r#rr~rZ enum_valuechoicer4r4r5parse_user_argument_for_enums rrrcCst|_td7adS)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)ror4r4r5set_creation_order$s r)rrrr>cOs>z |||WStk r8tdtddYnXdS)zXexecutes the given function, catches all exceptions and converts to a warning. z%s('%s') ignoredrrN)r?warnrVrW)rrrr4r4r5warn_exception1s rcCsFz(t||kr d|d|WS|WSWntk r@|YSXdS)Nz%s...r)rrA)r\Zlen_r4r4r5ellipses_string<s   rcsTeZdZUdZded<dddddfdd Zdd d d Zd d dddZZS)_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_hashrK Sequence[Any])r\numrr>cs<||d|}t||}td|t||f|_|S)Nz6 (this warning may be suppressed after %d occurrences)z%s_%d)rrrr)rir\rrZ interpolatedrYrr4r5rSs z_hash_limit_string.__new__rUcCs|jSrF)rrXr4r4r5__hash__]sz_hash_limit_string.__hash__rr)otherr>cCst|t|kSrF)r)rYrr4r4r5__eq__`sz_hash_limit_string.__eq__) rdrerfrgr7rrrrr4r4rr5rFs   r)msgrr>cCs(|rttj||dn t|tjdS)zdIssue a warning. If msg is a string, :class:`.exc.SAWarning` is used as the category. )rN)_warnings_warnr SAWarning)rrr4r4r5rdsrr)rrr>cCs |rt|d|}t|tjdS)z\Issue a warning with a parameterized string, limiting the number of registrations. N)rrrr)rrr4r4r5 warn_limitedqs rz)Dict[CodeType, Tuple[str, Type[Warning]]] _warning_tagsz Type[Warning]zCallable[[_F], _F])messagecategoryr>csfdd}|S)Ncsft|j<|SrF)rrrrrr4r5gosz#tag_method_for_warnings..gor4)rrrr4rr5tag_method_for_warnings~srz&^(?:sqlalchemy\.(?!testing)|alembic\.)zUnion[str, Warning]zOptional[Type[Warning]])rr stacklevelr>cCszt|}Wn&tk r&d}Ynd}YnXd}}|dk r|sdtt|jddsdd}|jt krd}t |j\}}|p|}|d|d}|j }|s|d7}q>|r>|r>qq>|dk rt j |||dd nt j ||dd dS) NrFrdrTz (rr)r ) rV _getframer0r|r_not_sa_pattern f_globalsrIf_coderf_backwarningsr)rrr frameZstacklevel_foundZwarning_tag_found_suffixZ _categoryr4r4r5rs4    rrzCallable[..., Optional[_T]])rretry_on_exceptionr>cs$gddddfdd }|S)zNDecorate the given function to be a no-op after it is called exactly once.rz Optional[_T]r#r8r>cs@}r<}z |||WSr4d|YnXdS)Nr)rkinsert)r#r8Z strong_fnZonce_fnroncerr4r5rs  zonly_once..gor4)rrrr4rr5 only_oncesrz%sqlalchemy/([a-z_]+/){0,2}[a-z_]+\.pyzunit(?:2|test2?/)zre.Pattern[str])tbexclude_prefixexclude_suffixr>cCs`d}t|d}||kr0|||r0|d7}q||krP|||rP|d8}q0|||dS)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)rrrrendr4r4r5chop_tracebacks   rcCs$d|}t}t|||dS)Nz&def set(obj, value): obj.%s = valuerj)rPcopyr)attrnamerrr4r4r5 attrsetters  r z^__.+__$cs*eZdZdZdZddfdd ZZS) TypingOnlyzA mixin class that marks a class as 'typing only', meaning it has absolutely no methods, attributes, or runtime functionality whatsoever. r4rTrUcsBt|jkr4dd|jD}|r4td|d|dtdS)NcSsh|]}t|s|qSr4)_dundersrrxr~r4r4r5 s z/TypingOnly.__init_subclass__..zClass z< directly inherits TypingOnly but has additional attributes r)r!rrHr^r__init_subclass__)ri remainingrr4r5r% s zTypingOnly.__init_subclass__)rdrerfrgrhr%rr4r4rr5r!sr!csDeZdZUdZded<ddfdd Zeddd d d ZZS) 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. rK ensure_kwargrTrUcsf|j}|j}|rX|D]B}t||}|r||}t|}|js||}t|||qt dSrF) r(rHr|rrrr _wrap_w_kwrQrr%)riZfn_regZclsdictrDrPrrwrappedrr4r5r%+s   zEnsureKWArg.__init_subclass__rrcs ddddfdd }t|S)Nrrcs|SrFr4)r#r8rr4r5wrap;sz$EnsureKWArg._wrap_w_kw..wrapr)rirr+r4rr5r)9szEnsureKWArg._wrap_w_kw) rdrerfrgr7r%rxr)rr4r4rr5r's   r'cCsjt|drt||S|}|jj|_t|dr4|j|_t|jdrT|jjrT|jj|_n|jrb|j|_|SdS)zAugment functools.update_wrapper() to work with objects with a ``__call__()`` method. :param fn: object with __call__ method rdrergN)rrrrdrerrg)wrapperr_fr4r4r5 wrap_callableAs     r.cCsd|kr|dSd}gg}d}t|}||kr||}|dkr|dkr|||dkr|||ddkr||dd|d7}q|dN}n*|dkr|dkr|gn|d||d7}q(dd|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'] "rrrr cSsg|]}d|qS)r)r{)rxrr4r4r5r%sz'quoted_token_parser..)rrr)r\r5r}idxlvcharr4r4r5quoted_token_parserYs$ $    r3)paramsrr>cstfdd}|S)Ncs:|jdk r|jpd}|r0t|fddD}||_|S)Nrcsi|] }|qSr4r4)rxparamrr4r5rsz8add_parameter_text..decorate..)rginject_param_text)rrer4rr4r5rs z$add_parameter_text..decorate)rr*)r4rrr4r7r5add_parameter_texts r8cCsN|dd}t|dkr|S|\}}|ds@|dt|St|SdS)N rrv)rrrJtextwrapdedent)rZ split_text firstliner&r4r4r5_dedent_docstrings   r=) given_doctext injecttextposr>cCst|pd}|d}t|dkr,|dt|d}|drP|ddddt|D}|dd|t|t|d}|d||||d}d |S)Nrr9rrcSsg|]\}}|s|qSr4)r)rxrliner4r4r5r%sz)inject_docstring_text..) r=rrrr:r;rr,minr{)r>r?r@doctextlinesZ injectlinesZblanksZ inject_posr4r4r5inject_docstring_texts      rEz(\s+):param (.+?):zDict[str, str])rC inject_paramsr>c Cs t|}g}d}|r|}t|}|dkr|r|dd}||krdt|dd}|rt d|d} | rdt| d}|||}nV|r| d|dgd}n<| s| ||dgd}n| dr| ||gq| |qd|S) Nr*rvrz(\s+)\Srr9z::) collectionsdeque splitlinespopleft _param_regrrlstriprr|rmrstriprKrr{) rCrFZdoclinesrDZ to_injectrArPr5indentm2r4r4r5r6s6   r6)rr>cCst|dkrdSt|dk}|r,|ddn|dd|dd}dd|D}|r`d|Sd d|dd|dfSdS) zTrims a list of strings from the middle and return a string of up to four elements. Strings greater than 11 characters will be truncatedrNrrr cSs,g|]$}t|dkr$d|ddn|qS) z%s..N)rr#r4r4r5r%sz$repr_tuple_names..r)z %s, ..., %s)rr{)rflagresr4r4r5repr_tuple_namess  ( rTcCstrdS|rtdndSdS)NTzDcython extensions were expected to be installed, but are not presentF)r ImportError)Zraise_r4r4r5has_compiled_extsrVc@seZdZeZdS)_MissingN)rdrerfenumautoMissingr4r4r4r5rWsrW)N)FF)T)T)r4r4r4r4)r4Nr4)NNrFN)NNN)TN)N)F)r)N)Nr)F)rg __future__rrHrX functoolsrrGrrXr|rVr:rrrtypingrrrr r r r r rrrrrrrrrrrrrrrZ_has_cyrrrr r!r#r%r&r'r(Zpy314Z annotationlibr)r*r6r8rHrjr9r<rBrEZpy310rQrRrqrsrrrrrrrrrrrrrrrrrrrrr0r1rArErTrUr`rarbryr{rZnon_memoized_propertypropertyZro_memoized_propertyZro_non_memoized_propertyrrZ!HasMemoized_ro_memoized_attributerrrrrrrrrrrr\rrrrrrrrrZ FastIntFlagEnumrrrrrrrKrrrrr7rcompiler rrZ_SQLA_REZ _UNITTEST_RErZNoneTyper r"r!r'r.r3r8r=rErLr6rTrVrWrZZ MissingOrr4r4r4r5 s                                   !;  \+5 9u %N& C V +M.   #<"  "      3     )  ,   +