o DDi@sddlZddlmZddlmZddlmZmZmZm Z m Z m Z m Z m Z mZmZmZer@ddlmZddlmZmZddlmZedZGd d d ZGd d d e eZGd ddeZdS)N)ABC)contextmanager) CallablecastClassVar CollectionGenericIteratorMappingType TYPE_CHECKINGTypeVarUnion)CSTNode)BaseMetadataProvider ProviderT)MetadataWrapper_Tc@s eZdZdS)_UNDEFINED_DEFAULTN)__name__ __module__ __qualname__rrS/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/_metadata_dependent.pyr$src@s8eZdZdZdegefddfddZdefddZdS) LazyValueam The class for implementing a lazy metadata loading mechanism that improves the performance when retriving expensive metadata (e.g., qualified names). Providers including :class:`~libcst.metadata.QualifiedNameProvider` use this class to load the metadata of a certain node lazily when calling :func:`~libcst.MetadataDependent.get_metadata`. callablereturnNcCs||_t|_dSN)rr return_value)selfrrrr__init__1s zLazyValue.__init__cCs |jtur ||_tt|jSr)rrrrrrrrr__call__5s   zLazyValue.__call__)rrr__doc__rrr r"rrrrr(src @seZdZUdZededeffed<dZee ded<dd d Z e de dfd d Z e d ddedfddZefdeddddedefddZdS)MetadataDependentz The low-level base class for all classes that declare required metadata dependencies. :class:`~libcst.CSTVisitor` and :class:`~libcst.CSTTransformer` extend this class. rrmetadatarMETADATA_DEPENDENCIESrNcCs i|_dSr)r%r!rrrr Js zMetadataDependent.__init__cCsXz|jWSty+t}t|D] }t|tr ||jqt ||_|jYSw)z Returns all metadata dependencies declared by classes in the MRO of ``cls`` that subclass this class. Recursively searches the MRO of the subclass for metadata dependencies. ) &_INHERITED_METADATA_DEPENDENCIES_CACHEAttributeErrorsetinspectgetmro issubclassr$updater& frozenset)cls dependenciescrrrget_inherited_dependenciesMs     z,MetadataDependent.get_inherited_dependencieswrapperrccs"|||_dVi|_dS)a Context manager that resolves all metadata dependencies declared by ``self`` (using :func:`~libcst.MetadataDependent.get_inherited_dependencies`) on ``wrapper`` and caches it on ``self`` for use with :func:`~libcst.MetadataDependent.get_metadata`. Upon exiting this context manager, the metadata cache on ``self`` is cleared. N) resolve_manyr2r%)rr3rrrresolveas  zMetadataDependent.resolvekeyzBaseMetadataProvider[_T]nodedefaultcCs||vrt|jdt|jd||jvr!t|jd|tur/|j|||}n|j||}t|tr>|}t t |S)z Returns the metadata provided by the ``key`` if it is accessible from this visitor. Metadata is accessible in a subclass of this class if ``key`` is declared as a dependency by any class in the MRO of this class. z$ is not declared as a dependency in z.METADATA_DEPENDENCIES.z@ is a dependency, but not set; did you forget a MetadataWrapper?) r2KeyErrorrtyper%rget isinstancerrr)rr6r7r8valuerrr get_metadataps     zMetadataDependent.get_metadata)rN)rrrr#r object__annotations__r&rrr classmethodr2rr r5rr rr>rrrrr$;s&   r$)r*abcr contextlibrtypingrrrrrr r r r r rlibcst._nodes.baserlibcst.metadata.base_providerrrlibcst.metadata.wrapperrrrrr$rrrrs  4