o i{A@sddlmZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZdd lmZdd lmZdd lmZdd lmZmZerxddlmZmZmZddlmZddl m!Z!m"Z"ddl#m$Z$ddl%m&Z&dZ'eGdddZ(edddZddZ)e d[ddd d\d(d)Z*e d[dddddddd*dd*dd*d*d+ d]d>d?Z+e d[dddddddddd*dd*d*d+ d]d@dAZ,eddddddddBd^dEdFZ-edddddddddGd_dJdKZ.d*dLd`dPdQZ/dadRdSZ0dd*dTdbdXdYZ1dS)c) annotationsN)contextmanager) dataclass)invalidate_caches)Path)dedent) TYPE_CHECKING)inspect)visit)LinesCollection)load)ModuleObject)IteratorMappingSequence)ModulesCollection)DocstringOptionsDocstringStyle)Parser) Extensionsgriffe_c@s:eZdZUdZded< ded< ded< d d d Zd S) TmpPackagezA temporary package. The `tmpdir` and `path` parameters can be passed as relative path. They will be resolved to absolute paths after initialization. rtmpdirstrnamepathreturnNonecCs|j|_|j|_dSN)rresolver)selfr"\/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/griffe/_internal/tests.py __post_init__0s zTmpPackage.__post_init__N)rr)__name__ __module__ __qualname____doc____annotations__r$r"r"r"r#r!s rmodule module_namecoderr,rIterator[tuple[str, Path]]ccsbtjtd }t||d}|jt|dd||fVWddS1s*wYdS)aBCreate a Python file containing the given code in a temporary directory. Parameters: code: The code to write to the temporary file. module_name: The name of the temporary module. Yields: module_name: The module name, as to dynamically import it. module_path: The module path. prefix.pyutf8encodingN)tempfileTemporaryDirectory_TMPDIR_PREFIXr write_textr)r-r,rtmpfiler"r"r#temporary_pyfile5s   "r:Tinitinitspackagemodules(Sequence[str] | Mapping[str, str] | Noner<boolr=Iterator[TmpPackage]c cs |pi}t|trt|d}ddd}tjtdk}t|}dt|j }||}|j d i||r=| d | D]5\} } |} t| j D])} | dr`| | jt| dd qL| | } | j d i||ru| d qLqAt|||VWd d S1swYd S) a(Create a package containing the given modules in a temporary directory. Parameters: package: The package name. Example: `"a"` gives a package named `a`, while `"a/b"` gives a namespace package named `a` with a package inside named `b`. If `init` is false, then `b` is also a namespace package. modules: Additional modules to create in the package. If a list, simply touch the files: `["b.py", "c/d.py", "e/f"]`. If a dict, keys are the file names and values their contents: `{"b.py": "b = 1", "c/d.py": "print('hey from c')"}`. init: Whether to create an `__init__` module in the top package. inits: Whether to create `__init__` modules in subpackages. Yields: A temporary package. T)parentsexist_okr/. __init__.py)r1z.pyir2r3Nr") isinstancelistdictfromkeysr5r6r7rjoinpartsmkdirjoinpathtouchitemsendswithr8rr) r>r?r<r= mkdir_kwargsr tmpdirpath package_name package_pathr,module_contents current_pathpartr"r"r#temporary_pypackageGs2    "rZF) r<r= extensionsdocstring_parserdocstring_optionslines_collectionmodules_collectionallow_inspection store_sourceresolve_aliasesresolve_externalresolve_implicitsearch_sys_pathr[Extensions | Noner\DocstringStyle | Parser | Noner]DocstringOptions | Noner^LinesCollection | Noner_ModulesCollection | Noner`rarbrc bool | NonerdreIterator[Module]c cst|rtjng}t||||d }t|j|jg||||||| | | | | dd VWddS1s3wYdS)aCreate and visit a temporary package. Parameters: package: The package name. Example: `"a"` gives a package named `a`, while `"a/b"` gives a namespace package named `a` with a package inside named `b`. If `init` is false, then `b` is also a namespace package. modules: Additional modules to create in the package. If a list, simply touch the files: `["b.py", "c/d.py", "e/f"]`. If a dict, keys are the file names and values their contents: `{"b.py": "b = 1", "c/d.py": "print('hey from c')"}`. init: Whether to create an `__init__` module in the top package. inits: Whether to create `__init__` modules in subpackages. extensions: The extensions to use. docstring_parser: The docstring parser to use. By default, no parsing is done. docstring_options: Docstring parsing options. lines_collection: A collection of source code lines. modules_collection: A collection of modules. allow_inspection: Whether to allow inspecting modules when visiting them is not possible. store_source: Whether to store code source in the lines collection. resolve_aliases: Whether to resolve aliases. resolve_external: Whether to try to load unspecified modules to resolve aliases. Default value (`None`) means to load external modules only if they are the private sibling or the origin module (for example when `ast` imports from `_ast`). resolve_implicit: When false, only try to resolve an alias if it is explicitly exported. search_sys_path: Whether to search the system paths for the package. Yields: A module. r;F search_pathsr[r\r]r^r_r`rarbrcrdforce_inspectionN)sysrrZr rr)r>r?r<r=r[r\r]r^r_r`rarbrcrdrern tmp_packager"r"r#temporary_visited_packagexs&1  "rrc cs|rtjng}t||||dd}z9t|j|jg||||||| | | | | dd VWttjD]}||ks?| |drFtj |dq1t n!ttjD]}||ks`| |drgtj |dqRt wWddS1swwYdS)aCreate and inspect a temporary package. Parameters: package: The package name. Example: `"a"` gives a package named `a`, while `"a/b"` gives a namespace package named `a` with a package inside named `b`. If `init` is false, then `b` is also a namespace package. modules: Additional modules to create in the package. If a list, simply touch the files: `["b.py", "c/d.py", "e/f"]`. If a dict, keys are the file names and values their contents: `{"b.py": "b = 1", "c/d.py": "print('hey from c')"}`. init: Whether to create an `__init__` module in the top package. inits: Whether to create `__init__` modules in subpackages. extensions: The extensions to use. docstring_parser: The docstring parser to use. By default, no parsing is done. docstring_options: Docstring parsing options. lines_collection: A collection of source code lines. modules_collection: A collection of modules. allow_inspection: Whether to allow inspecting modules. store_source: Whether to store code source in the lines collection. resolve_aliases: Whether to resolve aliases. resolve_external: Whether to try to load unspecified modules to resolve aliases. Default value (`None`) means to load external modules only if they are the private sibling or the origin module (for example when `ast` imports from `_ast`). resolve_implicit: When false, only try to resolve an alias if it is explicitly exported. search_sys_path: Whether to search the system paths for the package. Yields: A module. r;TrmrFN) rprrZr rrtupler?keys startswithpopr)r>r?r<r=r[r\r]r^r_r`rarbrcrdrernrqrr"r"r#temporary_inspected_packages<1   "rw)r,r[parentr\r]r^r_rx Module | Nonec cst|}t||d+\}} |pt}||| <t|| |||||||d } | | j|<| VWddS1s9wYdS)aHCreate and visit a temporary module with the given code. Parameters: code: The code of the module. module_name: The name of the temporary module. extensions: The extensions to use when visiting the AST. parent: The optional parent of this module. docstring_parser: The docstring parser to use. By default, no parsing is done. docstring_options: Docstring parsing options. lines_collection: A collection of source code lines. modules_collection: A collection of modules. Yields: The visited module. r+)filepathr-r[rxr\r]r^r_N)rr:r splitlinesr r_) r-r,r[rxr\r]r^r__rr*r"r"r#temporary_visited_modules&   "r})r, import_pathsr[rxr\r]r^r_r~list[Path] | Nonec cst||dG\} } |pt}||| <z#t|| |||||||d } | | j|<| VW|tjvr5tj|=tn |tjvrBtj|=twWddS1sQwYdS)aCreate and inspect a temporary module with the given code. Parameters: code: The code of the module. module_name: The name of the temporary module. import_paths: Paths to import the module from. extensions: The extensions to use when visiting the AST. parent: The optional parent of this module. docstring_parser: The docstring parser to use. By default, no parsing is done. docstring_options: Docstring parsing options. lines_collection: A collection of source code lines. modules_collection: A collection of modules. Yields: The inspected module. r+)rzr~r[rxr\r]r^r_N)r:r r{r r_rpr?r) r-r,r~r[rxr\r]r^r_r|rr*r"r"r#temporary_inspected_module4s2      "r) return_leafobjectsrrcGsH|std|d}|}|ddD] }||j||}q|r"|S|S)a<Link objects together, vertically. Parameters: *objects: A sequence of objects. The first one is at the top of the tree. return_leaf: Whether to return the leaf instead of the root. Raises: ValueError: When no objects are provided. Returns: The top or leaf object. $At least one object must be providedrN ValueError set_memberr)rrtopleafobjr"r"r#vtreehs  rcGs8|std|d}|ddD] }||j|q|S)a=Link objects together, horizontally. Parameters: *objects: A sequence of objects. All objects starting at the second become members of the first. Raises: ValueError: When no objects are provided. Returns: The first given object, with all the other objects as members of it. rrrNr)rrrr"r"r#htrees  r) leaf_packagerrrr csP|dfddtD}|s!|djd}||d_t|d|iS)aoLink objects together, vertically. Parameters: path: The complete module path, like `"a.b.c.d"`. leaf_package: Whether the deepest module should also be a package. return_leaf: Whether to return the leaf instead of the root. Raises: ValueError: When no objects are provided. Returns: The top or leaf module. rFc s2g|]\}}t|tgd|dRdqS)NrG)rz)r r).0indexrrMr"r# s2z module_vtree..r)split enumeraterz with_stem _filepathr)rrrr?rzr"rr# module_vtrees  r)r-rr,rrr.r) r>rr?r@r<rAr=rArrB) r>rr?r@r<rAr=rAr[rfr\rgr]rhr^rir_rjr`rArarArbrArcrkrdrArerArrl)r-rr,rr[rfrxryr\rgr]rhr^rir_rjrrl)r-rr,rr~rr[rfrxryr\rgr]rhr^rir_rjrrl)rrrrArr)rrrr)rrrrArrArr )2 __future__rrpr5 contextlibr dataclassesr importlibrpathlibrtextwraprtypingr!griffe._internal.agents.inspectorr griffe._internal.agents.visitorr griffe._internal.collectionsr griffe._internal.loaderr griffe._internal.modelsr rcollections.abcrrrr#griffe._internal.docstrings.parsersrrgriffe._internal.enumerationsr griffe._internal.extensions.baserr7rr:rZrrrwr}rrrrr"r"r"r#s              0CI-3