U ={f@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__b/var/www/html/Darija-Ai-API/env/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) rvaluerConstantstrr 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 SyntaxErrorbodyr FunctionDefnameClassDefr!f_back) r/framelnumlines_Z block_linesr-Z block_treestmtrrr_extract_source_from_frame2s(  rCFboolzdict[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. ) rgetsourcelinesr5rCr.rrrr7r )r/rEr%rAr-visitorrrrextract_docstrings_from_clsPs  rH)F) __doc__ __future__rrrr)typingr NodeVisitorrr.rCrHrrrrs