File: //usr/local/lib/python3.10/test/test_importlib/__pycache__/util.cpython-310.opt-1.pyc
o
�i�H � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZmZ d dl m
Z
d dl Z d dl Z d dl
Z
d dlZ
d dlmZmZ d dlmZ d dlmZ d dlmZ d dlZd dlZd dlZd dlZdd lm Z dd
lm Z e� � Z de _ de _!d ej"v r�d e _ d ej"vr�d e _!e� � a#dt#_$dt#_%dt#_&dt#_'d
t#_(dd� Z)e)� dd� Z*dFdd�Z+dFdd�Z,dFdd�Z-dZ.ej/dvr�e0�1� Z2e2e0kr�e0�3� Z2e
j$�4e2�s�dZ.e*d �d Z5e6ej7�e6e5j7�d �Z7d d � Z8dGd d!�Z9d"d#� Z:ej;d$d%� �Z<ej;dGdd&�d'd(��Z=ej;d)d*� �Z>G d+d,� d,�Z?G d-d.� d.e?�Z@G d/d0� d0e?�ZAd1d2� ZBd3d4� ZCej;d5d6� �ZDej;d7d8� �ZEd9d:� ZFG d;d<� d<�ZGdHd>d?�ZHG d@dA� dAe jI�ZJG dBdC� dC�ZKG dDdE� dEeK�ZLdS )I� N)� machinery�util�invalidate_caches)�ResourceReader)�Path�PurePath)�support)�
import_helper)� os_helper� )�data01)� zipdata01�errno� importlibZ _testcapic C s` t jD ]*} tjD ]$}tj| }tj�| |�}tj�|�r,| t_|t_ |t_
|t_ d S qqd S �N) �sys�pathr �EXTENSION_SUFFIXES�
EXTENSIONS�name�os�join�exists�ext�filename� file_path)r r r r � r �5/usr/local/lib/python3.10/test/test_importlib/util.py�_extension_details) s
���r c C s4 d| v rdnd}t �| �}t j| |dd�}||d�S )zAImport a module from importlib both w/ and w/o _frozen_importlib.�.)r r )�_frozen_importlib�_frozen_importlib_external)�freshZblocked��Frozen�Source)r Zimport_fresh_module)� module_namer" �frozen�sourcer r r �import_importlib9 s
�
r) c
K s� |d u rt j}n t|t�s|| }d�|| j�}| |f}t�||�}| j|_| j|_ ||_
|� � D ]\}}|| } t ||| � q1|S )Nz{}_{})
�unittestZTestCase�
isinstance�type�format�__name__�types� new_class�
__module__Z_NAMEZ_KIND�items�setattr)
�cls�kind�base�kwargsr �basesZ specialized�attr�values�valuer r r �specialize_classB s
r< c K s0 t | d|fi |��}t | d|fi |��}||fS )Nr$ r% )r< )r4 r6 r7 r' r( r r r � split_frozenU s r= c K s t | |fi |��S r )r= )Z
test_classr6 r7 r r r � test_both[ s r> T)�win32�cygwinFr% r# c C s t �t d�| �S )zRClass decorator that nullifies tests requiring a case-insensitive
file system.z&requires a case-insensitive filesystem)r* �skipIf�CASE_INSENSITIVE_FS)�testr r r �case_insensitive_testsn s
��rD � c C sZ t j�||d �}t|ddd��
}|�|� W d � n1 s w Y d�| |�|fS )N�.py�w�utf-8��encodingz{}.{})r r r �open�writer- )�parentr Zpkg_dir�contentr Zsubfiler r r � submoduleu s
�rO c C sB t | d��}|�d� t�|�W d � S 1 sw Y dS )znReads a pyc file and returns the unmarshalled code object within.
No header validation is performed.
�rb� N)rK �seek�marshal�load)Zpyc_pathZpyc_fr r r �get_code_from_pyc| s
$�rU c g s� � | D ] }|dv rt d�|���ztj|= W q ty Y qw zdV W | D ]}ztj|= W q' ty8 Y q'w dS | D ]}ztj|= W q= tyN Y q=w w )z�Uncache a module from sys.modules.
A basic sanity check is performed to prevent uncaching modules that either
cannot/shouldn't be uncached.
)r rS Zimpzcannot uncache {0}N)�
ValueErrorr- r �modules�KeyError)�namesr r r r �uncache� s4 �� � � � ��rZ )�pkgc # s< � � fdd�t jD �}t�d ���}t� g|�R � �c t�|��N t� tj �
|� �}|r9tj �
|d�}t� � � n
|d }|d u rCd}|d urct |ddd��
}|�
|� W d � n1 s^w Y |V W d � n1 spw Y W d � n1 sw Y W d � d S W d � d S 1 s�w Y d S ) Nc s" g | ]
}|� d �d � kr|�qS )r r )� partition)�.0�n�r r r �
<listcomp>� s " z temp_module.<locals>.<listcomp>z __init__.pyrF rE rG rH rI )r rW r
Ztemp_cwdrZ r Z
DirsOnSysPathr r r r �mkdirrK rL )r rN r[ � conflicts�cwd�location�modpathZmodfiler r_ r � temp_module� s0 � � �� ��"�rf c k s� � i }zPdg fdg fdg fdi ffD ] \}}t t|�||< || v r)| | }| |= n|}tt||� qt| �r?td�| �� ���dV W |�� D ]
\}}tt||� qGdS |�� D ]
\}}tt||� qXw )a, Context manager to manage the various importers and stored state in the
sys module.
The 'modules' attribute is not supported as the interpreter state stores a
pointer to the dict that the interpreter uses internally;
reassigning to sys.modules does not have the desired effect.
� meta_pathr �
path_hooks�path_importer_cachezunrecognized arguments: {0}N)�getattrr r3 �lenrV r- �keysr2 )r7 Z originalsr9 �defaultZ new_valuer; r r r � import_state� s. �
� ���rn c @ s6 e Zd ZdZi d�dd�Zdd� Zdd� Zd d
� Zd S ) �
_ImporterMockz0Base class to help with creating importer mocks.)� module_codec G s� i | _ i | _|D ]R}|�d�s|}n |d td� � }d|vr"d }n||kr/|�dd�d }n|}t�|�}| |_d|_||_ ||_
||krJdg|_ || j |< ||v rZ|| | j|< qd S )N� .__init__r r r z<mock __file__>z<mock __path__>) rW rp �endswithrk �rsplitr/ �
ModuleType�
__loader__�__file__� __package__r9 �__path__)�selfrp rY r � import_name�package�moduler r r �__init__� s.
��z_ImporterMock.__init__c C s
| j | S r �rW )ry r r r r � __getitem__� s
z_ImporterMock.__getitem__c C s t | j�� � | _| j�� | S r )rZ rW rl �_uncache� __enter__�ry r r r r� � s
z_ImporterMock.__enter__c G s | j �d d d � d S r )r� �__exit__)ry �exc_infor r r r� � s z_ImporterMock.__exit__N)r. r1 � __qualname__�__doc__r} r r� r� r r r r ro � s ro c @ s" e Zd ZdZddd�Zdd� ZdS )� mock_modulesz!Importer mock using PEP 302 APIs.Nc C s || j vrd S | S r r~ )ry �fullnamer r r r � find_module s
zmock_modules.find_modulec C s\ || j vrt�| j | tj |< || jv r)z| j| � W n ty( tj |= � w | j | S r )rW � ImportErrorr rp � Exception)ry r� r r r � load_module
s
�
zmock_modules.load_moduler )r. r1 r� r� r� r� r r r r r� s
r� c @ s* e Zd ZdZd dd�Zdd� Zdd� ZdS )
� mock_specz!Importer mock using PEP 451 APIs.Nc C sD z| j | }W n
ty Y d S w tj||j| t|dd �d�}|S )Nrx )�loader�submodule_search_locations)rW rX r �spec_from_file_locationrv rj )ry r� r rM r| �specr r r � find_spec s �
�zmock_spec.find_specc C s |j | jvrt�| j|j S r )r rW r� )ry r� r r r �
create_module& s zmock_spec.create_modulec C s, z | j |jj � W d S ty Y d S w r )rp �__spec__r rX �ry r| r r r � exec_module+ s
�zmock_spec.exec_module)NN)r. r1 r� r� r� r� r� r r r r r� s
r� c s. t jr
t�d�� �S t�� �� fdd��}|S )ztDecorator to protect sys.dont_write_bytecode from mutation and to skip
tests that require it to be set to False.zrelies on writing bytecodec s0 t j}dt _z
� | i |��}W |t _|S |t _w )NF)r �dont_write_bytecode)�argsr7 ZoriginalZ to_return��fxnr r �wrapper7 s �z&writes_bytecode_files.<locals>.wrapper)r r� r* �skip� functools�wraps)r� r� r r� r �writes_bytecode_files2 s
r� c
C sN z t �t j�| �� W dS ty& } z|jtjkr� W Y d}~dS d}~ww )z�Ensure that the __pycache__ directory for PEP 3147 pyc file exists.
:param bytecode_path: File system path to PEP 3147 pyc file.
N)r ra r �dirname�OSErrorr �EEXIST)�
bytecode_path�errorr r r �ensure_bytecode_pathC s ���r� c c s* � t j}| t _z dV W |t _dS |t _w )z&Adjust and restore sys.pycache_prefix.N)r �pycache_prefix)�prefixZ _orig_prefixr r r �temporary_pycache_prefixO s �r� c g s� � d}g }i }d}d}z�t �� }||d< t� }| D ]{}|�d�s#|} n |dtd� � } |�| � | tjv r:tj| = |�d�}
|} |
dd� D ]} t j
� | | �} t j
� | �s`t �
| � |�| � qGt j
� | |
d d �} t| d d
d ��}
|
�|�|�� W d � n1 s�w Y |�| � | ||< qt|� }|�� t|gd �}|�� |V W |dur�|�ddd� |dur�|�ddd� t�|� dS |dur�|�ddd� |dur�|�ddd� t�|� w )
aR Temporarily create each named module with an attribute (named 'attr')
that contains the name passed into the context manager that caused the
creation of the module.
All files are created in a temporary directory returned by
tempfile.mkdtemp(). This directory is inserted at the beginning of
sys.path. When the context manager exits all created files (source and
bytecode) are explicitly deleted.
No magic is performed when creating packages! This means that if you create
a module within a package you must also create the package's __init__ as
well.
z attr = {0!r}Nz.rootr} rq r ���rF rG rH rI �r )�tempfile�mkdtemp�setrr rk �addr rW �splitr r r r ra �appendrK rL r- rZ r� rn r� r
�rmtree)rY r( Z
created_paths�mappingZ
state_managerZuncache_managerZtemp_dirZ import_namesr rz Z
name_partsr Z directory�filer r r �create_modulesZ s\ �
� �
� r� c s � �fdd�}|S )z A mock sys.path_hooks entry.c s | � vrt ��S r )r� )�entry��entries�importerr r �hook� s z mock_path_hook.<locals>.hookr )r� r� r� r r� r �mock_path_hook� s r� c @ s e Zd Zdd� ZdS )�CASEOKTestBasec s0 d}t � fdd�|D ��|kr� �d� d S d S )N)s PYTHONCASEOK� PYTHONCASEOKc 3 s � | ] }|� j jjjv V qd S r )r �_bootstrap_external�_os�environ)r] �xr� r r � <genexpr>� s � �z4CASEOKTestBase.caseok_env_changed.<locals>.<genexpr>z/os.environ changes not reflected in _os.environ)�anyZskipTest)ry Z should_existZ
possibilitiesr r� r �caseok_env_changed� s ���z!CASEOKTestBase.caseok_env_changedN)r. r1 r� r� r r r r r� � s r� r c sN G � ��fdd�dt �}d}t�|�}|� }tj||d|d�}||_||_|S )Nc sH e Zd Zdd� Z�fdd�Z�fdd�Z� �fdd�Z� �fd d
�Zd S ) z create_package.<locals>.Readerc S s | S r r �ry r{ r r r �get_resource_reader� s z2create_package.<locals>.Reader.get_resource_readerc � || _ t� t�r
� �� S r ��_pathr+ r� �ry r )r� r r �
open_resource� �
z,create_package.<locals>.Reader.open_resourcec r� r r� )ry �path_r� r r �
resource_path� r� z,create_package.<locals>.Reader.resource_pathc sJ || _ t�t�r
��� D ]}|�d�}t|�dkr"|d |kr" dS q dS )N�/r r TF)r� r+ r� r� rk )ry r� r� �parts��contentsr r r � is_resource� s
�z*create_package.<locals>.Reader.is_resourcec 3 s$ � t �t�r��� D ]}|V q
d S r )r+ r� )ry r� r� r r r� � s �
�z'create_package.<locals>.Reader.contentsN)r. r1 r� r� r� r� r� r� r �r� r� r r r �Reader� s
r� Ztestingpackagezdoes-not-exist)�origin�
is_package)r r/ rt r �
ModuleSpecr� ru ) r� r r� r� r� r r| r� r� r r� r �create_package� s #
�r� c @ s� e Zd Zejdd� �Zdd� Zdd� Zdd� Ze �
e j d k d
�d d � �Z
d
d� Zdd� Zdd� Zdd� Zdd� Ze �
e j dk d�dd� �Ze �
e j dk d�dd � �Zd d � Zd S ) �CommonResourceTestsc C s t �r )�NotImplementedError)ry r{ r r r r �execute� s zCommonResourceTests.executec C s | � tjd� d S �N�
utf-8.file)r� r r. r� r r r �test_package_name� s z%CommonResourceTests.test_package_namec C s | � td� d S r� �r� r r� r r r �test_package_object� s z'CommonResourceTests.test_package_objectc C s d}| � t|� d S r� r� r� r r r �test_string_path� s z$CommonResourceTests.test_string_path)� � z requires os.PathLike supportc C s t d�}| �t|� d S r� )r r� r r� r r r �test_pathlib_path� s z%CommonResourceTests.test_pathlib_pathc C sN t t�}|jd }| �t�� | �t|� W d � d S 1 s w Y d S r� )r rv rM � assertRaisesrV r� r )ry r � full_pathr r r �test_absolute_path� s
"�z&CommonResourceTests.test_absolute_pathc C �< | � t�� | �td� W d � d S 1 sw Y d S )Nz../data01/utf-8.file)r� rV r� r r� r r r �test_relative_path� � "�z&CommonResourceTests.test_relative_pathc C s t jtj= | �tjd� d S r� )r rW r r. r� r� r r r �$test_importing_module_as_side_effect� s
z8CommonResourceTests.test_importing_module_as_side_effectc C r� r� )r� � TypeErrorr� r. r� r r r �test_non_package_by_name r� z,CommonResourceTests.test_non_package_by_namec C sF | � t�� tjd }| �|d� W d � d S 1 s w Y d S )Nztest.test_importlib.utilr� )r� r� r rW r� r� r r r �test_non_package_by_package s
"�z/CommonResourceTests.test_non_package_by_package)r� z No ResourceReader in Python 2c C s8 t �d�}t|t� d�}| �|d� | �|jjd� d S �Ns
Hello, world!�r� r r� )�io�BytesIOr� �FileNotFoundErrorr� � assertEqualru r� )ry �
bytes_datar{ r r r �test_resource_opener s
z(CommonResourceTests.test_resource_openerc C s: t �d�}t}t||d�}| �|d� | �|jjd� d S r� )r� r� rv r� r� r� ru r� )ry r� r r{ r r r �test_resource_path s
z&CommonResourceTests.test_resource_pathc C sL t t� t� d�}| �t�� | �|d� W d � d S 1 s w Y d S )Nr� r� )r� r� r� r� r� r r r �test_useless_loader s � "�z'CommonResourceTests.test_useless_loaderN)r. r1 r� �abc�abstractmethodr� r� r� r� r* rA r � version_infor� r� r� r� r� r� r� r� r� r r r r r� � s$
r� c @ s0 e Zd ZdZedd� �Zedd� �Zdd� ZdS )� ZipSetupBaseNc C s> t | jj�}|j}t|d �| _tj�| j� t �
d�| _ d S )Nzziptestdata.zipZ ziptestdata) r �
ZIP_MODULErv rM �str� _zip_pathr r r� r �
import_module�data)r4 Z data_pathZdata_dirr r r �
setUpClass% s
zZipSetupBase.setUpClassc C sx z t j�| j� W n ty Y nw z
t j| j= t j| jj= W n t y) Y nw z| `| `W d S t
y; Y d S w r ) r r �remover� rV ri rW r r. rX �AttributeError)r4 r r r �
tearDownClass- s" �
�
�zZipSetupBase.tearDownClassc C s t �� }| jt jg|�R � d S r )r Z
modules_setupZ
addCleanupZmodules_cleanup)ry rW r r r �setUp@ s zZipSetupBase.setUp)r. r1 r� r� � classmethodr r r r r r r r� " s
r� c @ s e Zd ZeZdS )�ZipSetupN)r. r1 r� r
r� r r r r r E s r r )rE )Tr )Mr� �builtins�
contextlibr r� r r r r Z
importlib.abcr r� rS r Zos.path�pathlibr r rC r Z test.supportr r
r* r r� r/ rE r r
�SimpleNamespaceZBUILTINSZ good_nameZbad_name�builtin_module_namesr r r r r r r r) r< r= r> rB �platformrv �upperZ changed_name�lowerr Zsource_importlib� staticmethod�
__import__rD rO rU �contextmanagerrZ rf rn ro r� r� r� r� r� r� r� r� r� �ABCr� r� r r r r r �<module> s�
�
)
9
2J#