o i@sFddlZddlmZddlmZmZddlZddl m Z m Z m Z ddl mZmZmZmZGdddeZejd ejd ejd fZejd ejd ejdgesWejdfndesaejdfndejderresrejdfndRZejdZgesejdfnderejdfndRZ ddeddeefddZdS)N)IntFlag)OptionalUnion)DUAL_ARG_OPCODESCellVarFreeVar)PY311PY312PY313PY314c@s@eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdS) CompilerFlagszPossible values of the co_flags attribute of Code object. Note: We do not rely on inspect values here as some of them are missing and furthermore would be version dependent. r @iiiN)__name__ __module__ __qualname____doc__ OPTIMIZED NEWLOCALSVARARGS VARKEYWORDSNESTED GENERATORNOFREE COROUTINEITERABLE_COROUTINEASYNC_GENERATORFUTURE_GENERATOR_STOPFUTURE_ANNOTATIONSr&r&B/home/ubuntu/.local/lib/python3.10/site-packages/bytecode/flags.pyr sr STORE_NAME LOAD_NAME DELETE_NAME GET_AWAITABLE GET_AITER GET_ANEXTBEFORE_ASYNC_WITHr&SETUP_ASYNC_WITH END_ASYNC_FORASYNC_GEN_WRAP YIELD_VALUE YIELD_FROMRETURN_GENERATORbytecode)z_bytecode.Bytecodez_bytecode.ConcreteBytecodez_bytecode.ControlFlowGraphis_asynccCstd}t|tjtjtjfsd}t||t|tjr!|n|}t|tj}d}|s/dn|jo4|j }d}d} d} t |} | D]} t| tj tj tj tjfrQqA| j} | tvr[d}qA| tvrbd}qA| tkrtrtt| }tj tj tj tjfr tt| }tj tj tj tjfsy|jdksJ|jd@dkrd} qAd}qAd} qA| tvrd} qA| tjvrd}qA|s| tvrt| jdtst| jdtrd}qAtr| tjvrt| jttfrd}qA|r|tjO}|s|tj O}||j!tj"tj#Btj$Btj%B@O}|dvrl|j!tj&@s|j!tj'@r| r|tj'O}ng|tj&O}na|j!tj(@r3|r-d }t||tj(O}nK|rE| r?|tj'O}n?|tj&O}n9| sK| rc|r]| rW|tj'O}n'|tj&O}n!|tj)O}n|rk|tj&O}n|rstd | sy| r~|tj)O}||j!tj*@O}|S) aInfer the proper flags for a bytecode based on the instructions. Because the bytecode does not have enough context to guess if a function is asynchronous the algorithm tries to be conservative and will never turn a previously async code into a sync one. Parameters ---------- bytecode : Bytecode | ConcreteBytecode | ControlFlowGraph Bytecode for which to infer the proper flags is_async : bool | None, optional Force the code to be marked as asynchronous if True, prevent it from being marked as asynchronous if False and simply infer the best solution based on the opcode and the existing flag if None. rzIExpected a Bytecode, ConcreteBytecode or ControlFlowGraph instance not %sTFRESUMEr)NTzThe ITERABLE_COROUTINE flag is set but bytecode thatcan only be used in async functions have been detected. Please unset that flag before performing inference.ziThe is_async argument is False but bytecodes that can only be used in async functions have been detected.)+r isinstance _bytecodeBytecodeConcreteBytecodeControlFlowGraph ValueError_get_instructionscellvarsfreevarsiter SetLinenoLabelTryBeginTryEndopcodeUNOPTIMIZED_OPCODES ASYNC_OPCODESYIELD_VALUE_OPCODEr nextnameargGENERATOR_LIKE_OPCODES_opcodehasfreerrr haslocalrrr flagsrrrrr!r#r"rr$)r5r6rRmsg instructions is_concrete optimizedhas_free known_asyncknown_generatorpossible_generator instr_iterinstrrGnir&r&r' infer_flags@s                           r^)N)rGrOenumrtypingrrr5r:r\rrrutilsr r r r r opmaprHrIrJrNboolr^r&r&r&r'sJ