UgdZddlZddlZddlZddlZddlZGddej ZGddejZ Gdde Z d Z d Z d Z dd Zd ZdZdZy)zJ Utilities for manipulating the Abstract Syntax Tree of Python constructs Nc4eZdZdZfdZdZdZdZxZS) NameVisitorzQ NodeVisitor that builds a set of all of the named identifiers in an AST cBt||i|t|_yN)super__init__setnames)selfargskwargs __class__s 1lib/python3.12/site-packages/datashader/macros.pyrzNameVisitor.__init__s $)&)U cN|jj|jyr)r addidr nodes r visit_NamezNameVisitor.visit_Names twwrct|dr&|jj|jyt|dr&|jj|jyy)Nargr)hasattrr rrrrs r visit_argzNameVisitor.visit_argsC 4  JJNN488 $ T4 JJNN477 #!rc Ltjd}|jDcgc]}|j|s|}}|r)t |Dcgc]}t |ddc}dz}nd}t |||zDcgc]}dt|zc}Scc}wcc}wcc}w)aD Returns a list of new names that are not already present in the AST. New names will have the form _N, for N a non-negative integer. If the AST has no existing identifiers of this form, then the returned names will start at 0 ('_0', '_1', '_2'). If the AST already has identifiers of this form, then the names returned will not include the existing identifiers. Parameters ---------- num_names: int The number of new names to return Returns ------- list of str z^_(\d+)$Nr_)recompiler matchmaxintrangestr)r num_namesprop_renmatching_names start_numbers r get_new_nameszNameVisitor.get_new_namess&**[)%)ZZDZ7==3C!ZD NCNqAabE NCDqHLLlL9$<=?="#c!f =? ? EC?sBB B?B!) __name__ __module__ __qualname____doc__rrrr* __classcell__rs@rrr s $ ?rrc"eZdZdZfdZxZS)ExpandVarargTransformerz Node transformer that replaces the starred use of a variable in an AST with a collection of unstarred named variables. c@t||i|||_||_y)a Parameters ---------- starred_name: str The name of the starred variable to replace expand_names: list of stf List of the new names that should be used to replace the starred variable N)rr starred_name expand_names)r r4r5r r rs rrz ExpandVarargTransformer.__init__?s' $)&)((r)r+r,r-r.rr/r0s@rr2r2:s ) )rr2ceZdZdZy)ExpandVarargTransformerStarredc|jj|jk(r9|jDcgc]#}t j ||j %c}S|Scc}w)N)rctx)valuerr4r5astNamer9)r rnames r visit_Starredz,ExpandVarargTransformerStarred.visit_StarredQsY ::==D-- -%%'%9=CHH$((3%' 'K's(AN)r+r,r-r>rrr7r7Osrr7ctjtj|}t j |}|S)z2 Get the AST representation of a function )textwrapdedentinspect getsourcer;parse)fn fn_sourcefn_asts rfunction_to_astrIYs4  1 1" 56IYYy !F Mrc,ddl}|j|S)z9Convert AST to source code string using the astor packagerN)astor to_source)r;rKs r ast_to_sourcerMjs ??3 rct|tjsJ|jd}t|tjsJt |d|j zdS)zO Compile function AST into a code object suitable for use in eval/exec rz<%s>exec)mode) isinstancer;Modulebody FunctionDefrr=)rH fndef_asts rcompile_function_astrVpsQ fcjj )) ) AI i 11 1 66INN2 @@rct|tjsJ|jd}t|tjsJt |}t j}|}t|D]}|j}|j}|j}~tj|} | j|t|| | |jS)Nr)rQr;rRrSrTrVrC currentframer#f_backf_locals f_globalscopyupdateevalr=) rH stacklevelrUcode current_frame eval_framer eval_locals eval_globalsscopes rfunction_ast_to_functionrfzs fcjj )) ) AI i 11 1  'D((*MJ : && %%K''L IIl #E LL u   rc.tj|S)N)r)r;rr=s r _build_argris 77t rcft|tjsJtj|}|j d}t|tj sJ|j}|j}|s td|sJt|tr|}n |j}t}|j||j|}t}|||j|} | j d} | jjj!|D cgc]} t#| c} d| j_t} | j| || j$vrtdj'|g|_tj*| | Scc} w)a Given a function AST that use a variable length positional argument (e.g. *args), return a function that replaces the use of this argument with one or more fixed arguments. To be supported, a function must have a starred argument in the function signature, and it may only use this argument in starred form as the input to other functions. For example, suppose expand_number is 3 and fn_ast is an AST representing this function... def my_fn1(a, b, *args): print(a, b) other_fn(a, b, *args) Then this function will return the AST of a function equivalent to... def my_fn1(a, b, _0, _1, _2): print(a, b) other_fn(a, b, _0, _1, _2) If the input function uses `args` for anything other than passing it to other functions in starred form, an error will be raised. Parameters ---------- fn_ast: ast.FunctionDef expand_number: int Returns ------- ast.FunctionDef rzmInput function AST does not have a variable length positional argument (e.g. *args) in the function signaturerhNzNThe variable length positional argument {n} is used in an unsupported context )r')rQr;rRr\deepcopyrSrTr vararg ValueErrorr$rrvisitr*r7extendrir formatdecorator_listfix_missing_locations) rH expand_numberrUfn_args fn_vararg vararg_namebefore_name_visitorr5expand_transformer new_fn_ast new_fndef_astr=after_name_visitors rexpand_function_ast_varargsr|sF fcjj )) )]]6 "F AI i 11 1nnGI *+ + 9)S! mm &-f%&44]CL8#\ eFmOOA&M""+78<4 <8!%M%Z((... F[F  "Ij) ) 9s!F.cRttrdkr tdfd}|S)aX Decorator to expand the variable length (starred) argument in a function signature with a fixed number of arguments. Parameters ---------- expand_number: int The number of fixed arguments that should replace the variable length argument Returns ------- function Decorator Function rz,expand_number must be a non-negative integercLt|}t|}t|dS)N)r_)rIr|rf)rFrHfn_expanded_astrss r_expand_varargsz'expand_varargs.._expand_varargss& $5fmL'AFFr)rQr"rm)rsrs` rexpand_varargsrs0 mS )]Q->GHHG r)r)r.rr\rCr;rA NodeVisitorrNodeTransformerr2r7rIrMrVrfrir|rr?rrrsq  ,?#//,?^)c11)*%<" A!:cLr