X-PhUdZddlmZddlmZmZddlmZmZm Z m Z m Z m Z m Z mZmZeeeeeefZeeeeefZded<dd ZddZddZddZd dZdS)!zTConstant folding of expressions. For example, 3 + 5 can be constant folded into 8. ) annotations)FinalUnion) ComplexExpr Expression FloatExprIntExprNameExprOpExprStrExpr UnaryExprVarr CONST_TYPESexprr cur_mod_idstrreturnConstantValue | NonecHt|tr|jSt|tr|jSt|tr|jSt|t r|jSt|t r|jdkrdS|jdkrdS|j}t|trJ|j rC|j ddd|kr|j }t|tr|Snt|trEt!|j|}t!|j|}||t'|j||SnAt|t*r,t!|j|}|t/|j|SdS) aReturn the constant value of an expression for supported operations. Among other things, support int arithmetic and string concatenation. For example, the expression 3 + 5 has the constant value 8. Also bind simple references to final constants defined in the current module (cur_mod_id). Binding to references is best effort -- we don't bind references to other modules. Mypyc trusts these to be correct in compiled modules, so that it can replace a constant expression (or a reference to one) with the statically computed value. We don't want to infer constant values based on stubs, in particular, as these might not match the implementation (due to version skew, for example). Return None if unsuccessful. TrueTFalseF.rN) isinstancer valuer rrr namenoderis_finalfullnamersplit final_valuerr constant_fold_exprleftrightconstant_fold_binary_opopr rconstant_fold_unary_op)rrrrr#r$s R/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/mypy/constant_fold.pyr"r"s$$  z$  z$ ""z$ $$:z D( # #: 9  4 Y' ! !5y tS ! !    $$S!,,Q/:==$E%+//  D& ! !:!$)Z88"4:z::   1*47D%@@ @ D) $ $:"49j99  )$'599 9 4r&r# ConstantValuer$ct|tr&t|trt|||St|tr&t|trt |||St|tr&t|trt |||St|tr&t|trt |||S|dkr/t|t rt|t r||zS|dkr/t|t rt|tr||zS|dkr/t|trt|t r||zS|dkr6t|ttfrt|t r||zS|dkr6t|t r!t|ttfr||zS|dkr6t|ttfrt|t r||z S|dkr6t|t r!t|ttfr||z SdS)N+*-)rintconstant_fold_binary_int_opfloatconstant_fold_binary_float_oprcomplex)r&r#r$s r(r%r%Os<$<E3!7!7<*2tU;;;$>:eU#;#;>,Ru=== D% >Zs%;%;>,Ru=== D#  >:eU#;#;>,Ru=== SyyZc**yz%/E/Eye| sz$,,E31G1Ge| sz$,,E31G1Ge| SyyZsEl33y 5'8R8Rye| sz$00ZU|5T5Te| sz$e 55*UG:T:Te| sz$00ZU|5T5Te| 4r)r/int | float | Nonec|dkr||zS|dkr||z S|dkr||zS|dkr |dkr||z Sn|dkr |dkr||zSn|dkr |dkr||zSno|dkr||zS|d kr||zS|d kr||z S|d kr |dkr||zSn<|d kr |dkr||z Sn*|d kr$|dkr||z}t|tsJ|SdS)Nr,r.r-/r//%&|^z<>**)rr/r&r#r$rets r(r0r0rsT Syye| Syye| se| s A::%<   t A::5=   s A::%<   se| se| se| t A::5=   t A::5=   t A::+Cc3'' ' ' 'J 4r) int | float float | Nonect|tr"t|tr J|||f|dkr||zS|dkr||z S|dkr||zS|dkr |dkr||z Sn|dkr |dkr||zSnq|dkr |dkr||zSn_|dkrY|dkrt|ts|dkr8 ||z}t|ts J||S#t$rYdSwxYwdS) Nr,r.r-r6rr7r8r<)rr/r1 OverflowErrorr=s r(r2r2sX4%%T*UC*@*@TTBeCTTTT Syye| se| se| s A::%<   t A::5=   s A::%<   t 1HHE3//HD1HH Ek"#u--22s222 !   tt  4s7C C+*C+rc|dkrt|ttfr| S|dkrt|tr|S|dkrt|ttfr|SdS)Nr.~r,)rr/r1)r&rs r(r'r'sg SyyZU|44yv sz%--v sz%#u66 4r)N)rrrrrr)r&rr#r*r$r*rr)r&rr#r/r$r/rr4)r&rr#r?r$r?rr@)r&rrr*rr4)__doc__ __future__rtypingrr mypy.nodesrrrr r r r r rr/boolr1r3rr*r__annotations__r"r%r0r2r'r)r(rLsR #"""""                      c445 45 55551111h    F!!!!H<r)