l j&UdZddlmZddlZddlZddlmZmZmZddlm Z ddl m Z m Z dgZ eeefZded < eeefZded < eeefZded < eeefZded < eeefZded < eeeefZded< eeefZded< iad ed<iaded<iaded<e ejjjZddZ dS)zIRead and parse all the Manim modules and extract documentation from them.) annotationsN) AttributeName Subscript)Path)Any TypeAliasparse_module_attributesr AliasInfoAliasCategoryDictModuleLevelAliasDictModuleTypeVarDict AliasDocsDictDataDict TypeVarDictALIAS_DOCS_DICT DATA_DICT TYPEVAR_DICTreturn+tuple[AliasDocsDict, DataDict, TypeVarDict]cl tststrtttfStdD]}|t}t|j}|d d|d<d |}| d}i}d}d}i}g} d} tj tj|D]} t| tjurt| jtjurt| jjt&ur| jj} d} | | r9| t-| d}i||<||}d}n|r| |d <n| r| | t| tjurt| jtjur| jjd ksjt| jtjurRt| jjtjur-| jjjd kr| jjd kr| j}n| g}|D]&} t>j d kot| tj!u}t| tj"uoXt| j#tjuo8| j#jd ko(t| j$tjuo| jdu}|s|r|r | j%jn | j$j}| j}t|tj&ur\t|jtjur<|jjdkr,|j'j(}d d|D}ntj)|}|*dd}| i|d<|d}d|i||<||}t| tj+urt| j,dtjurt| jtj-urt| jj.tjurf| jj.j/drBtj)| j*dd|| j,dj<td}t| tj"ur| j$}nCt| tj+ur&t-| j,dkr| j,d}nd}t|tjur-t| tj+ur |j|vs |j} $d} (t-|dkr |t|<t-| dkr | t|<|r |t|<tttfS)aRead all files, generate Abstract Syntax Trees from them, and extract useful information about the type aliases defined in the files: the category they belong to, their definition and their description, separating them from the "regular" module attributes. Returns ------- ALIAS_DOCS_DICT : :class:`AliasDocsDict` A dictionary containing the information from all the type aliases in Manim. See :class:`AliasDocsDict` for more information. DATA_DICT : :class:`DataDict` A dictionary containing the names of all DOCUMENTED module-level attributes which are not a :class:`TypeAlias`. TYPEVAR_DICT : :class:`TypeVarDict` A dictionary containing the definitions of :class:`TypeVar` objects, organized by modules. z*.pyz.py.zutf-8)encodingNz [CATEGORY]doc TYPE_CHECKINGtyping) r Unionz | c3>K|]}tj|VdS)N)astunparse).0elems ^/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/utils/docbuild/module_parsing.py z*parse_module_attributes..s=0026CK--000000znpt. definitionrTypeVar_z\_)0rrr MANIM_ROOTrglobresolve relative_tolistparts removesuffixjoin read_textr"iter_child_nodesparsetypeExprvalueConstantstrstrip startswithlenappendIftestridrattrbodysys version_infor AnnAssign annotationtargetnamersliceeltsr#replaceAssigntargetsCallfuncendswith) module_pathmodule_name_t1module_name_t2 module_namemodule_content module_dict category_dict alias_infomodule_typevars data_list data_namenodestring section_str category_name inner_nodes is_type_alias"is_annotated_assignment_with_value alias_namedefinition_nodeunion_elementsr*rKs r&r r >s08)8|8 <77!''//X8X8 $,,..:::FFn233+B/<)B)BCC} %} %DT ch&&$$ 44)**c11)//11* $$[11 0$*3{+;+;+=+=$>$D$D$F$FM13K .$/ $>M!%JJ0(.Ju%%0$$Y///T cf$$TY3833 8W8WTY3=88 11SX== IO.(:: INo==*. #f #I %I %$/ODJJ#-4OJJ#-//T_--9/*k9/T[))SX5/ $. 3!($F(2?!RDKNJ&*jO _-->> !67738CC+14??)8)>)C%*ZZ00:H000&& &)[%A%A !+!3!3FB!?!?J$,*, B(3B 1=z0JM*-!.z!:JJJ#*,,T\!_--99TZ((CH44TZ_--99 *33I>>:;>+ ;;gc5))$DLO$67" ::..MQ[FF$ZZ3:--#dl2C2Cq2H2H!\!_FF!F<<38++JJ#*,,/1Q1Q & II $IISI %V {  a  +6OK ( y>>A  %.Ik "  8(7L % I| 33r()rr)!__doc__ __future__rr"rGrrrpathlibrrrr __all__dictr=r __annotations__r r rrr2rrrrr__file__r0parentr.r r(r&rssOOO"""""" **********!!!!!!!! $ %CH~ %%%% $CN33333#'s,='=">>>>> $CH~----G %9 9: :::: 3S >***** c#445 5555T!#####   T(^^ # # % % , 3 : u4u4u4u4u4u4r(