a _g@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__q/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/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|jtjurNt|jj}|j rH||j |j <d|_ dSr) rvaluerConstantstrr AnnAssigninspectcleandocr r )r r 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|ur|j}zt|\}}WntyJYn0t||dd}t|}zt |}Wnt yYnH0|j d}t |tj r|jdkr|j d}t |tjr|j|jkr|S|j}qdS)NrZdedent_workaround)r currentframe getmodulef_lineno findsourceOSErrorgetblockr/rparse SyntaxErrorbodyr FunctionDefnameClassDefr"f_back) r0framelnumlines_Z block_linesr.Z block_treestmtrrr_extract_source_from_frame3s(    rDFboolzdict[str, str])r0 use_inspectrcCsd|r0zt|\}}Wq8ty,iYS0nt|}|s@iSt|}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. ) rgetsourcelinesr6rDr/rrrr8r )r0rFr&rBr.visitorrrrextract_docstrings_from_clsQs   rI)F) __doc__ __future__rrrr*typingr NodeVisitorrr/rDrIrrrrs