3
\. @ s d Z ddlZddlZddlZddlZddlmZmZ ddgZ G dd de
Z G dd d e
Z d d
d Z
d!d d
ZefddZG dd deZd"ddZd#ddZefddZdd Zd$ddZed k reejd k red ej d nejd= eejd dS )%aZ runpy.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( e Zd ZdZdd Zdd Zdd ZdS ) _TempModulezCTemporarily replace a module in sys.modules with an empty namespacec C s || _ tj|| _g | _d S )N)mod_nametypes
ModuleTypemodule
_saved_module)selfr r
/usr/lib64/python3.6/runpy.py__init__ s z_TempModule.__init__c C sB | j }y| jjtj| W n tk
r0 Y nX | jtj|< | S )N)r r appendsysmodulesKeyErrorr
)r r r
r
r __enter__ s z_TempModule.__enter__c G s. | j r| j d tj| j< n
tj| j= g | _ d S )Nr )r r r r )r argsr
r
r __exit__' s
z_TempModule.__exit__N)__name__
__module__ __qualname____doc__r r r r
r
r
r r s r c @ s$ e Zd Zdd Zdd Zdd ZdS )_ModifiedArgv0c C s || _ t | _| _d S )N)valueobject _saved_value _sentinel)r r r
r
r r / s z_ModifiedArgv0.__init__c C s0 | j | jk rtdtjd | _ | jtjd< d S )Nz Already preserving saved valuer )r r RuntimeErrorr argvr )r r
r
r r 3 s z_ModifiedArgv0.__enter__c G s | j | _| jtjd< d S )Nr )r r r r r! )r r r
r
r r 9 s z_ModifiedArgv0.__exit__N)r r r r r r r
r
r
r r . s r c
C sn |dk r|j | |dkr(d}|}d} n |j}|j}|j} |dkrH|j}|j ||| d|||d t| | |S )z)Helper to run code in nominated namespaceN)r __file__
__cached__r
__loader__ __package____spec__)updateloaderorigincachedparentexec)
codeZ run_globals init_globalsr mod_specpkg_name script_namer( fnamer* r
r
r _run_code> s(
r3 c C s^ |dkr |n|j }t|6}t|" |jj}t| |||||| W dQ R X W dQ R X |j S )z5Helper to run code in new namespace with sys modifiedN)r) r r r
__dict__r3 copy) r- r. r r/ r0 r1 r2 temp_module mod_globalsr
r
r _run_module_codeX s r8 c
; C s4 | j dr|d| jd\}}}|ry t| W nJ tk
r| } z.|jd ksj|j|krl|j |jd rl W Y d d }~X nX tjj| }|d k rt|d rddl m
} dj | |d}|t | yt
jj| }W nJ ttttfk
r } z$d}||j | t| j| | W Y d d } ~ X nX |d kr6|d | |jd k r| d
ksX| jd r`|d y| d }
t|
|S |k
r } z$| tjkr |d|| f W Y d d }~X nX |j} | d kr|d| y| j| } W n2 tk
r } z|t ||W Y d d }~X nX | d kr*|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)r r0 z:Error while finding module specification for {!r} ({}: {})zNo module named %s__main__z .__main__z%Cannot use package as __main__ modulez %s; %r is a package and cannot zbe directly executedz0%r is a namespace package and cannot be executedz No code object available for %sz3%s; %r is a package and cannot be directly executed)
startswith
rpartition
__import__ ImportErrornamer r gethasattrwarningsr; formatRuntimeWarning importlibutil find_specAttributeError TypeError
ValueErrortyper submodule_search_locationsendswith_get_module_detailsr( get_code)
r errorr0 _eZexistingr; msgspecZexZ
pkg_main_namer( r- r
r
r rP f sZ
,
rP c @ s e Zd ZdZdS )_ErrorzBError that _run_module_as_main() should report without a tracebackN)r r r r r
r
r
r rW s rW Tc C s y0|s| dkr t | t\} }}ntt\} }}W n: tk
rj } z dtj|f }tj| W Y dd}~X nX tjd j}|r|jtj d< t
||dd|S )a Runs 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 ) rP rW _get_main_module_detailsr
executableexitr r4 r) r! r3 )r Z
alter_argvr/ r- excrU Z main_globalsr
r
r _run_module_as_main s r\ Fc C s@ t | \} }}|dkr| }|r,t||||S t|i |||S dS )znExecute a module's code without importing it
Returns the resulting top level namespace dictionary
N)rP r8 r3 )r r. run_nameZ alter_sysr/ r- r
r
r r s c C s~ d}t j| }t j|= zVyt|S tk
rh } z,|t|krV| d|t jd f | W Y d d }~X nX W d |t j|< X d S )Nr< zcan't find %r module in %rr )r r rP r@ strpath)rR Z main_nameZ
saved_mainr[ r
r
r rX s
rX c C sT t |d}t|}W d Q R X |d krLt |d}t|j |d}W d Q R X ||fS )Nrbr, )openr compileread)r] r2 fr- r
r
r _get_code_from_file s re c C s |dkr d}|j dd }t| }d}t|jdkrFt|jdkrFd}t|tdsX|rxt|| \}}t|||||d S tj j
d| zRt \}} }t |4}
t
| |
jj} t|| ||| |j S Q R X W dQ R X W dytj j| W n tk
r Y nX X dS )
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
r9 r FZimpZ NullImporterT)r0 r1 )r> r rM r r
isinstancere r8 r r_ insertrX r r r
r4 r3 r5 removerL ) Z path_namer. r] r0 ZimporterZis_NullImporterr- r2 r r/ r6 r7 r
r
r r s0
"r< z!No module specified for execution)file)NNNNN)NNNNN)T)NNF)NN) r r importlib.machineryrG importlib.utilr Zpkgutilr r __all__r r r r3 r8 r@ rP ExceptionrW r\ r rX re r r lenr! printstderrr
r
r
r s6
:
1