o <Æ&iÒã@sDddlZddlmZmZmZddlmZGdd„dejƒZej Z dS)éN)ÚListÚOptionalÚTupleé)Ú _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úOC:\wamp64\www\opt\env\Lib\site-packages\torch/package/find_file_dependencies.pyÚrun s  z_ExtractModuleReferences.runcstƒ ¡||_i|_dSr )ÚsuperÚ__init__r r)Ú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Úmodulerr r!r)rr"r!r#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_str/r-z'_ExtractModuleReferences._grab_node_strc Cs´t|jdƒrÖ|jjdkrØz¶| |jd¡}g}d}t|jƒdkr2|jdjD] }| | |¡¡q&nt|dƒrQ|jD]}|j dkrP|j jD] }| | |¡¡qEq:t|jƒdkra|  |jd¡}nt|dƒrw|jD] }|j dkrv|  |j ¡}qi|gkr¢d |j |df<|j d d d d}||krŸ| ||¡}d |j |df<WdSWdS| ||¡}|D]}|d kr¸d |j ||f<qªd |j |df<qªWdStyÕ} zWYd} ~ dSd} ~ wwdSdS)NÚidÚ __import__réÚkeywordsÚfromlistérTÚ.r)Úmaxsplitr%)ÚhasattrÚfuncr/r.ÚargsÚlenÚeltsÚappendr2Úargr*r,rÚrsplitrÚ Exception) rr"r!r3rÚvÚkeywordZtop_namer#ÚerrrÚ visit_Call2sPÿ    €    € þ ù€ÿÙz#_ExtractModuleReferences.visit_Call)Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú classmethodÚstrrrrrrÚintrr$r'r,r.rCÚ __classcell__rrrrrs(  r) r ÚtypingrrrZ _importlibrÚ NodeVisitorrrZfind_files_source_depends_onrrrrÚs   X