o 2h@sdZddlZddlZGdddejZGdddeZGdddeZeZdd d Z ed fd d Z dddZ dddZ dddZ dddZdddZdS)z4AST node annotation support. Adapted from Tangent. Nc@s2eZdZdZd ddZddZddZd d ZdS) NoValuez2Base class for different types of AST annotations.NcCst|||dS)N)default)getanno)selfnoderr`/var/www/html/chatgem/venv/lib/python3.10/site-packages/tensorflow/python/autograph/pyct/anno.pyof$sz NoValue.ofcCst|||dSN)setanno)rrvaluerrradd_to'szNoValue.add_tocCs t||Sr )hasanno)rrrrrexists* zNoValue.existscCs t|jSr )strname)rrrr__repr__-rzNoValue.__repr__r )__name__ __module__ __qualname____doc__r r rrrrrrr!s   rc@s(eZdZdZdZdZdZdZdZdZ dS) BasiczhContainer for basic annotation keys. The enum values are used strictly for documentation purposes. z>Qualified name, as it appeared in the code. See qual_names.py.zBThis node should be preserved as is and not processed any further.zWhen a node is annotated with this, the remainder of the block should be indented below it. The annotation contains a tuple (new_body, name_map), where `new_body` is the new indented block and `name_map` allows renaming symbols.zaInformation about the source code that converted code originated from. See origin_information.py.zvUser directives associated with a statement or a variable. Typically, they affect the immediately-enclosing statement.zQA special annotation containing additional test code to be executed in for loops.N) rrrrQNSKIP_PROCESSINGINDENT_BLOCK_REMAINDERORIGIN DIRECTIVESEXTRA_LOOP_TESTrrrrr1src@sLeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdS)StaticzrContainer for static analysis annotation keys. The enum values are used strictly for documentation purposes. z5Symbol is a parameter to the function being analyzed.z2The scope for the annotated node. See activity.py.z3The scope for the argument list of a function call.z7The scope for the test node of a conditional statement.z`The scope for the main body of a statement (True branch for if statements, main body for loops).zeThe scope for the orelse body of a statement (False branch for if statements, orelse body for loops).z=Reaching definition information. See reaching_definitions.py.zQThe value of DEFINITIONS that applied to the original code before any conversion.zWLocal function definitions that may exist when exiting the node. See reaching_fndefs.pyzDSymbols defined when entering the node. See reaching_definitions.py.z4Symbols live when exiting the node. See liveness.py.z5Symbols live when entering the node. See liveness.py.z/Static type information. See type_inference.py.z4Types of closure symbols at each detected call site.z0Static value information. See type_inference.py.N)rrrrIS_PARAMSCOPE ARGS_SCOPE COND_SCOPE BODY_SCOPE ORELSE_SCOPE DEFINITIONSORIG_DEFINITIONSDEFINED_FNS_INDEFINED_VARS_IN LIVE_VARS_OUT LIVE_VARS_INTYPES CLOSURE_TYPESVALUErrrrrIs.r ___pyct_annocCs"t||stStt||Sr )hasattr frozensetgetattrkeys)r field_namerrrr3us r3cCs2|tust||r|t||vrt|||S|Sr )FAILr0r2)rkeyrr4rrrr{srcCst||o |t||vSr )r0r2)rr6r4rrrrsrcCsBt||i}t||||||<||jvr|j|f7_dSdSr )r2setattr_fields)rr6r r4 annotationsrrrr s   r cs@t|}||=|st|tfdd|jD|_dSdS)Nc3s|] }|kr|VqdSr r).0fr4rr szdelanno..)r2delattrtupler8)rr6r4r9rr<rdelannos  r@cCs0t|||drt||t|||d|ddSdS)Nr<)rr r) from_nodeto_noder6r4rrrcopyannos  rCc CsDt|D]}|D]}t|||rt|||t||||q qdS)a+Recursively copies annotations in an AST tree. Args: node: ast.AST copy_map: Dict[Hashable, Hashable], maps a source anno key to a destination key. All annotations with the source key will be copied to identical annotations with the destination key. field_name: str N)gastwalkrr r)rcopy_mapr4nkrrrdups  rI)r/)renumrDEnumrrrobjectr5r3rrr r@rCrIrrrrs )