o hYA@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"W/var/www/html/openai_agents/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]]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_implicitsearch_sys_pathrXExtensions | NonerYDocstringStyle | Parser | NonerZDocstringOptions | Noner[LinesCollection | Noner\ModulesCollection | Noner]r^r_r` bool | NonerarbIterator[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. r8F search_pathsrXrYrZr[r\r]r^r_r`raforce_inspectionN)sysrrWr rr)r;r<r9r:rXrYrZr[r\r]r^r_r`rarbrk tmp_packager"r"r#temporary_visited_packagexs&1  "roc 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. r8TrjrCN) rmrrWr rrtupler<keys startswithpopr)r;r<r9r:rXrYrZr[r\r]r^r_r`rarbrkrnrr"r"r#temporary_inspected_packages<1   "rt)r,rXparentrYrZr[r\ru 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-rXrurYrZr[r\N)rr7r splitlinesr r\) r-r,rXrurYrZr[r\_rr*r"r"r#temporary_visited_modules&   "rz)r, import_pathsrXrurYrZr[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+)rwr{rXrurYrZr[r\N)r7r rxr r\rmr<r) r-r,r{rXrurYrZr[r\ryrr*r"r"r#temporary_inspected_module4s2      "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~rtopleafobjr"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_packager~rrr 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. rCc s2g|]\}}t|tgd|dRdqS)NrD)rw)r r).0indexrrJr"r# s2z module_vtree..r~)split enumeraterw with_stem _filepathr)rrr~r<rwr"rr# module_vtrees  r)r-rr,rrr.r) r;rr<r=r9r>r:r>rr?) r;rr<r=r9r>r:r>rXrcrYrdrZrer[rfr\rgr]r>r^r>r_r>r`rhrar>rbr>rri)r-rr,rrXrcrurvrYrdrZrer[rfr\rgrri)r-rr,rr{r|rXrcrurvrYrdrZrer[rfr\rgrri)rrr~r>rr)rrrr)rrrr>r~r>rr )2 __future__rrmr2 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.baserr4rr7rWrortrzr}rrrr"r"r"r#s              0CI-3