o w!g?@sddlmZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZmZdd lmZdd lmZdd lmZdd lmZdd lmZmZerxddlmZmZmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&dZ'eGdddZ(edddYddZ)e dZddd d[d(d)Z*e dZdddddddd*dd*dd*d+ d\d=d>Z+e dZdddddddddd*dd*d+ d\d?d@Z,eddddddddAd]dDdEZ-edddddddddFd^dIdJZ.d*dKd_dOdPZ/d`dQdRZ0dd*dSdadWdXZ1dS)b) annotationsN)contextmanager) dataclass)invalidate_caches)Path)dedent) TYPE_CHECKINGAny)inspect)visit)LinesCollection)load)ModuleObject)IteratorMappingSequence)ModulesCollection)DocstringStyle)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"O/mnt/skqttb/ctump_chatbot/chatbot/lib/python3.10/site-packages/_griffe/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]]ccs^tjtd}t||d}|t|||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.pyN)tempfileTemporaryDirectory_TMPDIR_PREFIXr write_textr)r-r,rtmpfiler"r"r#temporary_pyfile5s   "r7Tinitinitspackagemodules(Sequence[str] | Mapping[str, str] | Noner9boolr:Iterator[TmpPackage]c cs|pi}t|trt|d}ddd}tjtdi}t|}dt|j }||}|j d i||r=| d | D]3\} } |} t| j D]'} | dr^| | t| qL| | } | j d i||rs| d qLqAt|||VWddS1swYdS) 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.pyiNr") isinstancelistdictfromkeysr2r3r4rjoinpartsmkdirjoinpathtouchitemsendswithr5rr) r;r<r9r: mkdir_kwargsr tmpdirpath package_name package_pathr,module_contents current_pathpartr"r"r#temporary_pypackageGs2    "rWF) r9r: extensionsdocstring_parserdocstring_optionslines_collectionmodules_collectionallow_inspection store_sourceresolve_aliasesresolve_externalresolve_implicitrXExtensions | NonerYDocstringStyle | Parser | NonerZdict[str, Any] | Noner[LinesCollection | Noner\ModulesCollection | Noner]r^r_r` bool | NoneraIterator[Module]c csbt||||d}t|j|jg|||||| | | | | dd VWddS1s*wYdS)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: Additional 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. Yields: A module. r8F search_pathsrXrYrZr[r\r]r^r_r`raforce_inspectionN)rWr rr)r;r<r9r:rXrYrZr[r\r]r^r_r`ra tmp_packager"r"r#temporary_visited_packagexs$/ "rmc cst||||db}z7t|j|jg|||||| | | | | dd VWttjD]}||ks6||dr=tj |dq(t n!ttjD]}||ksW||dr^tj |dqIt wWddS1snwYdS)abCreate 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: Additional 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. Yields: A module. r8TrirCN) rWr rrtuplesysr<keys startswithpopr)r;r<r9r:rXrYrZr[r\r]r^r_r`rarlrr"r"r#temporary_inspected_packages:/  "rs)r,rXparentrYrZr[r\rt Module | Nonec cst|}t||d+\}} |pt}||| <t|| |||||||d } | | j|<| VWddS1s9wYdS)aSCreate 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: Additional docstring parsing options. lines_collection: A collection of source code lines. modules_collection: A collection of modules. Yields: The visited module. r+)filepathr-rXrtrYrZr[r\N)rr7r splitlinesr r\) r-r,rXrtrYrZr[r\_rr*r"r"r#temporary_visited_modules&   "ry)r, import_pathsrXrtrYrZr[r\rzlist[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: Additional docstring parsing options. lines_collection: A collection of source code lines. modules_collection: A collection of modules. Yields: The inspected module. r+)rvrzrXrtrYrZr[r\N)r7r rwr r\ror<r) r-r,rzrXrtrYrZr[r\rxrr*r"r"r#temporary_inspected_module.s2      "r|) return_leafobjectsrr}cGsH|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)r}r~topleafobjr"r"r#vtreebs  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)r~rrr"r"r#htreeys  r) leaf_packager}rrrcsP|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. rCc s2g|]\}}t|tgd|dRdqS)NrD)rv)rr).0indexrrJr"r# s2z module_vtree..r})split enumeraterv with_stem _filepathr)rrr}r<rvr"rr# module_vtrees  r)r-rr,rrr.r) r;rr<r=r9r>r:r>rr?)r;rr<r=r9r>r:r>rXrbrYrcrZrdr[rer\rfr]r>r^r>r_r>r`rgrar>rrh)r-rr,rrXrbrtrurYrcrZrdr[rer\rfrrh)r-rr,rrzr{rXrbrtrurYrcrZrdr[rer\rfrrh)r~rr}r>rr)r~rrr)rrrr>r}r>rr)2 __future__rror2 contextlibr dataclassesr importlibrpathlibrtextwraprtypingrr _griffe.agents.inspectorr _griffe.agents.visitorr _griffe.collectionsr _griffe.loaderr _griffe.modelsrrcollections.abcrrrr_griffe.docstrings.parsersr_griffe.enumerationsr_griffe.extensions.baserr4rr7rWrmrsryr|rrrr"r"r"r#s              0@F-3