U Qh@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__I./venv/lib/python3.8/site-packages/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) rvaluerZConstantstrr Z AnnAssigninspectZcleandocr 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)rZ currentframeZ getmodulef_linenoZ findsourceOSErrorZgetblockr+rparse SyntaxErrorZbodyrZ FunctionDefnameZClassDefrf_back) r,framelnumlines_Z block_linesr*Z block_treeZstmtrrr_extract_source_from_frame3s(  r8Fboolzdict[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. ) rZgetsourcelinesr/r8r+rrrr0r )r,r:r"r7r*Zvisitorrrrextract_docstrings_from_clsQs  r;)F) __doc__Z __future__rrrr&typingrZ NodeVisitorrr+r8r;rrrrs