o 0 h@s<ddlZddlmZddlmZGdddejZejZdS)N)Optional) _resolve_namec seZdZdZedededeeeeeffddZ fddZ d ed e defd d Z d dZ ddZddZddZddZZS)_ExtractModuleReferenceszR Extract the list of global variables a block of code will read and write srcpackagereturncCs*||}t|}||t|jSN)astparsevisitlist referenceskeys)clsrrvisitortreerX/var/www/vscode/kcb/lib/python3.10/site-packages/torch/package/find_file_dependencies.pyrun s  z_ExtractModuleReferences.runcst||_i|_dSr )super__init__rr)selfr __class__rrrs  z!_ExtractModuleReferences.__init__ module_namelevelcCs|dkr t||j|S|S)Nr)rr)rrrrrr _absmodulesz#_ExtractModuleReferences._absmodulecCs |jD] }d|j|jdf<qdS)NT)namesrname)rnodealiasrrr visit_Imports z%_ExtractModuleReferences.visit_ImportcCsX||j|jdur dn|j}|jD]}|jdkr"d|j||jf<qd|j|df<qdS)Nr*T)rmodulerrrr)rr rr!rrrvisit_ImportFrom"s   z)_ExtractModuleReferences.visit_ImportFromcC|jSr valuerr rrr_grab_node_int-z'_ExtractModuleReferences._grab_node_intcCr&r r'r)rrr_grab_node_str0r+z'_ExtractModuleReferences._grab_node_strcst|jdr|jjdkrz|jd}g}d}t|jdkr1|fdd|jdjDnt|drO|jD]}|j dkrN|fd d|j jDq9t|jd kr_ |jd }nt|dru|jD] }|j d krt |j }qg|gkrd j |df<|j d ddd}||kr||}d j |df<WdSWdS||}|D]}|dkrd j ||f<qd j |df<qWdStyYdSwdSdS)Nid __import__rc3|]}|VqdSr r,.0vrrr ;sz6_ExtractModuleReferences.visit_Call..keywordsfromlistc3r0r r1r2r5rrr6?s  rT.r)maxsplitr#)hasattrfuncr-r,argslenextendeltsr7argr(r*rrsplitr Exception)rr rr8rkeywordtop_namer!rr5r visit_Call3sL"          z#_ExtractModuleReferences.visit_Call)__name__ __module__ __qualname____doc__ classmethodstrr tuplerrrintrr"r%r*r,rG __classcell__rrrrrs(  r) r typingr _importlibr NodeVisitorrrfind_files_source_depends_onrrrrs    X