U
e5d/ @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZm Z ddgZ
G dd de Z G dd de Z
d d d
Zd d d Zefd
dZG dd deZd!ddZd"ddZefddZdd Zd#ddZedkr
ee jd k re d e j d ne jd = ee jd dS )$ N) read_code get_importer
run_modulerun_pathc @ s$ e Zd Zdd Zdd Zdd ZdS ) _TempModulec C s || _ t|| _g | _d S N)mod_nametypes
ModuleTypemodule
_saved_moduleselfr r /usr/lib64/python3.8/runpy.py__init__ s z_TempModule.__init__c C sB | j }z| jtj| W n tk
r0 Y nX | jtj|< | S r )r r appendsysmodulesKeyErrorr 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__r r r r r r r r s r c @ s$ e Zd Zdd Zdd Zdd ZdS )_ModifiedArgv0c C s || _ t | _| _d S r )valueobject _saved_value _sentinel)r r r r r r 1 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 5 s z_ModifiedArgv0.__enter__c G s | j | _| jtjd< d S r )r# r r" r r% r r r r r ; s z_ModifiedArgv0.__exit__Nr r r r r r 0 s r c
C sn |d k r| | |d kr(d }|}d } n |j}|j}|j} |d krH|j}|j ||| d |||d t| | |S )N)r __file__
__cached____doc__
__loader__ __package____spec__)updateloaderorigincachedparentexec)
codeZ run_globals init_globalsr mod_specpkg_name script_namer- fnamer/ r r r _run_code@ s*
r8 c
C s^ |d kr |n|j }t|6}t|" |jj}t| |||||| W 5 Q R X W 5 Q R X | S r )r. r r r __dict__r8 copy) r2 r3 r r4 r5 r6 r7 temp_module mod_globalsr r r _run_module_codeZ s r= c
C s2 | dr|d| d\}}}|rz t| W nH tk
rz } z*|jd ksh|j|krj| |jd sj W 5 d }~X Y nX tj| }|d k rt|dsddl m
} dj | |d}|t | zt
j| }W nJ ttttfk
r } z"d}|| | t| j| | W 5 d } ~ X Y nX |d kr2|d | |jd k r| d
ksT| d r\|d z| d }
t|
|W S |k
r } z"| tjkr |d
|| f W 5 d }~X Y nX |j} | d kr|d| z| | } W n2 tk
r } z|t ||W 5 d }~X Y 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 r5 z: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 executedz No code object available for %s)
startswith
rpartition
__import__ ImportErrornamer r gethasattrwarningsr@ formatRuntimeWarning importlibutil find_specAttributeError TypeError
ValueErrortyper submodule_search_locationsendswith_get_module_detailsr- get_code)
r errorr5 _eZexistingr@ msgspecZexZ
pkg_main_namer- r2 r r r rU h sd
,
rU c @ s e Zd ZdS )_ErrorN)r r r r r r r r\ s r\ Tc
C s z0|s| dkr t | t\} }}ntt\} }}W n: tk
rj } z dtj|f }t| W 5 d }~X Y nX tjd j}|r|jtj d< t
||d d|S )NrA z%s: %sr ) rU r\ _get_main_module_detailsr
executableexitr r9 r. r% r8 )r Z
alter_argvr4 r2 excrZ Z main_globalsr r r _run_module_as_main s ra Fc C s@ t | \} }}|d kr| }|r,t||||S t|i |||S d S r )rU r= r8 )r r3 run_nameZ alter_sysr4 r2 r r r r s c
C s d}t j| }t j|= z\zt|W W NS tk
rn } z*|t|kr\| d|t jd f | W 5 d }~X Y nX W 5 |t j|< X d S )NrA zcan't find %r module in %rr )r r rU rE strpath)rW Z main_nameZ
saved_mainr` r r r r] s
r] c C sf t jt |}t|}t|}W 5 Q R X |d kr^t|}t| |d}W 5 Q R X ||fS )Nr1 ) osrd abspathfsdecodeio open_coder compileread)rb r7 Z decoded_pathfr2 r r r _get_code_from_file s rm c C s$ |d kr d}| dd }t| }d}t|jdkrFt|jdkrFd}t|td sX|rxt|| \}}t|||||dS tj
d| znt
\}} }t|P}
t| < |
jj} t|| ||| | W 5 Q R W 5 Q R W S Q R X W 5 Q R X W 5 ztj | W n t k
r Y nX X d S ) Nz
r> r FZimpZ NullImporterT)r5 r6 )rC r rR r r
isinstancerm r= r rd insertremoverQ r] r r r r9 r8 r: ) Z path_namer3 rb r5 ZimporterZis_NullImporterr2 r7 r r4 r; r< r r r r s<
8rA z!No module specified for execution)file)NNNNN)NNNNN)T)NNF)NN) r importlib.machineryrL importlib.utilrh r re Zpkgutilr r __all__r! r r r8 r= rE rU Exceptionr\ ra r r] rm r r lenr% printstderrr r r r
sL
:
1