Mh;FddlZddlZddlZddlmZmZmZddlmZddl m Z ddl m Z m Z mZmZmZmZmZmZddlmZddlmZmZmZmZmZmZmZmZGdd Zd ej d e d e!fd Z"dee ej e fd ee eej e ffdZ#dS)N)ChainMap OrderedDictdeque)suppress) FrameType)AnyTupleIterableListMappingDictUnionSet)getattr_static) CannotEval has_ast_namecopy_ast_without_contextis_standard_typesof_standard_typesis_anyof_type ensure_dictceZdZdeeeffdZededdfdZ de j defdZ de j defdZ d Zd Zd Zd Zde jfd ZdZdee je je je jfdeeeeeffdZde jdeee j effdZde jdeeee j effdZdS) Evaluatornamesc"||_i|_dS)z Construct a new evaluator with the given variable names. This is a low level API, typically you will use `Evaluator.from_frame(frame)`. :param names: a mapping from variable names to their values. N)r_cache)selfrs N/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pure_eval/core.py__init__zEvaluator.__init__s  framereturnc |tt|jt|jt|jS)z Construct an Evaluator that can look up variables from the given frame. :param frame: a frame object, e.g. from a traceback or `inspect.currentframe().f_back`. )rrf_locals f_globals f_builtins)clsr"s r from_framezEvaluator.from_frame"sLs8  ' '  ( ( ( ) )     r!nodect|tjs4tdt |jtt5|j |}|turt|cdddS#1swxYwY | |x|j |<}|S#t$rt|j |<wxYw)z Find the value of the given node. If it cannot be evaluated safely, this raises `CannotEval`. The result is cached either way. :param node: an AST expression to evaluate :return: the value of the node z$node should be an ast.expr, not {!r}N) isinstanceastexpr TypeErrorformattype__name__rKeyErrorrr_handle)rr*results r __getitem__zEvaluator.__getitem__0s$)) `BII$t**J]^^__ _ h    [&F##                    )-d);); ;DK M    *DK   s#BBB B<<Ccftt5tj|cdddS#1swxYwYt |tjr( |j|jS#t$rtwxYwt |tj r$||j }|j }t||St |tjr||St |tjtjtjtjfr||St |tjr||St |tjr||St |tjr||St |tjr||St |tjr||St)z This is where the evaluation happens. Users should use `__getitem__`, i.e. `evaluator[node]`, as it provides caching. :param node: an AST expression to evaluate :return: the value of the node N)r Exceptionr- literal_evalr,Nameridr3r Attributevalueattrr Subscript_handle_subscriptr r rr _handle_containerUnaryOp _handle_unaryBinOp _handle_binopBoolOp_handle_boolopCompare_handle_compareCall _handle_call)rr*r=r>s rr4zEvaluator._handleKs)i  * *#D)) * * * * * * * * * * * * * * * * dCH % % + !z$'** ! ! !   ! cm , , +$E9D!%.. . cm , , +))$// / sxCGSXF G G +))$// / ck * * +%%d++ + ci ( ( +%%d++ + cj ) ) +&&t,, , ck * * +''-- - ch ' ' +$$T** *s6::A..Bc|jrt|j}fd|jD}t |t t tttttttttt t"t$t&t(st'|dkr9t |t*t,t.t0t2t4t6s@t'|dkrYt |t.t8t2t4t:r,d|D} ||S#t<$r }t|d}~wwxYwt'|dkrc|d}t |t>t@r& ||S#t<$r }t|d}~wwxYwt |tBtDtFrvtI|tt0tt*t,tJtL|D]}tO|dd ||S#t<$r }t|d}~wwxYwt |tPtRtTtVt*t,tXt.tZt0 r8tO|dd ||S#t<$r }t|d}~wwxYwt) Nc g|] }| SrN).0argrs r z*Evaluator._handle_call..us///cS ///r!rc2g|]}t|ddS)Fcheck_dict_valuesdeep)r)rOrPs rrQz*Evaluator._handle_call..s7"#UKKKr!FrTT).keywordsrfuncargsrsliceintrangeroundcomplexlisttupleabshexbinoctboolordfloatlenchrsetdictstr frozensetbytes bytearrayobjectdivmodpowr8r;r1allanysumrrrrsortedminmaxhashasciirepr)rr*rYrZerPxs` rrKzEvaluator._handle_callqs =  DI////TY/// #   ((4yyA~~tS$Yy&QQ4yyA~~tS&%C@@D (tT{" ( ( ( a' ( t99>>q'CdB%% ,,499$ ,,,$!+,dCc** ,UItS$ USSSNNA%a5uMMMMM,499$ ,,,$!+,fc3c4T9 ,"#DIIII,499$ ,,,$!+,s`8D== EEE F F1$F,,F1& H11 I;II) J44 K >KK c||j}d}t|j|jD]]\}}||}t |}t jtjt j tj t j tj t j tjt jtjt jtjt jtjt jtjt jdt jdi |}|t jt jfvr$t3|ddt3|dd |||}n#t4$r }t6|d}~wwxYw|s|cS|}_|S)NTc ||vSNrNabs rz+Evaluator._handle_compare..s a1fr!c ||vSrrNrs rrz+Evaluator._handle_compare..s !r!FrT)leftzipops comparatorsr1r-EqoperatoreqNotEqneLtltLtEleGtgtGtEgeIsis_IsNotis_notInNotInrr8r) rr*rr5oprightop_typeop_funcr}s rrIzEvaluator._handle_comparesSDITXt'788  IBKE2hhG  8;    8?,, 33  Gsvsy111!$%dKKKK!%5tLLLL ( u-- ( ( ( a' (  DD s E%% E</E77E<cZt||jddd}|jddD]w}t|jtjr|pt||dd}1G99G>cpt||jdd}t|j}tjt jtjt j tj t j tj t j i|} ||S#t$r }t|d}~wwxYwr)roperandr1rr-USubrnegUAddposNotnot_Invertinvertr8r)rr*r=rrr}s rrCzEvaluator._handle_unary s!  %e   tw-- Hhl Hhl GX] J     $2e99  $ $ $! # $s B B5(B00B5c |j}t|dtt|tt |j}t|tj r(t fd|j |j |j fD}nJt|tj rtt|tjr|j}|}t|dd ||S#t $rtwxYw)NFrTc(g|]}|dn|SrrN)rOprs rrQz/Evaluator._handle_subscript.."s5IDD47r!T)r=rrr1rlrr[r,r-SlicelowerupperstepExtSlicerIndexr8)rr*r=indexs` rr@zEvaluator._handle_subscripts TZ  UU T;1W1W      eSY ' ' #k5; CEE s| , ,  %++ $ KE%5tDDDD <      s -C55Dct|tjr|j}d|vrtn|j}fd|D}t|tjr|St|tjrt|Std|Dstt|tj r% t|S#t$rtwxYwt|tjsJfdt||jD} t|S#t$rtwxYw)z;Handle container nodes, including List, Set, Tuple and DictNc g|] }| SrNrN)rOeltrs rrQz/Evaluator._handle_container..?s***cS ***r!c3:K|]}t|ddVdS)FTrTN)r)rOrs r z.Evaluator._handle_container..FsC  KN cU F F F      r!c*g|]\}}||fSrNrN)rOrvalrs rrQz/Evaluator._handle_container..Ss&IIIhc3#tCy!IIIr!)r,r-r keysreltsr r rartrrkr/rrrl)rr*rpairss` rrAzEvaluator._handle_container4s~ dCH % % 9Dt||  9D****T*** dCH % % K dCI & & ;;   RV        dCG $ $ ! !4yy  ! ! !   !$)))))IIII#dDK2H2HIII ;;      sCC%&D55Erootc#Ktj|D]=}t|tjs ||}n#t$rY3wxYw||fV>dS)a5 Find all expressions in the given tree that can be safely evaluated. This is a low level API, typically you will use `interesting_expressions_grouped`. :param root: any AST node :return: generator of pairs (tuples) of expression nodes and their corresponding values. N)r-walkr,r.r)rrr*r=s rfind_expressionszEvaluator.find_expressionsYsHTNN  DdCH--  T     +      s> A  A cZtd||DS)aP Find all interesting expressions in the given tree that can be safely evaluated, grouping equivalent nodes together. For more control and details, see: - Evaluator.find_expressions - is_expression_interesting - group_expressions :param root: any AST node :return: A list of pairs (tuples) containing: - A list of equivalent AST expressions - The value of the first expression node (which should be the same for all nodes, unless threads are involved) c3,K|]}t| |VdSr)is_expression_interesting)rOpairs rrz.~sA! ! ($/! ! ! ! ! ! ! r!)group_expressionsr)rrs rinteresting_expressions_groupedz)Evaluator.interesting_expressions_groupedms@"!! ! --d33! ! !    r!N)r2 __module__ __qualname__r rmrr classmethodrr)r-r.r6r4rKrIrGrErBrCr@rr r rr rAASTr rrrNr!rrrs gc3h/      y  [    [  S6$CH$$$$$L@@@D!!!F&$$$B$#+$$$$ 4##)SWch>?# tUC% &####JSW%# :N1O( CG U4PSPX>[^K^E_@`      r!rr*r=r#c8tt5tj| ddddS#1swxYwYt ||rdSt |tjr+tt|j t|urdSdS)an Determines if an expression is potentially interesting, at least in my opinion. Returns False for the following expressions whose value is generally obvious: - Literals (e.g. 123, 'abc', [1, 2, 3], {'a': (), 'b': ([1, 2], [3])}) - Variables or attributes whose name is equal to the value's __name__. For example, a function `def foo(): ...` is not interesting when referred to as `foo` as it usually would, but `bar` can be interesting if `bar is foo`. Similarly the method `self.foo` is not interesting. - Builtins (e.g. `len`) referred to by their usual name. This is a low level API, typically you will use `interesting_expressions_grouped`. :param node: an AST expression :param value: the value of the node :return: a boolean: True if the expression is interesting, False otherwise NFT) r ValueErrorr-r9rr,r:getattrbuiltinsr;rq)r*r=s rrrs$ *    E4  u tSX & &$'68844==u 4s 8<< expressionsci}|D]W\}}tjt|}||g|fd|Xt |S)a Organise expression nodes and their values such that equivalent nodes are together. Two nodes are considered equivalent if they have the same structure, ignoring context (Load, Store, or Delete) and location (lineno, col_offset). For example, this will group together the same variable name mentioned multiple times in an expression. This will not check the values of the nodes. Equivalent nodes should have the same values, unless threads are involved. This is a low level API, typically you will use `interesting_expressions_grouped`. :param expressions: pairs of AST expressions and their values, as obtained from `Evaluator.find_expressions`, or `(node, evaluator[node])`. :return: A list of pairs (tuples) containing: - A list of equivalent AST expressions - The value of the first expression node (which should be the same for all nodes, unless threads are involved) r)r-dumpr setdefaultappendr`r)rr5r*r=rs rrrsz(F"== ex06677$U ,,Q/66t<<<<    r!)$r-rr collectionsrrr contextlibrtypesrtypingrr r r r r rrpure_eval.my_getattr_staticrpure_eval.utilsrrrrrrrrrr.rfrrrNr!rrs 4444444444HHHHHHHHHHHHHHHHHHHH666666                    l l l l l l l l ^ !CH!S!T!!!!H!8E#(C-,@#A!d5QUVYV^Q_adQdKeFf!!!!!!r!