o wZŽh‹ã@s<ddlZddlmZddlmZGdd„dejƒZejZdS)éN)ÚOptionalé)Ú _resolve_namec sˆeZdZdZedededeeeeeffdd„ƒZ ‡fdd„Z d ed e defd d „Z d d„Z dd„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úS/var/www/auris/lib/python3.10/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ÚlevelcCs|dkr t||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|jdur dn|j¡}|jD]}|jdkr"d|j||jf<qd|j|df<qdS)NrÚ*T)rÚmodulerrrr)rr rr!rrrÚvisit_ImportFrom"s   ùz)_ExtractModuleReferences.visit_ImportFromcCó|jSr ©Úvalue©rr rrrÚ_grab_node_int-óz'_ExtractModuleReferences._grab_node_intcCr&r r'r)rrrÚ_grab_node_str0r+z'_ExtractModuleReferences._grab_node_strcsžt|jdƒrË|jjdkrÍz´ˆ |jd¡}g}d}t|jƒdkr1| ‡fdd„|jdjDƒ¡nt|dƒrO|jD]}|j dkrN| ‡fd d„|j jDƒ¡q9t|jƒd kr_ˆ  |jd ¡}nt|dƒru|jD] }|j d krtˆ  |j ¡}qg|gkr d ˆj |df<|j d ddd}||krˆ ||¡}d ˆj |df<WdSWdSˆ ||¡}|D]}|dkr¶d ˆj ||f<q¨d ˆj |df<q¨WdStyÊYdSwdSdS)NÚidÚ __import__réc3ó|]}ˆ |¡VqdSr ©r,©Ú.0Úv©rrrÚ ;s€z6_ExtractModuleReferences.visit_Call..ÚkeywordsÚfromlistc3r0r r1r2r5rrr6?s€  ÿérTÚ.r)Úmaxsplitr#)ÚhasattrÚfuncr-r,ÚargsÚlenÚextendÚeltsr7Úargr(r*rÚrsplitrÚ Exception)rr rr8rÚkeywordZtop_namer!rr5rÚ visit_Call3sL"   ÿ€    € þ ù ÿÙz#_ExtractModuleReferences.visit_Call)Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú classmethodÚstrr ÚtuplerrrÚintrr"r%r*r,rFÚ __classcell__rrrrrs(  r) r ÚtypingrZ _importlibrÚ NodeVisitorrrZfind_files_source_depends_onrrrrÚs    X