o
DDis @ s d dl Z d dlmZ d dlmZmZmZmZmZm Z d dl
Z d dl m
Z
mZ d dlmZmZ d dlmZ d dlmZmZmZ G dd d eee ZG d
d d e jZG d d
d
eee ZG dd de jZ dS ) N)Path)Any
CollectionListMappingOptionalUnion) LazyValueMetadataDependent) calculate_module_and_packageModuleNameAndPackage)BatchableMetadataProvider)
QualifiedNameQualifiedNameSource
ScopeProviderc @ sV e Zd ZdZefZdejdee fddZ
e de dej
deeef de fdd Zd
S ) QualifiedNameProvideran
Compute possible qualified names of a variable CSTNode
(extends `PEP-3155 `_).
It uses the
:func:`~libcst.metadata.Scope.get_qualified_names_for` underlying to get qualified names.
Multiple qualified names may be returned, such as when we have conditional imports or an
import shadows another. E.g., the provider finds ``a.b``, ``d.e`` and
``f.g`` as possible qualified names of ``c``::
>>> wrapper = MetadataWrapper(
>>> cst.parse_module(dedent(
>>> '''
>>> if something:
>>> from a import b as c
>>> elif otherthing:
>>> from d import e as c
>>> else:
>>> from f import g as c
>>> c()
>>> '''
>>> ))
>>> )
>>> call = wrapper.module.body[1].body[0].value
>>> wrapper.resolve(QualifiedNameProvider)[call],
{
QualifiedName(name="a.b", source=QualifiedNameSource.IMPORT),
QualifiedName(name="d.e", source=QualifiedNameSource.IMPORT),
QualifiedName(name="f.g", source=QualifiedNameSource.IMPORT),
}
For qualified name of a variable in a function or a comprehension, please refer
:func:`~libcst.metadata.Scope.get_qualified_names_for` for more detail.
nodereturnc C s t | }|| d S N)QualifiedNameVisitorvisitselfr visitor r V/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/metadata/name_provider.py visit_Module: s z"QualifiedNameProvider.visit_Moduler namec sF | t|t }t trt fdd|D S t fdd|D S )z`Check if any of qualified name has the str name or :class:`~libcst.metadata.QualifiedName` name.c 3 s | ]}|j kV qd S r r .0qnr r r E s z1QualifiedNameProvider.has_name..c 3 s | ]}| kV qd S r r r r r r r" G s ) get_metadatar set
isinstancestrany)r r r qualified_namesr r r has_name> s
z QualifiedNameProvider.has_nameN)__name__
__module__ __qualname____doc__r METADATA_DEPENDENCIEScstModuler boolr staticmethodr
CSTNoder r& r r) r r r r r s "
r c s2 e Zd Zd
ddZdejdef fdd Z ZS ) r providerr r Nc C s
|| _ d S r )r4 )r r4 r r r __init__K s
z QualifiedNameVisitor.__init__r c sR | j t d r| j t fdd n| j t t dS )Nc s
S r )get_qualified_names_forr r scoper r R s
z/QualifiedNameVisitor.on_visit..T)r4 r# r set_metadatar r$ superon_visit)r r __class__r7 r r< N s z QualifiedNameVisitor.on_visit)r4 r r N) r* r+ r, r5 r/ r3 r1 r<
__classcell__r r r= r r J s
r c sx e Zd ZdZefZedddedee de
de de e e
f f
d d
Zd e
dd f fd
d Zdejde
fddZ ZS )FullyQualifiedNameProvidera
Provide fully qualified names for CST nodes. Like :class:`QualifiedNameProvider`,
but the provided :class:`QualifiedName` instances have absolute identifier names
instead of local to the current module.
This provider is initialized with the current module's fully qualified name, and can
be used with :class:`~libcst.metadata.FullRepoManager`. The module's fully qualified
name itself is stored as a metadata of the :class:`~libcst.Module` node. Compared to
:class:`QualifiedNameProvider`, it also resolves relative imports.
Example usage::
>>> mgr = FullRepoManager(".", {"dir/a.py"}, {FullyQualifiedNameProvider})
>>> wrapper = mgr.get_metadata_wrapper_for_path("dir/a.py")
>>> fqnames = wrapper.resolve(FullyQualifiedNameProvider)
>>> {type(k): v for (k, v) in fqnames.items()}
{: {QualifiedName(name='dir.a', source=)}}
Fuse_pyproject_toml root_pathpathsrB kwargsr c s fdd|D }|S )Nc s i | ]
}|t |d qS )rA )r )r pathrC rB r r
z s z8FullyQualifiedNameProvider.gen_cache..r )clsrC rD rB rE cacher rG r gen_cacheq s z$FullyQualifiedNameProvider.gen_cacherJ Nc s t | |j| _|j| _d S r )r; r5 r module_namepackage package_name)r rJ r= r r r5 s z#FullyQualifiedNameProvider.__init__r c C s8 t | | j| j}|| | |t| jtjdh dS )N)r sourceT)FullyQualifiedNameVisitorrL rN r r: r r LOCALr r r r r s
z'FullyQualifiedNameProvider.visit_Module)r* r+ r, r- r r. classmethodr r r& r1 r r r rK r5 r/ r0 r r? r r r= r r@ Z s$
r@ c @ sv e Zd ZededededefddZededededefdd Zd
edededd fd d
Z de
j de fddZ
d S )rP rL rN r r c C s` | d}t|t| }|dkr)|}|d|d }t||k r%td|d } | d| S )N.r z2attempted relative import beyond top-level package)lstriplenrsplit ImportError)rL rN r abs_namenum_dotsbitsr r r _fully_qualify_local s
z.FullyQualifiedNameVisitor._fully_qualify_localqnamec C sN |j tjkr|S |j}|j tjkr|ds|S t| ||j}tj ||dS )NrS r )
rO r BUILTINr IMPORT
startswithrP r\ dataclassesreplace)rL rN r] r new_namer r r _fully_qualify s z(FullyQualifiedNameVisitor._fully_qualifyr4 Nc C s || _ || _|| _d S r )rL rN r4 )r r4 rL rN r r r r5 s
z"FullyQualifiedNameVisitor.__init__r c s6 j t|}|d ur j | fdd|D dS )Nc s h | ] }t j j|qS r )rP rd rL rN )r r] r r r s
z5FullyQualifiedNameVisitor.on_visit..T)r4 r# r r: )r r qnamesr re r r< s
z"FullyQualifiedNameVisitor.on_visit)r* r+ r, r2 r& r\ r rd r@ r5 r/ r3 r1 r< r r r r rP s.
rP ) ra pathlibr typingr r r r r r libcstr/ libcst._metadata_dependentr r
libcst.helpers.moduler r libcst.metadata.base_providerr
libcst.metadata.scope_providerr r r r
CSTVisitorr r@ rP r r r r s 57