o DDi/@sddlmZddlmZddlmZmZmZmZm Z m Z m Z m Z m Z mZddlmZddlmZmZmZmZddlmZe rRddlmZddlmZmZdd lm Z e d Z!e d d d Z"eee"e"fZ#Gddde Z$Gdddeee"Z%Gdddee%e"Z&Gdddee%e"Z'dS))Path)MappingProxyType) GenericListMappingMutableMappingOptionalProtocolType TYPE_CHECKINGTypeVarUnion)BatchableCSTVisitor)_T_UNDEFINED_DEFAULT LazyValueMetadataDependent) CSTVisitor)CSTNode) _ModuleSelfTModule)MetadataWrapperzBaseMetadataProvider[object]_ProvidedMetadataTT) covariantc @sBeZdZddddedeedeedede ee ff d d Z dS) GenCacheMethodNF)timeoutuse_pyproject_toml root_pathpathsrrreturncCsdSN)selfrrrrr!r!V/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/metadata/base_provider.py__call__,szGenCacheMethod.__call__) __name__ __module__ __qualname__rrstrrintboolrobjectr$r!r!r!r#r+s rc seZdZUdZedefed<dZee ed<dde ddffdd Z d d de deffd d Z dddZdddeddfddZefdeddddeeeefdeffdd ZZS)BaseMetadataProvidera The low-level base class for all metadata providers. This class should be extended for metadata providers that are not visitor-based. This class is generic. A subclass of ``BaseMetadataProvider[T]`` will provider metadata of type ``T``. r _computedN gen_cachecachercs<ti|_|jr|durtd|jjd||_dS)Nz#Cache is required for initializing .)super__init__r-r. Exception __class__r%r/)r"r/r4r!r#r2Ks  zBaseMetadataProvider.__init__wrapperrcCsJi|_||||jWdn1swYtt|jS)z Resolves and returns metadata mapping for the module in ``wrapper``. This method is used by the metadata resolver and should not be called directly. N)r-resolve _gen_implmodulerdict)r"r6r!r!r#_genUs  zBaseMetadataProvider._genr9rcCdS)zR Override this method with a metadata computation implementation. Nr!r"r9r!r!r#r8gszBaseMetadataProvider._gen_implnodevaluecCs||j|<dS)zA Record a metadata value ``value`` for ``node``. N)r-)r"r>r?r!r!r# set_metadatamsz!BaseMetadataProvider.set_metadatakeyz BaseMetadataProvider[_MetadataT]defaultcsR|t|ur!|tur|j||}n|j|}t|tr|S|St|||S)z The same method as :func:`~libcst.MetadataDependent.get_metadata` except metadata is accessed from ``self._computed`` in addition to ``self.metadata``. See :func:`~libcst.MetadataDependent.get_metadata`. )typerr-get isinstancerr1 get_metadata)r"rAr>rBretr5r!r#rFss   z!BaseMetadataProvider.get_metadatar r9rrN)r%r&r'__doc__rMaybeLazyMetadataT__annotations__r.rrr+r2rr;r8r@rr r _MetadataTrF __classcell__r!r!r5r#r,7s0       r,c@eZdZdZdddZdS) VisitorMetadataProviderz The low-level base class for all non-batchable visitor-based metadata providers. Inherits from :class:`~libcst.CSTVisitor`. This class is generic. A subclass of ``VisitorMetadataProvider[T]`` will provider metadata of type ``T``. r9_ModuleTrNcCs||dSr )visitr=r!r!r#r8sz!VisitorMetadataProvider._gen_impl)r9rPrNr%r&r'rIr8r!r!r!r#rOsrOc@rN) BatchableMetadataProvidera The low-level base class for all batchable visitor-based metadata providers. Batchable providers should be preferred when possible as they are more efficient to run compared to non-batchable visitor-based providers. Inherits from :class:`~libcst.BatchableCSTVisitor`. This class is generic. A subclass of ``BatchableMetadataProvider[T]`` will provider metadata of type ``T``. r9rrNcCr<)z Batchables providers are resolved through _gen_batchable] so no implementation should be provided in _gen_impl. Nr!r=r!r!r#r8sz#BatchableMetadataProvider._gen_implrHrRr!r!r!r#rSs rSN)(pathlibrtypesrtypingrrrrrr r r r r libcst._batched_visitorrlibcst._metadata_dependentrrLrrrlibcst._visitorsrlibcst._nodes.baserlibcst._nodes.modulerrPrlibcst.metadata.wrapperr ProviderTrrJrr,rOrSr!r!r!r#s&  0      U