3
\�. � @ s� d dl Z d dlZd dlZd dlZd dlmZmZ ddgZG dd� de �Z
G dd� de �Z d d d
�Z d d d �Z
efd
d�ZG dd� de�Zd!dd�Zd"dd�Zefdd�Zdd� Zd#dd�Zedk� r�ee j�d k � r�ed e jd � ne jd = ee jd � dS )$� N)� read_code� get_importer�
run_module�run_pathc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� _TempModulec C s || _ tj|�| _g | _d S )N)�mod_name�types�
ModuleType�module�
_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 �append�sys�modules�KeyErrorr
)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__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)�value�object� _saved_value� _sentinel)r r r
r
r r / s z_ModifiedArgv0.__init__c C s0 | j | jk rtd��tjd | _ | 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 |d k r|j |� |d kr(d }|}d } n |j}|j}|j} |d krH|j}|j ||| d |||d� t| |� |S )N)r �__file__�
__cached__�__doc__�
__loader__� __package__�__spec__)�update�loader�origin�cached�parent�exec)
�codeZ run_globals� init_globalsr �mod_spec�pkg_name� script_namer( �fnamer* r
r
r � _run_code> s(
r3 c C s^ |d kr |n|j }t|��6}t|��" |jj}t| ||||||� W d Q R X W d Q R X |j� S )N)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 d�r|d��| jd�\}}}|r�y t|� W nJ tk
r| } z.|jd ksj|j|krl|j |jd � rl� W Y d d }~X nX tjj| �}|d k r�t|d� r�ddl m
} dj | |d�}|t |�� yt
jj| �}W nJ ttttfk
�r } z$d}||j | t| �j| ��| �W Y d d } ~ X nX |d k�r6|d | ��|jd k �r�| d
k�sX| jd ��r`|d ��y| d }
t|
|�S |k
�r� } z$| tjk�r�� |d|| f ��W Y d d }~X nX |j} | d k�r�|d| ��y| j| �} W n2 tk
�r } z|t |��|�W Y d d }~X nX | d k�r*|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__� ImportError�namer r �get�hasattr�warningsr; �format�RuntimeWarning� importlib�util� find_spec�AttributeError� TypeError�
ValueError�typer �submodule_search_locations�endswith�_get_module_detailsr( �get_code)
r �errorr0 �_�eZexistingr; �msg�specZexZ
pkg_main_namer( r- r
r
r rP f sZ
,
rP c @ s e Zd ZdS )�_ErrorN)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 d d }~X nX tjd j}|r�|jtj d< t
||d d|�S )Nr< z%s: %sr ) rP rW �_get_main_module_detailsr �
executable�exitr 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 | �\} }}|d kr| }|r,t||||�S t|i |||�S d S )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@ �str�path)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 )N�rbr, )�openr �compile�read)r] r2 �fr- r
r
r �_get_code_from_file� s re c C s |d kr d}|j d�d }t| �}d}t|�jdkrFt|�jdkrFd}t|td ��sX|rxt|| �\}}t|||||d�S tj j
d| � zRt � \}} }t |��4}
t
| �� |
jj} t|| ||| |�j� S Q R X W d Q R X W d ytj j| � W n tk
�r Y nX X d S ) Nz
<run_path>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 �importlib.machineryrG �importlib.utilr Zpkgutilr r �__all__r r r r3 r8 r@ rP � ExceptionrW r\ r rX re r r �lenr �print�stderrr
r
r
r �<module>
s4
:
1