g9lUddlmZddlZddlZddlmZmZddlmZddl m Z ddl m Z m Z mZddlmZmZddlmZdd lmZe r2ddlZdd lmZdd lmZdd lmZdd lmZddlm Z ddl!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'GddZ(ee)e*e)e fe(e+e(fZ, GddZ-dhZ.de/d< ddZ0 ddZ1ddZ2y)) annotationsN)module_from_specspec_from_file_location)isclass)Path) TYPE_CHECKINGAnyUnion) ast_childrenast_kind)ExtensionNotLoadedError)dynamic_import) ModuleType) Inspector) ObjectNode)Visitor) GriffeLoader)Alias AttributeClassFunctionModuleObjectc`eZdZdZddZddZddZddZddZ ddZ ddZ dd Z dd Z dd Z dd Z dd Z ddZddZ ddZddZ ddZ ddZd dZ d!dZy)" Extensionz!Base class for Griffe extensions.cDt|dt|d|y)zPVisit a node. Parameters: node: The node to visit. visit_cyN_s L/var/www/openai/venv/lib/python3.12/site-packages/_griffe/extensions/base.pyz!Extension.visit..%s4N)getattrr selfnodes r#visitzExtension.visits$ Ax~./@Fr%cFt|D]}|j|y)zhVisit children nodes. Parameters: node: The node to visit the children of. N)r r*r(r)childs r# generic_visitzExtension.generic_visit's "$'E JJu (r%cFt|d|jd|y)zTInspect a node. Parameters: node: The node to inspect. inspect_cyrr r!s r#r$z#Extension.inspect..6sr%N)r&kindr's r#inspectzExtension.inspect0s" > ,n=dCr%cb|jD] }|jr|j|"y)zxExtend the base generic inspection with extensions. Parameters: node: The node to inspect. N)childrenalias_target_pathr3r,s r#generic_inspectzExtension.generic_inspect8s' ]]E** U##r%c y)zRun when visiting a new node during static/dynamic analysis. Parameters: node: The currently visited node. Nr r(r)agentkwargss r#on_nodezExtension.on_nodeBr%c y)aRun when an Object has been created. Parameters: node: The currently visited node. obj: The object instance. agent: The analysis agent currently running. **kwargs: For forward-compatibility. Nr r(r)objr:r;s r# on_instancezExtension.on_instanceIr=r%c y)a Run when members of an Object have been loaded. Parameters: node: The currently visited node. obj: The object instance. agent: The analysis agent currently running. **kwargs: For forward-compatibility. Nr r?s r# on_memberszExtension.on_membersZr=r%c y)zRun when visiting a new module node during static/dynamic analysis. Parameters: node: The currently visited node. agent: The analysis agent currently running. **kwargs: For forward-compatibility. Nr r9s r#on_module_nodezExtension.on_module_nodedr=r%c y)zRun when a Module has been created. Parameters: node: The currently visited node. mod: The module instance. agent: The analysis agent currently running. **kwargs: For forward-compatibility. Nr r(r)modr:r;s r#on_module_instancezExtension.on_module_instancemr=r%c y)a Run when members of a Module have been loaded. Parameters: node: The currently visited node. mod: The module instance. agent: The analysis agent currently running. **kwargs: For forward-compatibility. Nr rGs r#on_module_memberszExtension.on_module_members~r=r%c y)zRun when visiting a new class node during static/dynamic analysis. Parameters: node: The currently visited node. agent: The analysis agent currently running. **kwargs: For forward-compatibility. Nr r9s r# on_class_nodezExtension.on_class_noder=r%c y)zRun when a Class has been created. Parameters: node: The currently visited node. cls: The class instance. agent: The analysis agent currently running. **kwargs: For forward-compatibility. Nr r(r)clsr:r;s r#on_class_instancezExtension.on_class_instancer=r%c y)aRun when members of a Class have been loaded. Parameters: node: The currently visited node. cls: The class instance. agent: The analysis agent currently running. **kwargs: For forward-compatibility. Nr rOs r#on_class_memberszExtension.on_class_membersr=r%c y)zRun when visiting a new function node during static/dynamic analysis. Parameters: node: The currently visited node. agent: The analysis agent currently running. **kwargs: For forward-compatibility. Nr r9s r#on_function_nodezExtension.on_function_noder=r%c y)aRun when a Function has been created. Parameters: node: The currently visited node. func: The function instance. agent: The analysis agent currently running. **kwargs: For forward-compatibility. Nr )r(r)funcr:r;s r#on_function_instancezExtension.on_function_instancer=r%c y)zRun when visiting a new attribute node during static/dynamic analysis. Parameters: node: The currently visited node. agent: The analysis agent currently running. **kwargs: For forward-compatibility. Nr r9s r#on_attribute_nodezExtension.on_attribute_noder=r%c y)aRun when an Attribute has been created. Parameters: node: The currently visited node. attr: The attribute instance. agent: The analysis agent currently running. **kwargs: For forward-compatibility. Nr )r(r)attrr:r;s r#on_attribute_instancezExtension.on_attribute_instancer=r%c y)aRun when an Alias has been created. Parameters: node: The currently visited node. alias: The alias instance. agent: The analysis agent currently running. **kwargs: For forward-compatibility. Nr )r(r)aliasr:r;s r#on_aliaszExtension.on_aliasr=r%c y)zRun when a package has been completely loaded. Parameters: pkg: The package (Module) instance. loader: The loader currently in use. **kwargs: For forward-compatibility. Nr )r(pkgloaderr;s r#on_package_loadedzExtension.on_package_loadedr=r%c y)zRun when wildcard imports are expanded into aliases. Parameters: alias: The alias instance. loader: The loader currently in use. **kwargs: For forward-compatibility. Nr )r(r_rcr;s r#on_wildcard_expansionzExtension.on_wildcard_expansionr=r%N)r)zast.ASTreturnNone)r)rrgrh)r)ast.AST | ObjectNoder:Visitor | Inspectorr;r rgrh) r)rir@rr:rjr;r rgrh) r)rirHrr:rjr;r rgrh) r)rirPrr:rjr;r rgrh) r)rirWrr:rjr;r rgrh) r)rir\rr:rjr;r rgrh) r)rir_rr:rjr;r rgrh)rbrrcrr;r rgrh)r_rrcrr;r rgrh)__name__ __module__ __qualname____doc__r*r.r3r7r<rArCrErIrKrMrQrSrUrXrZr]r`rdrfr r%r#rrs#+GD$  #   #      "   #   #      " #   #      "  #   #      " #   #      "  #   #      "  #   #      " #   #      "           r%rc(eZdZdZddZddZddZy) ExtensionszLThis class helps iterating on extensions that should run at different times.c0g|_|j|y)zrInitialize the extensions container. Parameters: *extensions: The extensions to add. N) _extensionsadd)r( extensionss r#__init__zExtensions.__init__s -/*r%cH|D]}|jj|y)zoAdd extensions to this container. Parameters: *extensions: The extensions to add. N)rrappend)r(rt extensions r#rszExtensions.add(s" $I    # #I .$r%c J|jD]}t||di|y)zCall the extension hook for the given event. Parameters: event: The triggered event. **kwargs: Arguments passed to the hook. Nr )rrr&)r(eventr;rxs r#callzExtensions.call1s())I %GIu % / /*r%N)rtrrgrh)rzstrr;r rgrh)rkrlrmrnrursr{r r%r#rprpsV/0r%rp dataclasseszset[str]builtin_extensionsctjj|jddd}t ||}|st d|dt |}|tj|<|jj||S)N.rz#Could not import module from path '') ospathbasenamersplitrr rsysmodulesrc exec_module)r module_namespecmodules r#_load_extension_pathrBs|''""4(//Q7:K "; 5D %(KD6QR&STT d #F%CKK KKF# Mr%cvd}t|tr|St|rt|tr|St|tr1t t |j\}}t|}n t|}i}|jd}|dkDs|r-dt|jvr|jdd\}}nd}|tvrd|}n+tjj!|r t#|}|s t)|}t|rt|tr|d i|S|r t-||d i|St1|j3Dcgc]*}t|st|ts |tus)|,}}|D cgc] } | d i| c} S#t$$r}t'd|d|d}~wwxYw#t*$r}t'd|d|d}~wt$$r}t'd|d||d}~wwxYw#t.$r}t'd|d |d |d}~wwxYwcc}wcc} w) aLoad a configured extension. Parameters: extension: An extension, with potential configuration options. Raises: ExtensionNotLoadedError: When the extension cannot be loaded, either because the module is not found, or because it does not expose the Extension attribute. ImportError will bubble up so users can see the traceback. Returns: An extension instance. N:rz_griffe.extensions.zExtension module 'z' could not be foundz!Error while importing extension 'z': z ' has no 'z ' attributer ) isinstancerr issubclassdictnextiteritemsr|countrdriverr~rrexistsr ImportErrorr rModuleNotFoundErrorr&AttributeErrorvarsvalues) rx ext_object import_pathoptionscolonsextension_nameerrorr@rtexts r#_load_extensionrMst"J)Y'yjI>{)T"#D):$;< W+& )n   s #F zfD,=,C,C!C&1&8&8a&@# ^((+K=9  $ m-k:J  r' 4Jzz*i@$G$$ 67:~6AA AJ'..00GCLZPSU^E_dgoxdx0'1 1jsCN'Nj 11A m),>{mK_*`agl l m# m),>{mK_*`agl l r),Mk]Z]^c]d*eflq q r )$[MN;K;W   2sr: F- G9H &H17H1H1H1H6- G 6GG  H G'' H 3HH  H.H))H.ct}|D]>}t|}t|tr|j|.|j |@ddlm}|jD]}t||us|S|jtd|S)zLoad configured extensions. Parameters: exts: Extensions with potential configuration options. Returns: An extensions container. r)DataclassesExtensionr}) rprrlistrs_griffe.extensions.dataclassesrrrtype)extsrtrxrrs r#load_extensionsrsJ i( c4 JNNC NN3  D%% 9, ,   &   67 r%)rr|rgr)rxz2str | dict[str, Any] | Extension | type[Extension]rgzExtension | list[Extension])rLoadableExtensionTypergrp)3 __future__rrrimportlib.utilrrr3rpathlibrtypingrr r _griffe.agents.nodes.astr r _griffe.exceptionsr _griffe.importerrasttypesr_griffe.agents.inspectorr_griffe.agents.nodes.runtimer_griffe.agents.visitorr_griffe.loaderr_griffe.modelsrrrrrrrr|rrrrpr~__annotations__rrrr r%r#rs# D,,;6+ 27.+PPy y xc4S>9d9oMN<00B H/X2AX2 X2vr%