o DDi @s<ddlZddlmZddlmZddlmZmZmZm Z m Z m Z m Z m Z mZmZmZddlmZmZmZddlmZddlmZer\ddlmZdd lmZdd lmZdd lmZm Z ed Z!d dde eede de de"fffddZ#dedde dde dfddZ$d ddedddfddZ%GdddZ&dS)N) ExitStack)MappingProxyType) Anycast CollectionIterableMappingMutableMapping MutableSetOptionalType TYPE_CHECKINGTypeVar)BatchableCSTVisitor visit_batched VisitorMethod)MetadataException)BatchableMetadataProvider)CSTNode)Module) CSTVisitorT)BaseMetadataProvider ProviderT_TwrapperMetadataWrapper providersreturnrrcCs||dd|DS)zW Returns map of metadata mappings from resolving ``providers`` on ``wrapper``. cSs i|] }t|tt|jqS)typerdict _computed.0prrP/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/metadata/wrapper.py 5s z"_gen_batchable..)r)rrrrr%_gen_batchable)s r'gatheredcCs,|D]}||vr||t|j|q|S)zM Recursively gathers all the given providers and their dependencies. )add_gather_providersMETADATA_DEPENDENCIES)rr(Prrr%r*8s   r*c s"tj}tt|t|}t|dkrt}|D]1}t|j|rMt|tr1| |q|j r?|j | n| j|<| |qfdd|D}t|}j|||O}t|dkrt|dkrddd|D}td|||8}t|dksdSdS)zx Updates the _metadata map on wrapper with metadata from the given providers as well as their dependencies. rcs(g|]}|jr|j|n|qSr) gen_cache_cachegetr"rrr% ^sz!_resolve_impl..z, cSsg|]}|jqSr)__name__)r#r,rrr%r1gsz"Detected circular dependencies in N)set _metadatakeysr*lenr+issubset issubclassrr)r-r.r/_genr'updatejoinr) rr completed remaining batchabler,initialized_batchablemetadata_batchnamesrr0r% _resolve_implEs2        rBc @seZdZUdZgdZded<ededeffed<edefed<d ifd dd e d edefd dfddZ d e fddZ e d%ddZdedd edeffddZdedd ededefffddZd&ddZ  d'd eed!eed"eed dfd#d$ZdS)(ra A wrapper around a :class:`~libcst.Module` that stores associated metadata for that module. When a :class:`MetadataWrapper` is constructed over a module, the wrapper will store a deep copy of the original module. This means ``MetadataWrapper(module).module == module`` is ``False``. This copying operation ensures that a node will never appear twice (by identity) in the same tree. This allows us to uniquely look up metadata for a node based on a node's identity. )__moduler4r.r_MetadataWrapper__modulerrr4r.Fmoduleunsafe_skip_copycacherNcCs"|s|}||_i|_||_dS)a :param module: The module to wrap. This is deeply copied by default. :param unsafe_skip_copy: When true, this skips the deep cloning of the module. This can provide a small performance benefit, but you should only use this if you know that there are no duplicate nodes in your tree (e.g. this module came from the parser). :param cache: Pass the needed cache to wrapper to be used when resolving metadata. N) deep_clonerDr4r.)selfrErFrGrrr%__init__s  zMetadataWrapper.__init__cCsdtt|jddS)NzMetadataWrapper( z z, ))textwrapindentreprrErIrrr%__repr__szMetadataWrapper.__repr__cCs|jS)a| The module that's wrapped by this MetadataWrapper. By default, this is a deep copy of the passed in module. :: mw = ModuleWrapper(module) # Because `mw.module is not module`, you probably want to do visit and do # your analysis on `mw.module`, not `module`. mw.module.visit(DoSomeAnalysisVisitor) )rDrNrrr%rEszMetadataWrapper.moduleproviderzBaseMetadataProvider[_T]cCs8||jvr |j|}n||g|}ttdtf|S)zR Returns a copy of the metadata mapping computed by ``provider``. r)r4 resolve_manyrrr)rIrPmetadatarrr%resolves  zMetadataWrapper.resolvercst|fdd|DS)z Returns a copy of the map of metadata mapping computed by each provider in ``providers``. The returned map does not contain any metadata from undeclared metadata dependencies that ``providers`` has. csi|]}|j|qSr)r4)r#krNrr%r&sz0MetadataWrapper.resolve_many..)rB)rIrrrNr%rQs zMetadataWrapper.resolve_manyvisitorrcCs:|||j|WdS1swYdS)z Convenience method to resolve metadata before performing a traversal over ``self.module`` with ``visitor``. See :func:`~libcst.Module.visit`. N)rSrEvisit)rIrUrrr%rVs  $zMetadataWrapper.visitvisitors before_visit after_leavecCsTt}|D] }|||qt|j|||WdS1s#wYdS)z Convenience method to resolve metadata before performing a traversal over ``self.module`` with ``visitors``. See :func:`~libcst.visit_batched`. N)r enter_contextrSrrE)rIrWrXrYstackvrrr%rs  $zMetadataWrapper.visit_batched)rr)rUrrr)NN)r2 __module__ __qualname____doc__ __slots____annotations__r robjectboolrJstrrOpropertyrEr rrSrrQrVrrr rrrrrr%rmsT          )'rK contextlibrtypesrtypingrrrrrr r r r r rlibcst._batched_visitorrrrlibcst._exceptionsrlibcst.metadata.base_providerrlibcst._nodes.baserlibcst._nodes.modulerlibcst._visitorsrrrrrbr'r*rBrrrrr%sF  4          (