U h@s~dZddlmZddlZddlZddlZddlmZGdddejZ ddd d d Z d d dddZ dd dddddZ dS)z4Utilities related to attribute docstring extraction.) annotationsN)AnycsVeZdZddfdd Zdddfdd Zd ddd d Zd ddddZZS)DocstringVisitorNone)returncs td|_i|_d|_dSN)super__init__targetattrsprevious_node_type)self __class__G/tmp/pip-unpacked-wheel-y076a95i/pydantic/_internal/_docs_extraction.pyr s zDocstringVisitor.__init__zast.ASTr)nodercst|}t||_|Sr)rvisittyper )r rZ node_resultrrrrs  zDocstringVisitor.visitz ast.AnnAssigncCst|jtjr|jj|_dSr) isinstancer astNameid)r rrrrvisit_AnnAssignsz DocstringVisitor.visit_AnnAssignzast.ExprcCsRt|jtjrNt|jjtrN|jtjkrNt|jj}|j rH||j |j <d|_ dSr) rvaluerConstantstrr Z AnnAssigninspectcleandocr r )r rZ docstringrrr visit_Exprs    zDocstringVisitor.visit_Expr)__name__ __module__ __qualname__r rrr __classcell__rrrrr srz list[str]r)sourcercCs(td|}|dr$d|}|S)N)  zdef dedent_workaround(): )textwrapdedentjoin startswith)r$ dedent_sourcerrr_dedent_source_lines(s  r-z type[Any]zlist[str] | None)clsrc Cst}|rt|t|kr|j}zt|\}}Wntk rLYnXt||dd}t|}zt |}Wnt k rYnHX|j d}t |tj r|jdkr|j d}t |tjr|j|jkr|S|j}qdS)NrZdedent_workaround)r currentframe getmodulef_lineno findsourceOSErrorgetblockr-rparse SyntaxErrorbodyrZ FunctionDefnameZClassDefr f_back) r.framelnumlines_Z block_linesr,Z block_treeZstmtrrr_extract_source_from_frame3s(  r?Fboolzdict[str, str])r. use_inspectrcCsf|r2zt|\}}Wq:tk r.iYSXnt|}|sBiSt|}t}|t||j S)aXMap model attributes and their corresponding docstring. Args: cls: The class of the Pydantic model to inspect. use_inspect: Whether to skip usage of frames to find the object and use the `inspect` module instead. Returns: A mapping containing attribute names and their corresponding docstring. ) rgetsourcelinesr4r?r-rrrr6r )r.rAr$r>r,Zvisitorrrrextract_docstrings_from_clsQs  rC)F) __doc__ __future__rrrr(typingrZ NodeVisitorrr-r?rCrrrrs