o ÕtBh{ã@sšdZddlmZddlmZmZmZmZmZm Z m Z m Z m Z m Z mZmZddlmZddlmZddlmZddlmZddd„Zddd„Zddd„ZdS)z½IRBuilder AST transform helpers shared between expressions and statements. Shared code that is tightly coupled to mypy ASTs can be put here instead of making mypyc.irbuild.builder larger. é)Ú annotations) ÚLDEFÚ BytesExprÚComparisonExprÚ ExpressionÚ FloatExprÚIntExprÚ MemberExprÚNameExprÚOpExprÚStrExprÚ UnaryExprÚVar)Ú BasicBlock)Ú is_tagged)Ú IRBuilder)Úconstant_fold_exprÚselfrÚerÚtruerÚfalseÚreturnÚNonecCsàt|tƒrC|jdvrC|jdkr)tƒ}t||j||ƒ| |¡t||j||ƒdStƒ}t||j||ƒ| |¡t||j||ƒdSt|tƒrW|jdkrWt||j ||ƒdSt ||||ƒ}|rbdS|  |¡}|  |||¡dS)N)ÚandÚorrÚnot) Ú isinstancer ÚoprÚprocess_conditionalÚleftÚactivate_blockÚrightr ÚexprÚ$maybe_process_conditional_comparisonÚacceptÚadd_bool_branch)rrrrÚnewÚresÚreg©r)úp/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/mypyc/irbuild/ast_helpers.pyrs"    rÚboolc Cs¼t|tƒr t|jƒdkrdS| |jd¡}| |jd¡}t|ƒr&t|ƒs(dS|jd}|dvr3dS|jd}|jd}t||ƒ} |j|| d} |j|dd} |j   | | ||||j ¡dS)aMTransform simple tagged integer comparisons in a conditional context. Return True if the operation is supported (and was transformed). Otherwise, do nothing and return False. Args: e: Arbitrary expression true: Branch target if comparison is true false: Branch target if comparison is false éFré)z==z!=úz>=)Ú can_borrowT) rrÚlenÚoperandsÚ node_typerÚ operatorsÚis_borrow_friendly_exprr$ÚbuilderÚcompare_tagged_conditionÚline) rrrrÚltypeÚrtyperÚ left_exprÚ right_exprÚ borrow_leftrr!r)r)r*r#8s      r#r"cCszt|ttttfƒr dSt|ttttfƒrt ||ƒdurdSt|tƒr/t|j t ƒr/|j t kr/dSt|tƒr;| |¡r;dSdS)zŽCan the result of the expression borrowed temporarily? Borrowing means keeping a reference without incrementing the reference count. TNF)rrrr rr r r r rÚnoderÚkindrÚis_native_attr_ref)rr"r)r)r*r5Xsÿ r5N) rrrrrrrrrr) rrrrrrrrrr+)rrr"rrr+)Ú__doc__Ú __future__rÚ mypy.nodesrrrrrrr r r r r rÚ mypyc.ir.opsrÚmypyc.ir.rtypesrÚmypyc.irbuild.builderrÚmypyc.irbuild.constant_foldrrr#r5r)r)r)r*Ús 8