o DDiK @sUddlmZmZmZmZmZddlZddlm Z dZ e e d<dZ e e d<eejejejejejejejejejejejejf Zde d e fd d Zde d ee fd d Zde dee d e fddZGdddejZGdddej Z!dej"dee efd ej"fddZ#e$Z%ej$e d<e%fde dej$ded ej&fddZ'e%fde dej$ded eej(ej)ffddZ*e%fde dej$ded ejfd d!Z+dS)")DictMappingOptionalSetUnionN) ensure_type__LIBCST_MANGLED_NAME_TEMPLATE_PREFIX_EMAN_DELGNAM_TSCBIL__TEMPLATE_SUFFIXvarreturncCst|tSN)r r )r rQ/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/helpers/_template.py mangled_name srcCs@t|vrt|vr|td\}}|td\}}|s|s|SdS)N)r r split)r prefixname_and_suffixnamesuffixrrrunmangled_name$s rtemplate template_varscCsZt|vst|vr td|D]}d|d}||vr"td|d||t|}q|S)Nz3Cannot parse a template containing reserved strings{}z+Template string is missing a reference to "z" referred to in kwargs)r r Exceptionreplacer)rrr originalrrrmangle_template.s  r c@seZdZdeeefddfddZdejdejdej fdd Z dej dej dej fd d Z dej dej dej fd d ZdejdejdejfddZdejdejdejfddZdejdejdejfddZdejdejdejfddZdejdejdejfddZdejdejdejfddZdejdejdejfddZdej dej dej!fddZ"dej#dej#dej#fddZ$dej%dej%dej%fd d!Z&dS)"TemplateTransformertemplate_replacementsr Ncsdd|D|_dd|D|_dd|D|_dd|D|_dd|D|_dd|D|_dd|D|_d d|D|_d d|D|_ d d|D|_ d d|D|_ d d|D|_ hdd|jDdd|jDdd|jDdd|jDdd|jDdd|jDdd|jDdd|jDdd|j Ddd|j Ddd|j Ddd|j Dfdd|D}|rt dtt|ddS)NcS"i|] \}}t|tjr||qSr) isinstancecstBaseExpression.0rvaluerrr @  z0TemplateTransformer.__init__..cSr#r)r$r% Annotationr'rrrr*Er+cSr#r)r$r% AssignTargetr'rrrr*Jr+cSr#r)r$r%Paramr'rrrr*Or+cSr#r)r$r% Parametersr'rrrr*Tr+cSr#r)r$r%Argr'rrrr*Yr+cSr#r)r$r%BaseSmallStatementr'rrrr*^r+cSr#r)r$r% BaseStatementr'rrrr*cr+cSr#r)r$r% BaseSuiter'rrrr*hr+cSr#r)r$r%SubscriptElementr'rrrr*mr+cSr#r)r$r% BaseSlicer'rrrr*rr+cSr#r)r$r% Decoratorr'rrrr*wr+cSg|]}|qSrrr(rrrr z0TemplateTransformer.__init__..cSr7rrr8rrrr9r:cSr7rrr8rrrr9r:cSr7rrr8rrrr9r:cSr7rrr8rrrr9r:cSr7rrr8rrrr9r:cSr7rrr8rrrr9r:cSr7rrr8rrrr9r:cSr7rrr8rrrr9r:cSr7rrr8rrrr9r:cSr7rrr8rrrr9r:cSr7rrr8rrrr9r:csh|]}|vr|qSrrr8supported_varsrr sz/TemplateTransformer.__init__..zTemplate replacement for "z" is unsupported)itemssimple_replacementsannotation_replacementsassignment_replacementsparam_replacementsparameters_replacementsarg_replacementssmall_statement_replacementsstatement_replacementssuite_replacementssubscript_element_replacementssubscript_index_replacementsdecorator_replacementsrnextiter)selfr"unsupported_varsrr;r__init__=s     zTemplateTransformer.__init__ original_node updated_nodecCs.t|j}|dus||jvr|S|j|Sr)rr)r? deep_clonerMrPrQvar_namerrr leave_Names zTemplateTransformer.leave_NamecC8|j}t|tjrt|j}||jvr|j|S|Sr) annotationr$r%Namerr)r@rR)rMrPrQrWrTrrrleave_Annotation    z$TemplateTransformer.leave_AnnotationcCrVr)targetr$r%rXrr)rArR)rMrPrQr[rTrrrleave_AssignTargetrZz&TemplateTransformer.leave_AssignTargetcCs(t|jj}||jvr|j|S|Sr)rrr)rBrRrSrrr leave_Params  zTemplateTransformer.leave_ParamcCs~t|jdkr=|jtjjkr=t|jdkr=|jdur=t|jdkr=|j tjjkr=t |jdj j }||j vr=|j |S|SNrr)lenparamsstar_argr% MaybeSentinelDEFAULT kwonly_params star_kwargposonly_params posonly_indrrr)rCrRrSrrrleave_Parameterss  z$TemplateTransformer.leave_ParameterscC8|j}t|tjrt|j}||jvr|j|S|Sr)r)r$r%rXrrDrR)rMrPrQargrTrrr leave_Args    zTemplateTransformer.leave_ArgcC\t|jdkr,|jd}t|tjr,|j}t|tjr,t|j}||jvr,|j| S|Sr^) r_bodyr$r%Exprr)rXrrFrRrMrPrQ body_node name_noderTrrrleave_SimpleStatementLine     z-TemplateTransformer.leave_SimpleStatementLinecCrir)r)r$r%rXrrErR)rMrPrQrqrTrrr leave_Exprs    zTemplateTransformer.leave_ExprcCrlr^) r_rmr$r%rnr)rXrrGrRrorrrleave_SimpleStatementSuitersz.TemplateTransformer.leave_SimpleStatementSuitecCst|jdkr>|jd}t|tjr>t|jdkr>|jd}t|tjr>|j}t|tjr>t|j}||j vr>|j | S|Sr^) r_rmr$r%SimpleStatementLinernr)rXrrGrR)rMrPrQstatement_noderprqrTrrrleave_IndentedBlocks       z'TemplateTransformer.leave_IndentedBlockcCrir)r)r$r%rXrrIrR)rMrPrQexprrTrrr leave_Index.rZzTemplateTransformer.leave_IndexcCsJ|j}t|tjr#|j}t|tjr#t|j}||jvr#|j|S|Sr) slicer$r%Indexr)rXrrHrR)rMrPrQindexryrTrrrleave_SubscriptElement;s    z*TemplateTransformer.leave_SubscriptElementcCrVr) decoratorr$r%rXrr)rJrR)rMrPrQrrTrrrleave_DecoratorNs    z#TemplateTransformer.leave_Decorator)'__name__ __module__ __qualname__rstrValidReplacementTyperOr%rXr&rUr,rYr-r\r.r]r/rhr0rkrvr2rrrnr1rtSimpleStatementSuiter3ru IndentedBlockrxr|r5rzr4r~r6rrrrrr!<s  X                 r!c@s6eZdZdeeddfddZdejddfddZdS) TemplateCheckerrr NcCs ||_dSr)r)rMrrrrrO[s zTemplateChecker.__init__nodecCs.|jD]}|jt|krtd|dqdS)NzTemplate variable "z" was not replaced properly)rr)rr)rMrr rrr visit_Name^s zTemplateChecker.visit_Name) rrrrrrOr%rXrrrrrrZsrtreer"cCst|}t||tjSr)r!rvisitr%CSTNode)rr" unmanglerrrrunmangle_nodesdsr_DEFAULT_PARTIAL_PARSER_CONFIGconfigcKNt|dd|D}t||}tt||tj}|tdd|D|S)a Accepts an entire python module template, including all leading and trailing whitespace. Any :class:`~libcst.CSTNode` provided as a keyword argument to this function will be inserted into the template at the appropriate location similar to an f-string expansion. For example:: module = parse_template_module("from {mod} import Foo\n", mod=Name("bar")) The above code will parse to a module containing a single :class:`~libcst.FromImport` statement, referencing module ``bar`` and importing object ``Foo`` from it. Remember that if you are parsing a template as part of a substitution inside a transform, its considered :ref:`best practice ` to pass in a ``config`` from the current module under transformation. Note that unlike :func:`~libcst.parse_module`, this function does not support bytes as an input. This is due to the fact that it is processed as a template before parsing as a module. cSh|]}|qSrrr8rrrr=r:z(parse_template_module..cSrrrr8rrrr=r:)r r% parse_modulerrModulerr)rrr"sourcemodule new_modulerrrparse_template_moduleos  rcKslt|dd|D}t||}t||}t|tjtjfs(td|jj d| t dd|D|S)a< Accepts a statement template followed by a trailing newline. If a trailing newline is not provided, one will be added. Any :class:`~libcst.CSTNode` provided as a keyword argument to this function will be inserted into the template at the appropriate location similar to an f-string expansion. For example:: statement = parse_template_statement("assert x > 0, {msg}", msg=SimpleString('"Uh oh!"')) The above code will parse to an assert statement checking that some variable ``x`` is greater than zero, or providing the assert message ``"Uh oh!"``. Remember that if you are parsing a template as part of a substitution inside a transform, its considered :ref:`best practice ` to pass in a ``config`` from the current module under transformation. cSrrrr8rrrr=r:z+parse_template_statement..zExpected a statement but got a !cSrrrr8rrrr=r:) r r%parse_statementrr$rvBaseCompoundStatementr __class__rrr)rrr"r statement new_statementrrrparse_template_statements   rcKr)u& Accepts an expression template on a single line. Leading and trailing whitespace is not valid (there’s nowhere to store it on the expression node). Any :class:`~libcst.CSTNode` provided as a keyword argument to this function will be inserted into the template at the appropriate location similar to an f-string expansion. For example:: expression = parse_template_expression("x + {foo}", foo=Name("y"))) The above code will parse to a :class:`~libcst.BinaryOperation` expression adding two names (``x`` and ``y``) together. Remember that if you are parsing a template as part of a substitution inside a transform, its considered :ref:`best practice ` to pass in a ``config`` from the current module under transformation. cSrrrr8rrrr=r:z,parse_template_expression..cSrrrr8rrrr=r:)r r%parse_expressionrrr&rr)rrr"r expressionnew_expressionrrrparse_template_expressions  r),typingrrrrrlibcstr%libcst.helpers.commonrr r__annotations__r r&r,r-r.r/r0r2r1r3r5r4r6rrrr CSTTransformerr! CSTVisitorrrrPartialParserConfigrrrrvrrrrrrrs         " %