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
d�ZG dd� de�Zd!dd�Zd"dd�Zefdd�Zdd� Zd#dd�Zedk�r
ee j�d k � r�e d e j d � ne jd = ee jd � dS )$� N)� read_code� get_importer�
run_module�run_pathc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� _TempModulec C s || _ t�|�| _g | _d S �N)�mod_name�types�
ModuleType�module�
_saved_module��selfr � r � /usr/lib64/python3.8/runpy.py�__init__ s z_TempModule.__init__c C sB | j }z| j�tj| � W n tk
r0 Y nX | jtj|< | S r )r r �append�sys�modules�KeyErrorr 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 )�value�object� _saved_value� _sentinel)r r r r r r 1 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 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__)�update�loader�origin�cached�parent�exec)
�codeZ run_globals� init_globalsr �mod_spec�pkg_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 | � d�r|d��| �d�\}}}|r�z t|� W nH tk
rz } z*|jd ksh|j|krj|� |jd �sj� W 5 d }~X Y nX tj�| �}|d k r�t|d�s�ddl m
} dj | |d�}|t |�� zt
j�| �}W nJ ttttfk
�r } z"d}||� | t| �j| ��| �W 5 d } ~ X Y nX |d k�r2|d | ��|jd k �r�| d
k�sT| �d ��r\|d ��z| d }
t|
|�W S |k
�r� } z"| tjk�r�� |d
|| f ��W 5 d }~X Y nX |j} | d k�r�|d| ��z| �| �} W n2 tk
�r } z|t |��|�W 5 d }~X Y 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 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__� 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 �errorr5 �_�eZexistingr@ �msg�specZexZ
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 �
executable�exitr 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 �str�path)rW Z main_nameZ
saved_mainr` r r r r] � s
��r] c C sf t j�t �|��}t�|��}t|�}W 5 Q R X |d kr^t�|��}t|�� |d�}W 5 Q R X ||fS )Nr1 ) �osrd �abspath�fsdecode�io� open_coder �compile�read)rb r7 Z decoded_path�fr2 r r r �_get_code_from_file� s rm c C s$ |d kr d}|� d�d }t| �}d}t|�jdkrFt|�jdkrFd}t|td ��sX|rxt|| �\}}t|||||d�S 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 �insert�removerQ 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% �print�stderrr r r r �
sL � �
�
:
�
1