a ¢º”h‹ã@s<ddlZddlmZddlmZGdd„dejƒZejZdS)éN)ÚOptionalé)Ú _resolve_namecs€eZdZdZeeeeeeeefdœdd„ƒZ ‡fdd„Z ee edœdd „Z d d „Z d d „Zdd„Zdd„Zdd„Z‡ZS)Ú_ExtractModuleReferenceszR Extract the list of global variables a block of code will read and write )ÚsrcÚpackageÚreturncCs*||ƒ}t |¡}| |¡t|j ¡ƒS©N)ÚastÚparseÚvisitÚlistÚ referencesÚkeys)ÚclsrrÚvisitorÚtree©rúR/var/www/auris/lib/python3.9/site-packages/torch/package/find_file_dependencies.pyÚrun s  z_ExtractModuleReferences.runcstƒ ¡||_i|_dSr )ÚsuperÚ__init__rr)Úselfr©Ú __class__rrrs z!_ExtractModuleReferences.__init__)Ú module_nameÚlevelrcCs|dkrt||j|ƒS|S)Nr)rr)rrrrrrÚ _absmodulesz#_ExtractModuleReferences._absmodulecCs |jD]}d|j|jdf<qdS)NT)ÚnamesrÚname)rÚnodeÚaliasrrrÚ visit_Imports z%_ExtractModuleReferences.visit_ImportcCsX| |j|jdurdn|j¡}|jD].}|jdkrDd|j||jf<q$d|j|df<q$dS)NrÚ*T)rÚmodulerrrr)rr rr!rrrÚvisit_ImportFrom"s   z)_ExtractModuleReferences.visit_ImportFromcCs|jSr ©Úvalue©rr rrrÚ_grab_node_int-sz'_ExtractModuleReferences._grab_node_intcCs|jSr r&r(rrrÚ_grab_node_str0sz'_ExtractModuleReferences._grab_node_strcsžt|jdƒrš|jjdkršzdˆ |jd¡}g}d}t|jƒdkrh| ‡fdd„|jdjDƒ¡n;óz6_ExtractModuleReferences.visit_Call..ÚkeywordsÚfromlistc3s|]}ˆ |¡VqdSr r.r/r2rrr3?sérTÚ.r)Úmaxsplitr#)ÚhasattrÚfuncr+r*ÚargsÚlenÚextendÚeltsr5Úargr'r)rÚrsplitrÚ Exception)rr rr6rÚkeywordZtop_namer!rr2rÚ visit_Call3s@"   ÿ         z#_ExtractModuleReferences.visit_Call)Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú classmethodÚstrr ÚtuplerrrÚintrr"r%r)r*rDÚ __classcell__rrrrrs$  r) r ÚtypingrZ _importlibrÚ NodeVisitorrrZfind_files_source_depends_onrrrrÚs  X