o is@sddlmZddlZddlZddlmZddlmZmZddl m Z m Z ddl m Z mZmZddlmZddlmZdd lmZdd lmZdd lmZmZdd lmZmZdd lmZm Z m!Z!ddl"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)ddl*m+Z+m,Z,ddl-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9erddl:m;Z;ddlZ>ddlm?Z?ddddZ@ dhdhddhddhdhdhdZA ddhZB dddddddd6d2d3ZCGd4d5d5ZDdS)7) annotationsN)suppress) TYPE_CHECKINGFinal)get_instance_names get_names) ast_childrenast_kindast_next) get_docstring)safe_get__all__)relative_to_absolute)get_parameters)LinesCollectionModulesCollection)KindTypeParameterKind)AliasResolutionErrorCyclicAliasError LastNodeError)ExprExprNamesafe_get_annotationsafe_get_base_classsafe_get_class_keywordsafe_get_conditionsafe_get_expression) Extensionsload_extensions) Alias AttributeClass Decorator DocstringFunctionModule Parameter Parameters TypeAlias TypeParameterTypeParameters)Path)DocstringOptionsDocstringStyle)Parserproperty staticmethod classmethod)r/r0r1abstractmethodcached dataclass)zabc.abstractmethodzfunctools.cachezfunctools.cached_propertyzcached_property.cached_propertyzfunctools.lru_cachezdataclasses.dataclassztyping.overloadztyping_extensions.overload) extensionsparentdocstring_parserdocstring_optionslines_collectionmodules_collection module_namestrfilepathr+coder5Extensions | Noner6 Module | Noner7DocstringStyle | Parser | Noner8DocstringOptions | Noner9LinesCollection | Noner:ModulesCollection | Nonereturnr%c Cs$t||||pt|||||d S)aParse and visit a module file. We provide this function for static analysis. It uses a [`NodeVisitor`][ast.NodeVisitor]-like class, the [`Visitor`][griffe.Visitor], to compile and parse code (using [`compile`][]) then visit the resulting AST (Abstract Syntax Tree). Important: This function is generally not used directly. In most cases, users can rely on the [`GriffeLoader`][griffe.GriffeLoader] and its accompanying [`load`][griffe.load] shortcut and their respective options to load modules using static analysis. Parameters: module_name: The module name (as when importing [from] it). filepath: The module file path. code: The module contents. 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. Returns: The module, with its members populated. )r7r8r9r:)Visitorr get_module) r;r=r>r5r6r7r8r9r:rHe/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/griffe/_internal/agents/visitor.pyvisitPs& rJcsTeZdZUdZ     dhdifdd Zdddjdd Zejd!krZdrdAdBZdsdtdFdGZdudIdJZdvdLdMZejd!krdwdOdPZdxdRdSZdydUdVZ dsdzdZd[Z d{d]d^Z!d|d`daZ"d}dcddZ#d~dfdgZ$Z%S)rFzoThis class is used to instantiate a visitor. Visitors iterate on AST nodes to extract data from them. Nr;r<r=r+r>r5rr6r@r7rAr8rBr9rCr:rDrENonec stt||_ ||_ ||_ ||_ ||_ d|_ ||_ |p$i|_ |p+t |_ | p2t |_ d|_dS)a$Initialize the visitor. Parameters: module_name: The module name. filepath: The module filepath. code: The module source code. extensions: The extensions to use when visiting. parent: An optional parent for the final module object. docstring_parser: The docstring parser to use. docstring_options: Docstring parsing options. lines_collection: A collection of source code lines. modules_collection: A collection of modules. NF)super__init__r;r=r>r5r6currentr7r8rr9rr: type_guarded) selfr;r=r>r5r6r7r8r9r: __class__rHrIrMs.    zVisitor.__init__Fstrictnodeast.ASTrTboolDocstring | NonecCs4t||d\}}}|durdSt||||j|jdS)NrS)lineno endlinenoparserparser_options)r r#r7r8)rPrUrTvaluerYrZrHrHrI_get_docstringszVisitor._get_docstring) z4Final[dict[type[ast.type_param], TypeParameterKind]]_type_parameter_kind_mapscopeEast.FunctionDef | ast.AsyncFunctionDef | ast.ClassDef | ast.TypeAliasrc str | Nonelist[TypeParameter]csfdd|jDS)Nc sPg|]$}t|jjt|tt|ddjdtt|ddjddqS)boundNr6member default_value)kindrgdefault)r)nameratypergetattrrN).0 type_paramrcrPrHrI s   z0Visitor._get_type_parameters..) type_paramsrPrUrcrHrrrI_get_type_parameterss  zVisitor._get_type_parameters5ast.FunctionDef | ast.AsyncFunctionDef | ast.ClassDefcCsgSNrHrurHrHrIrvsr%cCs.t|jdt|jtjdd}|||jjS)zBuild and return the object representing the module attached to this visitor. This method triggers a complete visit of the module nodes. Returns: A module instance. exec)modefilenameflagsoptimize) compiler>r<r=ast PyCF_ONLY_ASTrJrNmodule)rPtop_noderHrHrIrGs zVisitor.get_modulecCs t|dt||j|dS)ziExtend the base visit with extensions. Parameters: node: The node to visit. visit_N)ror generic_visitrPrUrHrHrIrJs z Visitor.visitcCst|D]}||qdS)zqExtend the base generic visit with extensions. Parameters: node: The node to visit. N)rrJ)rPrUchildrHrHrIr s  zVisitor.generic_visit ast.Modulec Cs|jjd||d|jjd||dt|j|j|j|||j|jdd|_ }|jjd|||d|jjd|||d | ||jjd |||d|jjd |||d d S) zWVisit a module node. Parameters: node: The node to visit. on_noderUagenton_module_nodestatic)rmr=r6 docstringr9r:analysis on_instancerUobjron_module_instance)rUmodr on_memberson_module_membersN) r5callr%r;r=r6r^r9r:rNr)rPrUrrHrHrI visit_modules    zVisitor.visit_module ast.ClassDefc sRjjddjjddg}jr+jdj}|fddjDnj}fddjD}fd d jD}tj|j  |t j jd ||j d d }|j|O_jj||_jjd|djjd|djjd|djjd|djj_dS)zaVisit a class definition node. Parameters: node: The node to visit. rr on_class_noderc3s.|]}tt|jdd|j|jdVqdS)Fr6 parse_stringsrYrZN)r"rrNrY end_lineno)rpdecorator_noderPrHrI 9s z)Visitor.visit_classdef..csg|] }t|jjdqS)rh)rrNrm)rpbaserUrPrHrIrsEsz*Visitor.visit_classdef..cs*i|]}|jdur|jt|jjdqS)Nr6)argrr]rN)rpkwrrHrI Fs$z*Visitor.visit_classdef..rbr) rmrYrZr decoratorstype_parametersbaseskeywordsruntimerrron_class_instance)rUclsrron_class_membersN)r5rdecorator_listrYextendrrr!rmrr^r*rvrOlabelsdecorators_to_labelsrN set_memberrr6)rPrUrrYrrclass_rHrrIvisit_classdef,sD      zVisitor.visit_classdefrlist[Decorator]set[str]cCsFt}|D]}|j}|tvr|t|q|tvr |t|O}q|S)zBuild and return a set of labels based on decorators. Parameters: decorators: The decorators to check. Returns: A set of labels. )set callable_pathbuiltin_decoratorsaddstdlib_decorators)rPrr decoratorrrHrHrIras  zVisitor.decorators_to_labelsfunctionr$c Csj|D]0}z |jdd\}}Wn tyYqw|dvo+||jko+|j|jd}|r2|SqdS)a<Check decorators to return the base property in case of setters and deleters. Parameters: decorators: The decorators to check. Returns: base_property: The property for which the setter/deleted is set. property_function: Either `"setter"` or `"deleter"`. .rz>setterdeleterr/N)rrsplit ValueErrorpathrN get_memberrm has_labels)rPrrrr prop_functionproperty_setter_or_deleterrHrHrIget_base_propertyss  zVisitor.get_base_property&ast.AsyncFunctionDef | ast.FunctionDefr set | Nonec sjjddjjdd|pt}g}d}jrMjdj}jD]$}t|jdd}|dur6q't||j|jd}| |||j t vO}q'nj}| |O}d |vrt jdtjjjd jjj d d } | j|O_jj| jjd | djjd| ddStfddtjD} tj|j| tjjjd |tjjdj jd d } || } |rjj| j | nP| r jjj} | dkr| | _| jdn8| dkr | | _ | jdn)jj| jj!t"j#t"j$hvr5jj| jr5jj| j| _jj| j=| j|O_jjd | djjd| djj!t"j$urm| jdkro| _%jj&_dSdSdS)zHandle a function definition node. Parameters: node: The node to visit. labels: Labels to add to the data object. rron_function_nodeFrrNrr/rhrrmr] annotationrYrZrrrrron_attribute_instancerUattrrc sLg|]"\}}}}t||t|jjdt|tr|nt|jdddqS)rhFr)rkrrl)r&rrNrm isinstancer<r)rprmrrkrlrrHrIrss z+Visitor.handle_function..rb) rmrYrZ parametersreturnsrrrrr6rrwritabler deletableon_function_instance)rUfuncrrM)'r5rrrrYrrNr"rappendrtyping_overloadrr rmrrr^rOrrr'rargsr$r*rvr overloadsmembersrrrrkrMODULECLASSrr6)rPrUrroverloadrYrdecorator_valuer attributerrproperty_function base_propertyrHrrIhandle_functions           &  zVisitor.handle_functionast.FunctionDefcC||dS)zdVisit a function definition node. Parameters: node: The node to visit. NrrrHrHrIvisit_functiondefzVisitor.visit_functiondefast.AsyncFunctionDefcCs|j|dhddS)zkVisit an async function definition node. Parameters: node: The node to visit. async)rNrrrHrHrIvisit_asyncfunctiondefszVisitor.visit_asyncfunctiondef ast.TypeAliasc Cs|jjd||d|jjd||d|jj}t|j|j|d}z |jt|dd}Wn t t fy7d}Ynwt |||j |j t|j||d||jd d }|j|||jjd |||d |jjd |||ddS)zgVisit a type alias node. Parameters: node: The node to visit. rron_type_alias_noderhTrSNrbr)rmr]rYrZrrr6rrron_type_alias_instance)rU type_aliasr)r5rrmidrr]rNr^r rAttributeErrorr(rYrr*rvr)rPrUrmr]rrrHrHrIvisit_typealiass, zVisitor.visit_typealias ast.Importc Cs|jD]A}|jr |jn|jddd}|jp|ddd}||jj|<t|||j|j|j dd}|j |||j j d|||dqdS) zXVisit an import node. Parameters: node: The node to visit. rrzrrrYrZrron_alias_instancealiasrUrN) namesasnamermsplitrNimportsrrYrrOrr5rrPrUrm alias_path alias_namerrHrHrI visit_import,s  zVisitor.visit_importast.ImportFromc Cs|jD]c}|js|jdkr|js|jjjrqt|||jj}|jdkr0|dd}|dd}n |jp5|j}||jj |<||jj d|krft |||j |j |j dd}|j|||jjd |||d qd S) z_Visit an "import from" node. Parameters: node: The node to visit. rz*r/z.*rrrrN)rrlevelrrNis_init_moduler rmreplacerrrrYrrOrr5rrrHrHrIvisit_importfromAs,      zVisitor.visit_importfromast.Assign | ast.AnnAssignrstr | Expr | Nonec Cs|jjd||d|jjd||d|j}t}d}|jtjur7zt|}Wn ty0YdSw| dnh|jtj urszt|}Wn tyMYdSwt |t r_|j r_|j}| dn@|jrm| d| dn2| dn,|jtjur|jdkrdSzt|}Wn tyYdSw|j}|durdS| d|sdSt|j|jd d }z |jt|d d }Wn ttfyd}Ynw|D]}d |vrq||jvr&t |jtjtjfrq|j|} ttt4|| j O}| j!r|s| j!}tt| j"r|s| j"}Wdn 1swYWdn 1s!wYt#||||j$|j%||j& dd} | j |O_ |'|| |dkrittddt(||jD|_)Wdn 1sdwY|jjd|| |d|jjd|| |dqdS)zHandle an attribute (assignment) node. Parameters: node: The node to visit. annotation: A potential annotation. rron_attribute_nodeNzmodule-attributezclass-attributezinstance-attributerMFrTrSrrr__all__cS*g|]}t|tr |nt|j|jdqSrrr<rrmr6rprmrHrHrIrsz,Visitor.handle_attribute..rrrr)*r5rrNrrkrrrKeyErrorrrrr is_classvarslicer]FUNCTIONrmrr6rr^r rrrrIf ExceptHandlerrrrrrrr rYrrOrr exports) rPrUrr6rrr]rrmexisting_memberrrHrHrIhandle_attributees                             zVisitor.handle_attribute ast.AssigncCr)z\Visit an assignment node. Parameters: node: The node to visit. N)rrrHrHrI visit_assignrzVisitor.visit_assign ast.AnnAssigncCs||t|j|jddS)zfVisit an annotated assignment node. Parameters: node: The node to visit. rN)rrrrNrrHrHrIvisit_annassignszVisitor.visit_annassign ast.AugAssigncCstt4|jjdko|jjot|jtj }|r0|jj ddt ||jDWddSWddS1s;wYdS)zfVisit an augmented assignment node. Parameters: node: The node to visit. rcSrrrrrHrHrIrsrz+Visitor.visit_augassign..N) rrtargetrrN is_moduleroprAddrrr )rPrU all_augmentrHrHrIvisit_augassigns    "zVisitor.visit_augassignast.IfcCsLt|jtjtjfrt|j|jdd}t|dvrd|_ | |d|_ dS)zVVisit an "if" node. Parameters: node: The node to visit. N)r6 log_level>typing.TYPE_CHECKINGrTF) rr6rr%ClassDefrtestrNr<rOr)rPrU conditionrHrHrIvisit_ifs    zVisitor.visit_if)NNNNN)r;r<r=r+r>r<r5rr6r@r7rAr8rBr9rCr:rDrErK)rUrVrTrWrErX)rUrdrcrerErf)rUrwrcrerErf)rEr%)rUrVrErK)rUrrErK)rUrrErK)rrrEr)rrrr$rErerx)rUrrrrErK)rUrrErK)rUrrErK)rUrrErK)rUrrErK)rUrrErK)rUr rr rErK)rUrrErK)rUrrErK)rUr!rErK)rUr(rErK)&__name__ __module__ __qualname____doc__rMr^sys version_inforTypeVarrtype_var TypeVarTupletype_var_tuple ParamSpec param_specra__annotations__rvrGrJrrrrrrrrrrr rrr r'r. __classcell__rHrHrQrIrFsL  <        5   j   $ ' k  rF)r;r<r=r+r>r<r5r?r6r@r7rAr8rBr9rCr:rDrEr%)E __future__rrr3 contextlibrtypingrr)griffe._internal.agents.nodes.assignmentsrr!griffe._internal.agents.nodes.astrr r (griffe._internal.agents.nodes.docstringsr %griffe._internal.agents.nodes.exportsr %griffe._internal.agents.nodes.importsr (griffe._internal.agents.nodes.parametersrgriffe._internal.collectionsrrgriffe._internal.enumerationsrrgriffe._internal.exceptionsrrrgriffe._internal.expressionsrrrrrrr griffe._internal.extensions.baserrgriffe._internal.modelsrr r!r"r#r$r%r&r'r(r)r*pathlibr+#griffe._internal.docstrings.parsersr,r-r.rrrrJrFrHrHrHrIsX      $ 8   3