Xh| UddlmZddlmZmZmZmZddlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZddlZddlZddlZddlZddlZddlZddlmZddlmZmZddl m!Z!m"Z"ddl#m$Z$dd lm%Z%m&Z&ejNd krdd l(m)Z)ndd lm)Z)e$Gd d eZ*e$GddeZ+e$GddeZ,e$GddeZ-ee,e-fZ.de dee dffdZ/e$eGddZ0de1dee1fdZ2de1dee1de1fdZ3dZ4de5e6e1d fe1zde5e6e1d ffd!Z7e$eGd"d#e0Z8Gd$d%e Z9e d&Z:Gd'd(e Z;Gd)d*Z<e<Z=d+Z>eZ?e@ZAGd,d-eBZCd.e1d/e;fd0ZDejd1ejd2ejd3ejd4ejd5ejd6ejd7ejd8ejd9ejd:ejd;ejd<ejd?ejd@ejdAejdBejdCejdDejdEiZ]eSeTeje6e1d ffeUdF<ejdGejdHejdIejdJiZceSeTeje6e1d ffeUdK<GdLdMZeGdNdOZfdee6e1d fdffdPZgd/e;de0fdQZhdRe:dSeSdeifdTZjdUeejdfd/e;fdVZlde dUejd/e;fdWZndXede dUejd/e;fdYZodZe1d/e;fd[ZpdUejd/e;fd\Zqd]Zrhd^ZseSe1eteue6ejejejejejejeje9er>7sr,r>funcr.ct|dd}t|}t|dd}t|dd}||r|r|r||vr t||Sy)z}Get unbound method for given bound method. Returns None if cannot get unbound method, or method is already unbound. __self__Nr/__dict__)getattrtype)r?owner owner_classnameinstance_dict_overridess r)_unbind_methodrI@sa D*d +Eu+K 4T *D%eZ>  ''D8O,O{D)) r,ceZdZUdZdZeed<dZeed<dZeed<dZ eed<dZ eed<dZ eed<dZ eed <dZ eed <ee Zeeed <d ZdZddeedffdZdZy)EvaluationPolicyz Definition of evaluation policy.Fallow_locals_accessallow_globals_accessallow_item_accessallow_attr_accessallow_builtins_accessallow_all_operationsallow_any_callsallow_auto_importdefault_factory allowed_callsc|jSr$)rN)r'valueitems r) can_get_itemzEvaluationPolicy.can_get_itemd%%%r,c|jSr$)rO)r'rXattrs r) can_get_attrzEvaluationPolicy.can_get_attrgr[r,Ndunders.c|jryyNT)rQ)r'r_abs r) can_operatezEvaluationPolicy.can_operatejs  $ $ %r,cv|jry||jvryt|}|r||jvryyyra)rRrVrI)r'r? owner_methods r)can_callzEvaluationPolicy.can_callnsC    4%% %%d+ LD,>,>>?KE{d222.F A) k!%2; %LN.5nd.K+&)@@@  A !.1! !sC''C:9C:ctt|j}t||j|j|dS)zTAllow accessing `__getiitem__` of allow-listed instances unless it was not modified.r*r)rrrr_getitem_methods)r'rXrYrs r)rZzSelectivePolicy.can_get_item!s;#9$:W:W#X # .. 115%   r,Nr_ct|j}|g}||j|t|Dcgc]2}|D]+}t ||j |j |||-4c}}Scc}}w)Nr)rrappendallrr_operator_dunder_methods)r'r_rbrcrobjectsdunderobjs r)rdzSelectivePolicy.can_operate,s&<  , ,' ## = NN1 & " %"&"9"9$($A$A&$I%@ &      s7A5 rr.c||jvr)|j|j||j|<|j|Sr$)r_safe_get_methodsr)r'rs r)rz(SelectivePolicy._operator_dunder_methodsAsI 66 6484J4J''5D ) )& 1,,V44r,c:|j|jdS)Nr*)rrr's r)rz SelectivePolicy._getitem_methodsH%%d&:&:MJJr,c:|j|jdS)Nr<rrrs r)rz SelectivePolicy._getattr_methodsLrr,c:|j|jdS)Nr|rrs r)rz%SelectivePolicy._getattribute_methodsPs%%d&:&:CTW>XU38_s%:!;X9>5:d3H #566p   5c? *5s5s8}5K#h-KKK#h-KKPs8}PP #h- r,rceZdZdZy)_DummyNamedTuplezK$s)Z/047>dr,rceZdZdZdZy)_IdentitySubscriptzQ&RS[(D'//7>>::4f-J Z ))r,)__add__)__sub__)__mul__) __truediv__) __floordiv__)__mod__)__pow__) __lshift__) __rshift__)__or__)__xor__)__and__) __matmul__BINARY_OP_DUNDERS)__eq__)__ne__r)__lt____gt__)__le____ge__)rr)rr) __contains__COMP_OP_DUNDERS)__neg__)__pos__) __invert____inv__)__not__UNARY_OP_DUNDERSceZdZdZy)ImpersonatingDuckzXA dummy class used to create objects of other classes without calling their ``__init__``Nrr%r,r)rrsbr,rcveZdZdZd deedeefdZdefdZdefdZ fdZ d efd Z d efd Z d Z xZS)_DuckzHA dummy class used to create objects pretending to have given attributes attributesitemsc0|xsi|_|xsi|_yr$rr)r'rrs r)__init__z_Duck.__init__s$*[b r,r]c |j|Sr$rr'r]s r)r<z_Duck.__getattr__st$$r,c||jvSr$r r s r) __hasattr__z_Duck.__hasattr__st&&r,c>gtt|jSr$)dirsuperr)r' __class__s r)__dir__z _Duck.__dir__s.U.doo..r,r(c |j|Sr$rr&s r)r*z_Duck.__getitem__zz#r,c |j|Sr$rr&s r) __hasitem__z_Duck.__hasitem__rr,c6|jjSr$)rkeysrs r)_ipython_key_completions_z_Duck._ipython_key_completions_szz  r,)NN)r/r0r1rhrrr rmr<rrr*rr __classcell__)rs@r)rrs[R!8D>!$!%%''/ss!r,rcXd}|jD]\}}t||s|}|Sr$)rr})node_opr_ropcandidate_dunders r) _find_dunderr!s8 F ' &  gr "%F& Mr,ctt|j}|jj D]\}}t ||st |||!|Sr$)rEVALUATION_POLICIESrrrrsetattr)rpolicyr(rXs r) get_policyr&sW %g&8&89 :F..446( U 63  FC '( Mr, policy_namerct|}d}|jD]0\}}t||rtjd|d|dd}2|S)NTz Override z is not valid with z evaluation policyF)r#rrwarningswarn)r'rr%all_goodr(rXs r)_validate_policy_overridesr,sg! -FH&,,. Uvs# MMC7"5k_DVW H  Or,nodec (t}|yt|tjrt |j St|tj rt |j}t |j}t|jt}|rP|j|||rt||d|Std|dt|dj dt|tj"r`t |j}d}d}t%|j&|j(D] \}}t |}d} t|t*}|smt|tj,rt*tj.}d}t|tj0rd } t|tj2rd } d}| s|r|d} | ru| d k(r||fn||f\} } | d k(s|r6|j|| | r#tt4| | | } |r| } | sd}|}td | dt|dj dt7d | d |St|tj8r |j:St|tj<rt?fd|j@DSt|tjBr$|j@D cgc]} t | c} St|tjDr$|j@D chc]} t | c} St|tjFrYtIt%|jJDcgc]}t |c}|jLDcgc]}t |c}St|tjNrItQt |jRt |jTt |jVSt|tjXrt |jZ}t|jt\}|rO|j||r t||dStd|dt|dj dt|tjbrgt |j:}t |jP}|je||r||Stdt|dj dt|tjfrti|jjSt|tjlrlt |j:}|jo||jprt||jpStdt|dj dt|tjrrDt |jt}|rt |j St |jvSt|tjxrt |jz}|j}|r4|j~s(|jDcgc]}t |}}||St|r,t|j|}|tur|St|St||}|tur|Std|dj dt7dtj|cc} wcc} wcc}wcc}w#t^$r%tad|jdt|wxYwcc}w)aEvaluate AST node in provided context. Applies evaluation restrictions defined in the context. Currently does not support evaluation of functions with keyword arguments. Does not evaluate actions that always have side effects: - class definitions (``class sth: ...``) - function definitions (``def sth: ...``) - variable assignments (``x = 1``) - augmented assignments (``x += 1``) - deletions (``del x``) Does not evaluate operations which do not return values: - assertions (``assert x``) - pass (``pass``) - imports (``import x``) - control flow: - conditionals (``if x:``) except for ternary IfExp (``a if x else b``) - loops (``for`` and ``while``) - exception handling The purpose of this function is to guard against unwanted side-effects; it does not give guarantees on protection from malicious code execution. Nrz Operation (`z`) forznot allowed in  modeTFis_rz Comparison (`z Comparison `z` not supportedc36K|]}t|ywr$)r).0ers r) zeval_node..gs>qYq'*>szbad operand type for unary z: z$Subscript access (`__getitem__`) forz not allowed in z$Attribute access (`__getattr__`) forzCall forzUnhandled node)Gr&r}r ExpressionrbodyBinOpleftrightr!rrrdrCrrDrComparezipops comparatorsrNotInInIsIsNotoperator ValueErrorConstantrXTuplerleltsListSetDictrrvaluesSliceslicelowerupperstepUnaryOpoperandrru TypeError SubscriptrZName_eval_node_nameid Attributer^r]IfExptestorelseCallr?rgkeywordsr6r_eval_return_type__new__ NOT_EVALUATED_create_duck_for_heap_typedump)r-rr%r8r9r_all_truenegaterrrbrcresultr3kvrXslice_rYr?argr6overridden_return_type return_types ` r)rrs6 F |$'G,,$ "G,$**g.tww(9: !!'470wtWQZ077$"7)62J%g&8&8%9? $ $G,TXXt'7'78! IBeW-EF"27Gb#)),-cff5G!Fb#&&)"Fb#)),"F!Fg (..(@t}tUm1U?g&2D2DWaQR2S6WXv6q!DII>>>$!/3yy9! !W%99$ /3yy9! !W%99$! 04 :11g&:04 <11g&<   $ " djj' * djj' * dii )  $ $$,,0tww(89 !!'51575'!*577 %"7)62K%g&8&8%9? $ &$**g.4::w/   uf -=  2 Kw112% 8  $!tww00$ &$**g.   udii 05$)), , 2 Kg001 7  $ "G, TYY0 0T[['2 2$!G, ??4 7;yyAIc7+ADA;  4= &7t||T7%S "%]:---d3 3+D$@K-/""  g001 7  %sxx~ 66g:9;< &#5dggYbe NPBs*3] 1]9] ]<](^.^ c t|}|jtj u}|rt |j||||StS#t$r t}YNwxYw)zyEvaluate return type of a given callable function. Returns the built-in type, a duck or NOT_EVALUATED sentinel. )rrCUNKNOWN_SIGNATUREreturn_annotationrempty_resolve_annotationr_)r?r-rsig not_emptys r)r]r]s^  o %%Y__rt;|j@||St;|||Scc} } wcc}}w) zKResolve annotation created by user with `typing` module and custom objects.rArNFr r _is_protocol)!r}rmrUrrrrAr rlenrr enumerate parametersr annotationr6rrrrrorrfromkeysrrjrr _eval_or_create_duck __supertype__r __value__)ryrpr?r-rorigin type_argsindexir(rXtype_argr]rrerfs r)roros j# &  G,   #FTgdJ7}} 7 Z( y>Q Q<   } $ v ()=)=)?@ OA|U6)   TYY!7TYYu-w7 7"8  9  5%Z0 /#tT7ST      j 9:: j !}}S[1'66<<>Aq&q#tT7CC   ^ , c*o >?? 9 J'*"8S$gFF J (#J$<$>1 s &J 6J node_idct|}|jr||jvr|j|S|jr||jvr|j|S|j r t t|rtt|S|jr|jr|j|S|js%|jstd|jdt|d)Nz Namespace access not allowed in r/z+ not found in locals, globals, nor builtins)r&rLrrMrrPrbuiltinsrCrSrrr NameError)rrr%s r)rUrU s  F !!g&?~~g&& ""w'//'Aw'' ##'(Bx)) G$7$7""7++  & &v/I/I.w/A/A.B% H  7)#NOPPr,ct|}|j|r4|js(|jDcgc]}t ||}}||St |Scc}wr$)r&rgr\r6rr`) duck_typer-rr%rhr6s r)r{r{sW  F y!$--3799=C #w'==$ %i 00>sAcRt} ||_|S#t$rYtSwxYw)zCreate an imitation of an object of a given type (a duck). Returns the duck or NOT_EVALUATED sentinel if duck could not be created. )rrrRr_)rducks r)r`r`)s5  D "    s  &&>numpyvoidrndarraypandascoreindexing _LocIndexerrrr _iLocIndexerrSeriesr DataFrameBUILTIN_GETITEMcZ|r|n t|Dcgc]}t||c}Scc}w)z=For use on immutable objects or with methods returning a copy)rrC)clssourceres r) _list_methodsrTs$06fCH FGCO FF Fs()rrrJr)rrcount dict_keysBUILTIN_GETATTRTF)rPrLrMrNrOrVrRrQrr) rrrrrrPrLrMrV)rPrLrMrOrNrRrQ)rrr)rrrrrIr$)rinspectrrrrtypingrr r r r r rrrrrrrrrr collectionsrBrrr) functoolsr dataclassesrrtypesrrIPython.utils.decoratorsrrr version_infotyping_extensionsr r"r4r:r>rrIrKrmryrrrkrlrrrrrrrrrlobjectr_ ExceptionrrAddSubMultDivFloorDivModPowLShiftRShiftBitOrBitXorBitAndMatMultrrrDrjEqNotEqLtLtEGtGtEr?rcmpopUSubUAddInvertNotrunaryoprrr!r&rir,ASTrr[r]rorUr{r`SUPPORTED_EXTERNAL_GETITEMbytesr defaultdictdeque OrderedDictChainMapUserDictUserList UserStringrrdict_non_mutating_methodslist_non_mutating_methodsr frozensetset_non_mutating_methodsrrabcKeysViewintfloatcomplexNUMERICS isdisjointrangeCounterelements most_common ALLOWED_CALLSrBUILTIN_OPERATIONSr#__all__) numeric_clsrs00r)rs<<   %(2*'g/$,,,;8;;,,,   z#556eHdN&;* !! !Hs#.  #  F4E#s(Oc)*sCx  &   DGzGWX   ()+K  Y !*s!*%6!*JGG\GG\HHlGG LL#GG\GG\JJJJII{JJ JJ KK;4S\\*E#J67"FFKII#FF GG !FF GG !FF  ;d399ouS#X67 HHlHHlJJ)GG\ >$tCKK(%S/9:cc !!4eE#s(OT,A&B).> % 9=   m7E#''4-(m73Dm7`HCHH?P"9? 9? 9? (( 9?  9?xQSQ+<Q&1#((1=N1    .)*$G @6s3x=3s9~+>>-1O 4 (( );  # #5# #423 #  #  #423##301##9# ##3# # 5!#"##$!)R}[7QRVfRfR%#&'#(;$$&?@)#*+#,;**,EF-#./#0;**,EF1#23#4;'')BC5#67#8;'')BC9#:;#<;))3s84=#>?#@;&&(ABA#B  C#D##E# J ( ( ( ( (  (  ( ( ( (^$ (' "!"e" '!;' $ " ." !#" !!5#L QSs;Y