o DDit @s:UddlmZmZddlmZmZmZmZmZm Z m Z m Z m Z m Z mZmZddlZddlmZmZddlmZddlmZmZmZmZddlmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(ddl)m*Z*zdd l+m,Z,Wne-yGd d d Z,Ynwhd d e*Ddd e*DZ.e e/e0d<de1de/de2fddZ3deddfddZ4Gddde5Z6dede eej7fddZ8de1de2fddZ9de1de ee1fd d!Z:d"e eej7de ee1fd#d$Z;d%e eej7d&ed'efd(e/d)e2ddf d*d+Zd/eee ed'effd(e/d,e=d0e2ddf d1d2Z?de1deee ej7ffd3d4Z@d(e/d5eej7gdfddfd6d7ZAde1deee eej7gdfffd8d9ZBde1deee ed'efffd:d;ZCdejDdeee ej7ffd?d@ZEdejDddf dKdLZJGdMdNdNeZKGdOdPdPeZLdS)Q)ismethod signature) AnyCallablecastDictget_type_hintsListOptionalSequenceSetTupleTypeUnionN)CSTTransformer CSTVisitor)CSTNodeT)CONSTRUCTED_LEAVE_MATCHER_ATTRCONSTRUCTED_VISIT_MATCHER_ATTRVISIT_NEGATIVE_MATCHER_ATTRVISIT_POSITIVE_MATCHER_ATTR) AllOfAtLeastNAtMostNBaseMatcherNodeextract extractallfindallmatches MatchIfTrue MatchMetadataMatchMetadataIfTrueOneOfreplace)TYPED_FUNCTION_RETURN_MAPPING) UnionTypec@s eZdZdS)r%N)__name__ __module__ __qualname__r)r)R/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/matchers/_visitors.pyr%5sr%cCh|]}d|jqS)visit_r&.0clsr)r)r* :r1cCr+)leave_r-r.r)r)r*r1;r2CONCRETE_METHODSobj attr_namereturncCsttt||dtS)z6Check if obj.attr is a property without evaluating it.N) isinstancegetattrtypeproperty)r5r6r)r)r* is_property?sr<kwargsMatchDecoratorMismatchcCstdi|S)Nr))r>)r=r)r)r*_match_decorator_unpicklerEsr?csLeZdZdededdffdd Zdeedeedfffd d ZZ S) r>funcmessager7Ncs(td|d|||_||_dS)NzInvalid function signature for z: )super__init__r@rA)selfr@rA __class__r)r*rCJs zMatchDecoratorMismatch.__init__).r>.cCst|j|jdffS)N)r@rA)r?r@rArDr)r)r* __reduce__Osz!MatchDecoratorMismatch.__reduce__) r&r'r(strrCr robjectrH __classcell__r)r)rEr*r>Is matchercCs.t|ttfrdd|jDStt|jjgS)NcSsg|] }tt|jjqSr))r9cstrFr&r/mr)r)r* Zsz/_get_possible_match_classes..)r8r"roptionsr9rMrFr&)rLr)r)r*_get_possible_match_classesXsrR annotationcCst|tp t|ddtuS)N __origin__)r8r%r9rrSr)r)r*_annotation_is_union_s rVcCs$t|r t|dgSttt|gS)N__args__)rVr9rrrJrUr)r)r*_get_possible_annotated_classesfs rXclassescCs&t}|D] }|tt|q|SN)setupdaterXr$)rYretvalr0r)r)r*(_get_valid_leave_annotations_for_classesms  r^possible_match_classesmeth.decorator_name expected_nonec Cst|}|r |dtdtdurt|jd|dddSd|vr&dSt|d}t|}|D]}|D] }t||r?nq6t|jd|dd|jdq2dS)Nr7@z( should only decorate functions that do z not return.z" decorated function cannot return z the type .) rgetr:r>r(rXr^ issubclassr&) r_r`rarb type_hintspossible_annotated_classespossible_returnsretrSr)r)r*_verify_return_annotation~s>    rkexpected_param_countc Cst|}t|j|kr$t|jd|d|d|dkr ddddt|D}|D])}t|}|D] }|D] } t|| rFnq=t|jd|d |j d d d q9q1dS) Nrcz& should decorate functions which take z parameterscSsg|] \}}|dkr|qS)r7r))r/kvr)r)r*rPsz1_verify_parameter_annotations..z can be called with  z4but the decorated function parameter annotations do znot include this type.) rlen parametersr>r(ritemsrXrfr&) r_r`rarlmeth_signatureparamsparamrhmatchrSr)r)r*_verify_parameter_annotationss@     rz decoratormapexpected_none_returnc Cs~|D]8\}}t|}tdd|D}|D]$}t|s#t|jd|r+t|jdt||||dt||||dqqdS)Ncss |] }t|tttfVqdSrZ)r8rrrrNr)r)r* s  z_check_types..zoMatcher decorators should only be used on methods of MatcherDecoratableTransformer or MatcherDecoratableVisitorz]The root matcher in a matcher decorator cannot be an AtLeastN, AtMostN or MatchIfTrue matcher)rb)rl)rurRanyrr>r(rkrz) r{rarlr|rLmethodsr_has_invalid_top_levelr`r)r)r* _check_typess<rcCsji}t|D],}t||s2t||}t|tgD] }d|tt|<qt|tgD] }d|tt|<q(q|S)z Set of gating matchers that we need to track and evaluate. We use these in conjunction with the call_if_inside and call_if_not_inside decorators to determine whether to call a visit/leave function. N)dirr<r9rrrr)r5visit_matchersr6r@rLr)r)r*_gather_matcherss   rr@cCs&|jtvrt|jd|dddS)Nrcz1 should not decorate functions that are concrete zvisit or leave methods.)r&r4r>r()rar@r)r)r*_assert_not_concrete%s  rcCi}t|D]A}t||rqt||}t|sqtttjgdf|}t|tg}|r0t d||D]}tt |}g| |d|R||<q2q|S)Nvisitr)) rr<r9rrrrMCSTNoderrrrer5constructed_visitorsfuncname possible_funcr@matchersrLcasted_matcherr)r)r*_gather_constructed_visit_funcs0*        rcCr)Nleaver)) rr<r9rrrrMrrrrrerr)r)r*_gather_constructed_leave_funcsOrrrnodemetadata_resolvercCsBi}|D]\}}|durt|||dr|||<q|||<q|SNr)rur)rrr new_matchersrL existing_noder)r)r*_visit_matchersks   rcCs4i}|D]\}}||urd||<q|||<q|SrZ)ru)rrrrLrr)r)r*_leave_matchers~s   r all_matcherscCs,t|tg}|D] }||durdSqdSNFT)r9rrr5requested_matchersrLr)r)r*_all_positive_matchers_trues  rcCs,t|tg}|D] }||durdSqdSr)r9rrr)r)r*_all_negative_matchers_falses  rcCst||o t||SrZ)rr)rr5r)r)r*_should_allow_visits r visit_funcscCs@|D]\}}t|||dr|D] }t||r||qqdSr)rurr)rrrrrL visit_funcr)r)r*_visit_constructed_funcss rc@sReZdZdZd#ddZedeeee j ffddZ e j deeee j fddfd dZ d e j de fd d Zd ededeee jffddZd e j deddfddZd e j deddfddZd ee je je j fdede fddZdee je je j fdeeee j eefdee j fddZd ee je je j fdedeeeee j ee j fffddZdee je je j fdeeee j eefdeeeee j ee j fffddZdee je je j fdeeee j eefd ee je je j ee j eeee j ee j ffgee je je j fffdee je je j ffd!d"Z dS)$MatcherDecoratableTransformera9 This class provides all of the features of a :class:`libcst.CSTTransformer`, and additionally supports various decorators to control when methods get called when traversing a tree. Use this instead of a :class:`libcst.CSTTransformer` if you wish to do more powerful decorator-based visiting. r7NcCsLt|d|_t||_t||_t|jddddt|jdddddS)NrrmTrlr|rF)rrC(_MatcherDecoratableTransformer__matchersr_extra_visit_funcsr_extra_leave_funcsrrGr)r)r*rCs$    z&MatcherDecoratableTransformer.__init__cC|jdur t||_|jSrZ)rrrGr)r)r* _matchers  z'MatcherDecoratableTransformer._matchersvaluecC ||_dSrZ)rrDrr)r)r*r rcCRt|j|||_t|j|j||t|jt|dt|jds#dSt ||SNr,T) rrrrrr9r:r&ron_visitrDrr)r)r*r z&MatcherDecoratableTransformer.on_visit original_node updated_nodecCst|jt|dt|jdrt|||}n|}tt|j D]!\}}| ||s.q#|D]}t|j|rCt |t jrC|||}q0q#t|j||_|SNr3)rrr9r:r&ron_leavereversedlistrrurr8rMrr)rDrrr]rL leave_funcs leave_funcr)r)r*rs"  z&MatcherDecoratableTransformer.on_leave attributecC8t|jt|dt|jd|drt|||SdSNr,_)rrr9r:r&ron_visit_attributerDrrr)r)r*r z0MatcherDecoratableTransformer.on_visit_attributecC<t|jt|dt|jd|drt|||dSdSNr3r)rrr9r:r&ron_leave_attributerDrrr)r)r*r( z0MatcherDecoratableTransformer.on_leave_attributerLcCt|||dSap A convenience method to call :func:`~libcst.matchers.matches` without requiring an explicit parameter for metadata. Since our instance is an instance of :class:`libcst.MetadataDependent`, we work as a metadata resolver. Please see documentation for :func:`~libcst.matchers.matches` as it is identical to this function. rrrDrrLr)r)r*r2 z%MatcherDecoratableTransformer.matchestreecCrap A convenience method to call :func:`~libcst.matchers.findall` without requiring an explicit parameter for metadata. Since our instance is an instance of :class:`libcst.MetadataDependent`, we work as a metadata resolver. Please see documentation for :func:`~libcst.matchers.findall` as it is identical to this function. rrrDrrLr)r)r*r@z%MatcherDecoratableTransformer.findallcCrap A convenience method to call :func:`~libcst.matchers.extract` without requiring an explicit parameter for metadata. Since our instance is an instance of :class:`libcst.MetadataDependent`, we work as a metadata resolver. Please see documentation for :func:`~libcst.matchers.extract` as it is identical to this function. rrrr)r)r*rSrz%MatcherDecoratableTransformer.extractcCrav A convenience method to call :func:`~libcst.matchers.extractall` without requiring an explicit parameter for metadata. Since our instance is an instance of :class:`libcst.MetadataDependent`, we work as a metadata resolver. Please see documentation for :func:`~libcst.matchers.extractall` as it is identical to this function. rrrr)r)r*rarz(MatcherDecoratableTransformer.extractall replacementcCt||||dSap A convenience method to call :func:`~libcst.matchers.replace` without requiring an explicit parameter for metadata. Since our instance is an instance of :class:`libcst.MetadataDependent`, we work as a metadata resolver. Please see documentation for :func:`~libcst.matchers.replace` as it is identical to this function. rr#rDrrLrr)r)r*r#tz%MatcherDecoratableTransformer.replacer7N)!r&r'r(__doc__rCr;rrr rMrrsetterboolrrrRemovalSentinelrrIrr MaybeSentinelrrr r!r rrrrr#r)r)r)r*rs #"          rc@sFeZdZdZd"ddZedeeee j ffddZ e j deeee j fddfd dZ d e j de fd d Zd e j ddfddZd e j deddfddZd e j deddfddZd ee je je j fdede fddZdee je je j fdeeee j eefdee j fddZd ee je je j fdedeeeee j ee j fffddZdee je je j fdeeee j eefdeeeee j ee j fffddZdee je je j fdeeee j eefdee je je j ee j eeee j ee j ffgee je je j fffdee je je j ffd d!ZdS)#MatcherDecoratableVisitora1 This class provides all of the features of a :class:`libcst.CSTVisitor`, and additionally supports various decorators to control when methods get called when traversing a tree. Use this instead of a :class:`libcst.CSTVisitor` if you wish to do more powerful decorator-based visiting. r7NcCsLt|d|_t||_t||_t|jddddt|jdddddS)NrrmTrr)rrC$_MatcherDecoratableVisitor__matchersrrrrrrGr)r)r*rCs$  z"MatcherDecoratableVisitor.__init__cCrrZ)rrrGr)r)r*rrz#MatcherDecoratableVisitor._matchersrcCrrZ)rrr)r)r*rrrcCrr) rrrrrr9r:r&rrrr)r)r*rrz"MatcherDecoratableVisitor.on_visitrcCst|jt|dt|jdrt||tt|j D]\}}| ||s*q|D] }t|j|r8||q,qt |j||_dSr) rrr9r:r&rrrrrrurr)rDrrLrrr)r)r*rs   z"MatcherDecoratableVisitor.on_leavercCrr)rrr9r:r&rrrr)r)r*rrz,MatcherDecoratableVisitor.on_visit_attributecCrr)rrr9r:r&rrrr)r)r*rrz,MatcherDecoratableVisitor.on_leave_attributerLcCrrrrr)r)r*rrz!MatcherDecoratableVisitor.matchesrcCrrrrr)r)r*rrz!MatcherDecoratableVisitor.findallcCrrrrr)r)r*rrz!MatcherDecoratableVisitor.extractcCrrrrr)r)r*r)rz$MatcherDecoratableVisitor.extractallrcCrrrrr)r)r*r#<rz!MatcherDecoratableVisitor.replacer) r&r'r(rrCr;rrr rMrrrrrrrIrrrrrrrr r!r rrrrr#r)r)r)r*rs "        r)Minspectrrtypingrrrrrr r r r r rrlibcstrMrr libcst._typesrlibcst.matchers._decoratorsrrrrlibcst.matchers._matcher_baserrrrrrrrrr r!r"r#libcst.matchers._return_typesr$typesr% ImportErrorr4rI__annotations__rJrr<r? Exceptionr>rrRrVrXr^rkintrzrrrrrMetadataDependentrrrrrrrrr)r)r)r*s8 <          2   - 4             Z