bMhzdZddlmZddlZddlZddlZddlmZGddejZ dd Z ddZ dddZ dS)z4Utilities related to attribute docstring extraction.) annotationsN)Anyc<eZdZd fd Zd fd Zdd Zdd ZxZS)DocstringVisitorreturnNonecrtd|_i|_d|_dSN)super__init__targetattrsprevious_node_type)self __class__s c/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pydantic/_internal/_docs_extraction.pyr zDocstringVisitor.__init__ s5 "& %' 8<nodeast.ASTrcrt|}t||_|Sr )r visittyper)rr node_resultrs rrzDocstringVisitor.visits,ggmmD)) "&t**r ast.AnnAssigncjt|jtjr|jj|_dSdSr ) isinstancer astNameid)rrs rvisit_AnnAssignz DocstringVisitor.visit_AnnAssigns1 dk38 , , )+.DKKK ) )rast.Exprc*t|jtjrot|jjtrR|jtjurAtj|jj}|j r||j |j <d|_ dSdSdSdSr ) rvaluerConstantstrr AnnAssigninspectcleandocr r)rr docstrings r visit_ExprzDocstringVisitor.visit_Exprs tz3< 0 0 4:+S11 '3=88()9::I{ 4*3 4;'DKKK    98r)rr)rrrr)rrrr)rr!rr)__name__ __module__ __qualname__r rr r* __classcell__)rs@rrr s====== ))))        rrsource list[str]rr%ctjd|}|drd|}|S)N)  zdef dedent_workaround(): )textwrapdedentjoin startswith)r/ dedent_sources r_dedent_source_linesr:(sHOBGGFOO44M ,,EE]DD rcls type[Any]list[str] | Nonecvtj}|r"tj|tj|ur|j} tj|\}}tj||dz d}t |} tj|}|j d}t|tj r|j dkr |j d}t|tj r|j |jkr|Sn #t$rYnwxYw#t $rYnwxYw|j}| dSdS)Nrdedent_workaround)r' currentframe getmodulef_lineno findsourcegetblockr:rparsebodyr FunctionDefnameClassDefr+ SyntaxErrorOSErrorf_back) r;framelnumlines_ block_linesr9 block_treestmts r_extract_source_from_framerU3si  " "E   U # #w'8'='= = =>D +"-e44q &.uTAXZZ/@AA 4[ A A  +!$=!9!9J&?1-D!$88,TYJ]=]=]#y|!$ 55+$)s|:S:S**#D    & 1 s$D D DD D*)D*F use_inspectbooldict[str, str]c|r+ tj|\}}n!#t$ricYSwxYwt|}|siSt |}t }|tj||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'getsourcelinesrLrUr:rrrrFr)r;rVr/rQr9visitors rextract_docstrings_from_clsr\Qs1 .s33IFAA   III ,C00  (00M  G MM#)M**+++ =s  ++)r/r0rr%)r;r<rr=)F)r;r<rVrWrrX) __doc__ __future__rrr'r5typingr NodeVisitorrr:rUr\rrrbs::"""""" s:<r