o ~j6h@sjdZddlmZddlZddlZddlZddlmZGdddejZ dd d Z dddZ ddddZ dS)z4Utilities related to attribute docstring extraction.) annotationsN)Anycs@eZdZdfdd Zdfdd Zdd d ZdddZZS)DocstringVisitorreturnNonecs td|_i|_d|_dSN)super__init__targetattrsprevious_node_type)self __class__r/var/www/html/chefvision.cloud.itp360.com/venv/lib/python3.10/site-packages/pydantic/_internal/_docs_extraction.pyr s  zDocstringVisitor.__init__nodeast.ASTrcst|}t||_|Sr)rvisittyper )r r node_resultrrrrs  zDocstringVisitor.visit ast.AnnAssigncCs t|jtjr|jj|_dSdSr) isinstancer astNameid)r rrrrvisit_AnnAssignsz DocstringVisitor.visit_AnnAssignast.ExprcCs^t|jtjr)t|jjtr+|jtjur-t|jj}|j r$||j |j <d|_ dSdSdSdSr) rvaluerConstantstrr AnnAssigninspectcleandocr r )r r docstringrrr visit_Exprs     zDocstringVisitor.visit_Expr)rr)rrrr)rrrr)rrrr)__name__ __module__ __qualname__r rrr% __classcell__rrrrr s  rsource list[str]rr cCs(td|}|drd|}|S)N)  zdef dedent_workaround(): )textwrapdedentjoin startswith)r* dedent_sourcerrr_dedent_source_lines(s  r4cls type[Any]list[str] | Nonec Cst}|rpt|t|uri|j}z t|\}}Wn ty%YnDwt||dd}t|}zt |}Wn t yEYn$w|j d}t |tj r[|jdkr[|j d}t |tjri|j|jkri|S|j}|sdSdS)Nrdedent_workaround)r" currentframe getmodulef_lineno findsourceOSErrorgetblockr4rparse SyntaxErrorbodyr FunctionDefnameClassDefr&f_back) r5framelnumlines_ block_linesr3 block_treestmtrrr_extract_source_from_frame3s.     rNF use_inspectbooldict[str, str]cCsb|rz t|\}}WntyiYSwt|}|siSt|}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. ) r"getsourcelinesr>rNr4rrrr@r )r5rOr*rJr3visitorrrrextract_docstrings_from_clsQs  rT)r*r+rr )r5r6rr7)F)r5r6rOrPrrQ) __doc__ __future__rrr"r/typingr NodeVisitorrr4rNrTrrrrs