File: //lib/python3.10/test/support/__pycache__/import_helper.cpython-310.pyc
o
E�h8 � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZddlm Z e j
d dd��Z dd� Z d d
� Z
d d � Zd dd�dd�Zdd� Zd dd�ZG dd� de�ZG dd� de�Zdd� Zd d � ZdS )!� N� )�unlinkTc c sR � | r$t �� � t �ddt� dV W d � dS 1 s w Y dS dV dS )z�Context manager to suppress package and module deprecation
warnings when importing them.
If ignore is False, this context manager has no effect.
�ignorez.+ (module|package)N)�warnings�catch_warnings�filterwarnings�DeprecationWarning)r � r �1/usr/lib/python3.10/test/support/import_helper.py�_ignore_deprecated_imports
s �
�"�
r c C s$ zt j| = W d S ty Y d S w �N)�sys�modules�KeyError)�namer r r
�unload s
�r c C sT t | � tjD ] }tj�|| d �}t|d � dD ] }ttjj||d�� qqdS )z�'Forget' a module was ever imported.
This removes the module from sys.modules and deletes any PEP 3147/488 or
legacy .pyc files.
z.py�c)� r � )� optimizationN) r r
�path�os�joinr � importlib�util�cache_from_source)�modname�dirname�source�optr r r
�forget$ s
��r c C sB t j�| �}tj�tj�| ��}tj�|| d �}t� ||� |S )a Move a PEP 3147/488 pyc file to its legacy pyc location.
:param source: The file system path to the source file. The source file
does not need to exist, however the PEP 3147/488 pyc file must exist.
:return: The file system path to the legacy pyc file.
r )
r r r r r r �abspathr �shutil�move)r �pyc_file�up_one�
legacy_pycr r r
�make_legacy_pyc4 s
r' Fr )� required_onc C sr t |��+ z
t�| �W W d � S ty. } ztj�t|��r#� t� t
|���d}~ww 1 s2w Y dS )ac Import and return the module to be tested, raising SkipTest if
it is not available.
If deprecated is True, any module or package deprecation messages
will be suppressed. If a module is required on a platform but optional for
others, set required_on to an iterable of platform prefixes which will be
compared against sys.platform.
N) r r �
import_module� ImportErrorr
�platform�
startswith�tuple�unittest�SkipTest�str)r �
deprecatedr( �msgr r r
r) B s
����r) c C sL i }t dd� | D ��}ttj�D ]}|| v s|�|�r#tj�|�||< q|S )Nc s s � | ]}|d V qdS )�.Nr )�.0r r r r
� <genexpr>V s � z+_save_and_remove_modules.<locals>.<genexpr>)r- �listr
r r, �pop)�names� orig_modules�prefixesr r r r
�_save_and_remove_modulesT s �r; c
C s� t |��h t|�}t|�}| h|�|�}t|�}|D ]}dtj|< qz>z |D ]}t|� q&W n tyI Y W t|� tj�|� W d � dS w t� | �W t|� tj�|� W d � S t|� tj�|� w 1 sow Y dS )a9 Import and return a module, deliberately bypassing sys.modules.
This function imports and returns a fresh copy of the named Python module
by removing the named module from sys.modules before doing the import.
Note that unlike reload, the original module is not affected by
this operation.
*fresh* is an iterable of additional module names that are also removed
from the sys.modules cache before doing the import. If one of these
modules can't be imported, None is returned.
*blocked* is an iterable of module names that are replaced with None
in the module cache during the import to ensure that attempts to import
them raise ImportError.
The named module and any modules named in the *fresh* and *blocked*
parameters are saved before starting the import and then reinserted into
sys.modules when the fresh import is complete.
Module and package deprecation messages are suppressed during this import
if *deprecated* is True.
This function will raise ImportError if the named module cannot be
imported.
N)
r r6 r; r
r �
__import__r* �updater r) )r �fresh�blockedr1 r8 r9 r r r r
�import_fresh_module] s2
� �
��r@ c @ �( e Zd ZdZdd� Zdd� Zdd� ZdS ) � CleanImporta, Context manager to force import to return a new module reference.
This is useful for testing module-level behaviours, such as
the emission of a DeprecationWarning on import.
Use like this:
with CleanImport("foo"):
importlib.import_module("foo") # new reference
c G sJ t j�� | _|D ]}|t jv r"t j| }|j|kr t j|j= t j|= qd S r )r
r �copy�original_modules�__name__)�self� module_names� module_name�moduler r r
�__init__� s
��zCleanImport.__init__c C � | S r r �rF r r r
� __enter__� � zCleanImport.__enter__c G s t j�| j� d S r )r
r r= rD �rF �
ignore_excr r r
�__exit__� s zCleanImport.__exit__N�rE �
__module__� __qualname__�__doc__rJ rM rQ r r r r
rB � s
rB c @ rA ) �
DirsOnSysPatha� Context manager to temporarily add directories to sys.path.
This makes a copy of sys.path, appends any directories given
as positional arguments, then reverts sys.path to the copied
settings when the context ends.
Note that *all* sys.path modifications in the body of the
context manager, including replacement of the object,
will be reverted at the end of the block.
c G s( t jd d � | _t j| _t j�|� d S r )r
r �original_value�original_object�extend)rF �pathsr r r
rJ � s zDirsOnSysPath.__init__c C rK r r rL r r r
rM � rN zDirsOnSysPath.__enter__c G s | j t_| jtjd d �< d S r )rX r
r rW rO r r r
rQ � s zDirsOnSysPath.__exit__NrR r r r r
rV � s
rV c C s t j�� fS r )r
r rC r r r r
�
modules_setup� s r[ c C s: dd� t j�� D �}t j�� t j�|� t j�| � d S )Nc S s" g | ]
\}}|� d �r||f�qS )z
encodings.)r, )r4 �k�vr r r
�
<listcomp>� s �z#modules_cleanup.<locals>.<listcomp>)r
r �items�clearr= )�
oldmodules� encodingsr r r
�modules_cleanup� s
rc )T)F)r r F)�
contextlibr �importlib.utilr r" r
r. r � os_helperr �contextmanagerr r r r' r) r; r@ �objectrB rV r[ rc r r r r
�<module> s*
3