U e5d/@sdZddlZddlZddlZddlZddlZddlZddlm Z m Z ddgZ Gddde Z Gdd d e Zd d d Zd!d d ZefddZGdddeZd"ddZd#ddZefddZddZd$ddZedkreejdkredejdnejd=eejddS)%aZrunpy.py - locating and running Python code using the module namespace Provides support for locating and running Python scripts using the Python module namespace instead of the native filesystem. This allows Python code to play nicely with non-filesystem based PEP 302 importers when locating support scripts as well as when importing modules. N) read_code get_importer run_modulerun_pathc@s(eZdZdZddZddZddZdS) _TempModulezCTemporarily replace a module in sys.modules with an empty namespacecCs||_t||_g|_dSN)mod_nametypes ModuleTypemodule _saved_moduleselfrr/usr/lib64/python3.8/runpy.py__init__s z_TempModule.__init__cCsB|j}z|jtj|Wntk r0YnX|jtj|<|Sr)rr appendsysmodulesKeyErrorr r rrr __enter__ s z_TempModule.__enter__cGs.|jr|jdtj|j<n tj|j=g|_dSNr)r rrrrargsrrr__exit__)s z_TempModule.__exit__N)__name__ __module__ __qualname____doc__rrrrrrrrs rc@s$eZdZddZddZddZdS)_ModifiedArgv0cCs||_t|_|_dSr)valueobject _saved_value _sentinel)rr rrrr1sz_ModifiedArgv0.__init__cCs0|j|jk rtdtjd|_|jtjd<dS)NzAlready preserving saved valuer)r"r# RuntimeErrorrargvr )rrrrr5s  z_ModifiedArgv0.__enter__cGs|j|_|jtjd<dSr)r#r r"rr%rrrrr;sz_ModifiedArgv0.__exit__N)rrrrrrrrrrr0src Csn|dk r|||dkr(d}|}d} n |j}|j}|j} |dkrH|j}|j||| d|||dt|||S)z)Helper to run code in nominated namespaceN)r__file__ __cached__r __loader__ __package____spec__)updateloaderorigincachedparentexec) codeZ run_globals init_globalsrmod_specpkg_name script_namer,fnamer.rrr _run_code@s*  r7c Cs^|dkr |n|j}t|6}t|"|jj}t|||||||W5QRXW5QRX|S)z5Helper to run code in new namespace with sys modifiedN)r-rrr __dict__r7copy) r1r2rr3r4r5r6 temp_module mod_globalsrrr_run_module_codeZsr<c Cs2|dr|d|d\}}}|rz t|WnHtk rz}z*|jdksh|j|krj||jdsjW5d}~XYnXtj|}|dk rt|dsddl m }dj ||d}|t |zt j|}WnJttttfk r} z"d}|| |t| j| | W5d} ~ XYnX|dkr2|d ||jdk r|d ksT|d r\|d z|d } t| |WS|k r}z"|tjkr|d ||fW5d}~XYnX|j} | dkr|d|z| |} Wn2tk r}z|t ||W5d}~XYnX| dkr(|d|||| fS)N.z#Relative module names not supported__path__r)warnz{mod_name!r} found in sys.modules after import of package {pkg_name!r}, but prior to execution of {mod_name!r}; this may result in unpredictable behaviour)rr4z:Error while finding module specification for {!r} ({}: {})zNo module named %s__main__z .__main__z%Cannot use package as __main__ modulez3%s; %r is a package and cannot be directly executedz0%r is a namespace package and cannot be executedzNo code object available for %s) startswith rpartition __import__ ImportErrornamerrgethasattrwarningsr?formatRuntimeWarning importlibutil find_specAttributeError TypeError ValueErrortypersubmodule_search_locationsendswith_get_module_detailsr,get_code) rerrorr4_eZexistingr?msgspecZexZ pkg_main_namer,r1rrrrThsd     ,         rTc@seZdZdZdS)_ErrorzBError that _run_module_as_main() should report without a tracebackN)rrrrrrrrr[sr[Tc Csz0|s|dkr t|t\}}}ntt\}}}Wn:tk rj}zdtj|f}t|W5d}~XYnXtjdj}|r|jtj d<t ||dd|S)aRuns the designated module in the __main__ namespace Note that the executed module will have full access to the __main__ namespace. If this is not desirable, the run_module() function should be used to run the module code in a fresh namespace. At the very least, these variables in __main__ will be overwritten: __name__ __file__ __cached__ __loader__ __package__ r@z%s: %sNr) rTr[_get_main_module_detailsr executableexitrr8r-r%r7)rZ alter_argvr3r1excrYZ main_globalsrrr_run_module_as_mains   r`FcCs@t|\}}}|dkr|}|r,t||||St|i|||SdS)znExecute a module's code without importing it Returns the resulting top level namespace dictionary N)rTr<r7)rr2run_nameZ alter_sysr3r1rrrrs c Csd}tj|}tj|=z\zt|WWNStk rn}z*|t|kr\|d|tjdf|W5d}~XYnXW5|tj|<XdS)Nr@zcan't find %r module in %rr)rrrTrDstrpath)rVZ main_nameZ saved_mainr_rrrr\s   r\c Csftjt|}t|}t|}W5QRX|dkr^t|}t||d}W5QRX||fS)Nr0) osrcabspathfsdecodeio open_codercompileread)rar6Z decoded_pathfr1rrr_get_code_from_files  rlc Cs$|dkr d}|dd}t|}d}t|jdkrFt|jdkrFd}t|tdsX|rxt||\}}t|||||d Stj d|znt \}} }t|P} t|<| jj} t|| ||| |W5QRW5QRWSQRXW5QRXW5ztj |Wnt k rYnXXdS) a_Execute code located at the specified filesystem location Returns the resulting top level namespace dictionary The file path may refer directly to a Python script (i.e. one that could be directly executed with execfile) or else it may refer to a zipfile or directory containing a top level __main__.py script. Nz r=rFZimpZ NullImporterT)r4r5)rBrrQrr isinstancerlr<rrcinsertremoverPr\rrr r8r7r9) Z path_namer2rar4ZimporterZis_NullImporterr1r6rr3r:r;rrrrs<   8r@z!No module specified for execution)file)NNNNN)NNNNN)T)NNF)NN)rrimportlib.machineryrKimportlib.utilrgr rdZpkgutilrr__all__r!rrr7r<rDrT Exceptionr[r`rr\rlrrlenr%printstderrrrrrsN    :    1