a DOg3@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.9/runpy.py__init__s z_TempModule.__init__cCs@|j}z|jtj|Wnty.Yn0|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|jurtdtjd|_|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|dur|||dur(d}|}d} n |j}|j}|j} |durH|j}|j||| d|||dt|||S)z)Helper to run code in nominated namespaceN)r__file__ __cached__r __loader__ __package____spec__)updateloaderorigincachedparentexec) code run_globals init_globalsrmod_specpkg_name script_namer,fnamer.rrr _run_code@s*  r8c Cs|dur |n|j}t|T}t|,|jj}t|||||||Wdn1sV0YWdn1st0Y|S)z5Helper to run code in new namespace with sys modifiedN)r-rrr __dict__r8copy) r1r3rr4r5r6r7 temp_module mod_globalsrrr_run_module_codeZs@r=c Csd|dr|d|d\}}}|rz t|WnJty|}z2|jdusf|j|krh||jdshWYd}~n d}~00tj|}|durt|dsddl m }dj ||d}|t |zt j|}WnvttttfyH} zTd}|d r|d |dd d |d 7}|| |t| j| | WYd} ~ n d} ~ 00|dur`|d||jdur|dks|dr|dz|d} t| |WS|y}z*|tjvr|d||fWYd}~n d}~00|j} | dur|d|z| |} Wn4tyB}z|t ||WYd}~n d}~00| durZ|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)rr5z:Error while finding module specification for {!r} ({}: {})z.pyz . Try using 'z' instead of 'z' as the module name.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 ValueErrorendswithtypersubmodule_search_locations_get_module_detailsr,get_code) rerrorr5_eexistingr@msgspecex pkg_main_namer,r1rrrrVhsj       0      $  rVc@seZdZdZdS)_ErrorzBError that _run_module_as_main() should report without a tracebackN)rrrrrrrrr`sr`Tc Csz0|s|dkr t|t\}}}ntt\}}}Wn<tyl}z$dtj|f}t|WYd}~n d}~00tjdj}|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__ rBz%s: %sNr) rVr`_get_main_module_detailsr executableexitrr9r-r%r8)r alter_argvr4r1excr\ main_globalsrrr_run_module_as_mains    rgFcCs@t|\}}}|dur|}|r,t||||St|i|||SdS)u"Execute a module's code without importing it. mod_name -- an absolute module name or package name. Optional arguments: init_globals -- dictionary used to pre-populate the module’s globals dictionary before the code is executed. run_name -- if not None, this will be used for setting __name__; otherwise, __name__ will be set to mod_name + '__main__' if the named module is a package and to just mod_name otherwise. alter_sys -- if True, sys.argv[0] is updated with the value of __file__ and sys.modules[__name__] is updated with a temporary module object for the module being executed. Both are restored to their original values before the function returns. Returns the resulting module globals dictionary. N)rVr=r8)rr3run_name alter_sysr4r1rrrrs c Csd}tj|}tj|=zpzt|WW|tj|<Styx}z2|t|vrb|d|tjdf|WYd}~n d}~00W|tj|<n |tj|<0dS)NrBzcan't find %r module in %rr)rrrVrFstrpath)rX main_name saved_mainrerrrras      racCstjt|}t|}t|}Wdn1s:0Y|durt| }t||d}Wdn1s|0Y||fS)Nr0) osrkabspathfsdecodeio open_codercompileread)rhr7 decoded_pathfr1rrr_get_code_from_files & .rwc Cs|dur d}|dd}t|}d}t|jdkrFt|jdkrFd}t|tdsX|rxt||\}}t|||||d Stj d|zt \}} }t |} t |d| jj} t|| ||| |WdWdWztj |WStyYS0S1s0YWdn1s20YWztj |WntybYn0n(ztj |WntyYn00dS) uExecute code located at the specified filesystem location. path_name -- filesystem location of a Python script, zipfile, or directory containing a top level __main__.py script. Optional arguments: init_globals -- dictionary used to pre-populate the module’s globals dictionary before the code is executed. run_name -- if not None, this will be used to set __name__; otherwise, '' will be used for __name__. Returns the resulting module globals dictionary. Nz r>rFimp NullImporterT)r5r6)rDrrTrr isinstancerwr=rrkinsertrarrr r9r8r:removerR) path_namer3rhr5importeris_NullImporterr1r7rr4r;r<rrrrsH  $6rBz!No module specified for execution)file)NNNNN)NNNNN)T)NNF)NN)rrimportlib.machineryrMimportlib.utilrqr rnpkgutilrr__all__r!rrr8r=rFrV Exceptionr`rgrrarwrrlenr%printstderrrrrrsB    =    6