o 2g@sdZddlmZddlZddlZddlZddlZddlZddlZddl m Z m Z ddl m Z GdddZd#d d Zd$ddZGdddeZd%ddZd&dd ZGd!d"d"ZdS)'zBasic introspection of modules.) annotationsN)Queue get_context) ModuleTypec@s$eZdZ      ddddZdS)ModulePropertiesNFnamestrfile str | Nonepathlist[str] | Noneall is_c_modulebool subpackagesreturnNonecCs,||_||_||_||_||_|pg|_dSNrr r rrr)selfrr r rrrrJ/home/garg/my-data/venv/lib/python3.10/site-packages/mypy/moduleinspect.py__init__s  zModuleProperties.__init__)rNNNFN)rr r r r r rr rrrr rr)__name__ __module__ __qualname__rrrrrrsrmodulerrrcCs.|jddur dStj|jdddvS)N__file__T)z.soz.pydz.dll)__dict__getosr splitext)rrrrr"srr r cCs(t|o|dotj|dd S)Nz.pycr)rendswithr"r exists)r rrr is_pyc_only*s(r&c@s eZdZdS) InspectErrorN)rrrrrrrr'.sr' package_idr c szt|Wnty}ztt||d}~wwtd|}tdd}tdd}t|ts4d}tdd}|durPzt|}Wn tyOd}Ynwt }|durj|rgfddt D}ng}nt j |jdd d d }d d|D}t||||||d S)zDUse runtime introspection to get information about a module/package.Nrr__path____all__cs<g|]\}}t|r|jjd|krjd|qS).)inspectismoduler).0rvalpackagerr Ks z*get_package_properties..r+cSsdSrr)rrrrUsz(get_package_properties..)prefixonerrorcSsg|]\}}}|qSrr)r.importerqualified_nameispkgrrrr2Wsr) importlib import_module BaseExceptionr'r getattr isinstancelist Exceptionrr, getmemberspkgutil walk_packagesrr) r(err r pkg_allis_cr all_packagesrr0rget_package_properties2s@         rHtasks Queue[str]resultsQueue[str | ModuleProperties]sys_path list[str]rc Cs\|t_ |}zt|}Wnty'}z |t|WYd}~qd}~ww||q)z0The main loop of a worker introspection process.TN)sysr r!rHr'putr )rIrKrMmodproprDrrrworker]s   rSc@sVeZdZdZdddZdddZddd Zdd dZdddZdddZ dddZ dS) ModuleInspecta9Perform runtime introspection of modules in a separate process. Reuse the process for multiple modules for efficiency. However, if there is an error, retry using a fresh process to avoid cross-contamination of state between modules. We use a separate process to isolate us from many side effects. For example, the import of a module may kill the current process, and we want to recover from that. Always use in a with statement for proper clean-up: with ModuleInspect() as m: p = m.get_package_properties('urllib.parse') rrcC |dSr)_startrrrrrz zModuleInspect.__init__cCs`tjdkr td}ntd}||_||_|jt|j|jtjfd|_ |j d|_ dS)Nlinux forkserverspawn)targetargsr) rOplatformrrrIrKProcessrSr procstartcounter)rctxrrrrV}s      zModuleInspect._startcCs|jdS)zFree any resources used.N)r` terminaterWrrrcloseszModuleInspect.closer(r rcCsz|j||}|dur|td|t|tr4|jdkr0||| |St||jd7_|S)zReturn some properties of a module/package using runtime introspection. Raise InspectError if the target couldn't be imported. NzProcess died when importing r) rIrP_get_from_queuerVr'r>r rbrerH)rr(resrrrrHs    z$ModuleInspect.get_package_propertiesModuleProperties | str | NonecCsZd}d} ||kr tdz|jjddWStjy'|js%YdSYnw|d7}q) z{Get value from the queue. Return the value read from the queue, or None if the process unexpectedly died. iXrTzTimeout waiting for subprocessg?)timeoutNrf) RuntimeErrorrKr!queueEmptyr`is_alive)rmax_iternrrrrgs zModuleInspect._get_from_queuecCs|SrrrWrrr __enter__szModuleInspect.__enter__r]objectcGrUr)re)rr]rrr__exit__rXzModuleInspect.__exit__N)rrr(r rr)rri)rrT)r]rrrr) rrr__doc__rrVrerHrgrqrsrrrrrTjs     rT)rrrr)r r rrrt)rIrJrKrLrMrNrr)ru __future__rr:r,r"rBrlrOmultiprocessingrrtypesrrrr&r@r'rHrSrTrrrrs"      +