o DDi@sjddlmZddlmZmZmZmZmZddlZ ddl m Z ddl m Z er,ddlmZGdddZdS) )Path) CollectionDictListMapping TYPE_CHECKINGN)StrPath)MetadataWrapper) ProviderTc @seZdZ  ddedeededdeded d f d d Ze d e de ee fffd dZ dddZded e de ffddZded efddZd S)FullRepoManagerF repo_root_dirpaths providersr timeoutuse_pyproject_tomlreturnNcCs0t||_i|_||_||_||_t||_dS)a$ Given project root directory with pyre and watchman setup, :class:`~libcst.metadata.FullRepoManager` handles the inter process communication to read the required full repository cache data for metadata provider like :class:`~libcst.metadata.TypeInferenceProvider`. :param paths: a collection of paths to access full repository data. :param providers: a collection of metadata provider classes require accessing full repository data, currently supports :class:`~libcst.metadata.TypeInferenceProvider` and :class:`~libcst.metadata.FullyQualifiedNameProvider`. :param timeout: number of seconds. Raises `TimeoutExpired `_ when timeout. N)r root_path_cache_timeout_use_pyproject_toml _providerslist_paths)selfr rrrrrZ/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/metadata/full_repo_manager.py__init__s zFullRepoManager.__init__cCs||jS)z The full repository cache data for all metadata providers passed in the ``providers`` parameter when constructing :class:`~libcst.metadata.FullRepoManager`. Each provider is mapped to a mapping of path to cache. ) resolve_cacher)rrrrcache.szFullRepoManager.cachecCsJ|js#i}|jD]}|j}|r||j|j|j|jd||<q||_dSdS)aq Resolve cache for all providers that require it. Normally this is called by :meth:`~FullRepoManager.get_cache_for_path` so you do not need to call it manually. However, if you intend to do a single cache resolution pass before forking, it is a good idea to call this explicitly to control when cache resolution happens. )rrN)rr gen_cacherrrr)rrproviderhandlerrrrr8s   zFullRepoManager.resolve_cachepathcs2|jvr td|fdd|jDS)aq Retrieve cache for a source file. The file needs to appear in the ``paths`` parameter when constructing :class:`~libcst.metadata.FullRepoManager`. .. code-block:: python manager = FullRepoManager(".", {"a.py", "b.py"}, {TypeInferenceProvider}) MetadataWrapper(module, cache=manager.get_cache_for_path("a.py")) ziThe path needs to be in paths parameter when constructing FullRepoManager for efficient batch processing.cs0i|]\}}|D] \}}|kr ||q qSr)items).0r!files_pathdatar#rr ^sz6FullRepoManager.get_cache_for_path..)r Exceptionrrr$)rr#rr)rget_cache_for_pathMs  z"FullRepoManager.get_cache_for_pathcCs*t|j|}||}t|d|S)a Create a :class:`~libcst.metadata.MetadataWrapper` given a source file path. The path needs to be a path relative to project root directory. The source code is read and parsed as :class:`~libcst.Module` for :class:`~libcst.metadata.MetadataWrapper`. .. code-block:: python manager = FullRepoManager(".", {"a.py", "b.py"}, {TypeInferenceProvider}) wrapper = manager.get_metadata_wrapper_for_path("a.py") T)cst parse_moduler read_textr,r )rr#modulerrrrget_metadata_wrapper_for_pathes  z-FullRepoManager.get_metadata_wrapper_for_path)r F)rN)__name__ __module__ __qualname__rrstrintboolrpropertyrrobjectrrr,r r1rrrrr s*   r )pathlibrtypingrrrrrlibcstr- libcst._typesrlibcst.metadata.wrapperr libcst.metadata.base_providerr r rrrrs