File: //usr/local/lib/python3.7/distutils/tests/__pycache__/support.cpython-37.pyc
B
!��g� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZ ddl m
Z
ddl m Z m
Z
mZmZmZ ddlmZ G dd� de�ZG d d
� d
e�ZG d d � d �ZG d
d� de�Zdd� Zdd� Zdd� ZdS )z&Support code for distutils test cases.� N)�deepcopy)�log)�DEBUG�INFO�WARN�ERROR�FATAL)� Distributionc s@ e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zd d
� Z� ZS ) �LoggingSilencerc s6 t � �� t�tj�| _tjj| _| jtj_g | _ d S )N)
�super�setUpr �
set_thresholdr � threshold�Log�_log�_old_log�logs)�self)� __class__� �3/usr/local/lib/python3.7/distutils/tests/support.pyr s
zLoggingSilencer.setUpc s$ t �| j� | jt j_t� �� d S )N)r r
r r r r r �tearDown)r )r r r r s
zLoggingSilencer.tearDownc C sT |t ttttfkr"tdt|� ��t|t�s>tdt |�j
��| j � |||f� d S )Nz%s wrong log levelz msg should be str, not '%.200s')
r r r r r �
ValueError�str�
isinstance� TypeError�type�__name__r �append)r �level�msg�argsr r r r s
zLoggingSilencer._logc s dd� }� fdd�| j D �S )Nc S s | | S )Nr )r r! r r r �_format) s z)LoggingSilencer.get_logs.<locals>._formatc s" g | ]\}}}|� kr|| �qS r r )�.0r r r! )�levelsr r �
<listcomp>+ s z,LoggingSilencer.get_logs.<locals>.<listcomp>)r )r r$ r" r )r$ r �get_logs( s
zLoggingSilencer.get_logsc C s
g | _ d S )N)r )r r r r �
clear_logs. s zLoggingSilencer.clear_logs) r �
__module__� __qualname__r r r r& r' �
__classcell__r r )r r r
s
r
c sH e Zd ZdZ� fdd�Z� fdd�Zdd� Zdd d
�Zdd d
�Z� Z S )�TempdirManagerz}Mix-in class that handles temporary directories for test cases.
This is intended to be used with unittest.TestCase.
c s t � �� t�� | _g | _d S )N)r r �os�getcwd�old_cwd�tempdirs)r )r r r r 8 s
zTempdirManager.setUpc sB t �| j� t� �� x&| jr<| j�� }t�|t j dk� qW d S )N)�nt�cygwin)
r, �chdirr. r r r/ �pop�shutilZrmtree�name)r �d)r r r r = s
zTempdirManager.tearDownc C s t �� }| j�|� |S )zjCreate a temporary directory that will be cleaned up.
Returns the path of the directory.
)�tempfile�mkdtempr/ r )r r6 r r r r8 F s zTempdirManager.mkdtemp�xxxc C sB t |ttf�rtjj|� }t|d�}z|�|� W d|�� X dS )zWWrites a file in the given path.
path can be a string or a sequence.
�wN) r �list�tupler, �path�join�open�write�close)r r= Zcontent�fr r r �
write_fileO s
zTempdirManager.write_file�fooc K s2 | � � }tj�||�}t�|� t|d�}||fS )a Will generate a test environment.
This function creates:
- a Distribution instance using keywords
- a temporary directory with a package structure
It returns the package directory and the distribution
instance.
)Zattrs)r8 r, r= r> �mkdirr )r Zpkg_name�kwZtmp_dirZpkg_dirZdistr r r � create_dist] s
zTempdirManager.create_dist)r9 )rD )
r r( r) �__doc__r r r8 rC rG r* r r )r r r+ 2 s
r+ c @ s e Zd ZdZdd� Zdd� ZdS )� DummyCommandzAClass to store options for retrieval via set_undefined_options().c K s& x |� � D ]\}}t| ||� q
W d S )N)�items�setattr)r �kwargsrF �valr r r �__init__r s zDummyCommand.__init__c C s d S )Nr )r r r r �ensure_finalizedv s z DummyCommand.ensure_finalizedN)r r( r) rH rN rO r r r r rI o s rI c s( e Zd Z� fdd�Z� fdd�Z� ZS )� EnvironGuardc s t t| ��� ttj�| _d S )N)r rP r r r, �environ� old_environ)r )r r r r | s zEnvironGuard.setUpc sn x0| j �� D ]"\}}tj�|�|kr |tj|< q W x(ttj�� �D ]}|| j krBtj|= qBW tt| �� � d S )N)
rR rJ r, rQ �getr< �keysr rP r )r �key�value)r r r r � s
zEnvironGuard.tearDown)r r( r) r r r* r r )r r rP z s rP c C s( t � }|dkrt�d��t�|| � dS )ak Helper for tests that need the xxmodule.c source file.
Example use:
def test_compile(self):
copy_xxmodule_c(self.tmpdir)
self.assertIn('xxmodule.c', os.listdir(self.tmpdir))
If the source file can be found, it will be copied to *directory*. If not,
the test will be skipped. Errors during copy are not caught.
Nz>cannot find xxmodule.c (test must run in the python build dir))�_get_xxmodule_path�unittestZSkipTestr4 �copy)Z directory�filenamer r r �copy_xxmodule_c� s
r[ c
C sf t �d�} tj�tj�t�d�tj�| dd�tj�| ddddd�g}x |D ]}tj�|�rJ|S qJW d S )N�srcdirz
xxmodule.cZModulesz..)� sysconfig�get_config_varr, r= r> �dirname�__file__�exists)r\ Z
candidatesr= r r r rW � s
rW c C s~ t jdkrtj�d�| _n`t�d�rzt�d�}|dkr@dg| _n:tj dkrRg | _n(|�
d�\}}}d d
� |� t j �D �| _dS ) a� Function needed to make build_ext tests pass.
When Python was built with --enable-shared on Unix, -L. is not enough to
find libpython<blah>.so, because regrtest runs in a tempdir, not in the
source directory where the .so lives.
When Python was built with in debug mode on Windows, build_ext commands
need their debug attribute set, and it is not done automatically for
some reason.
This function handles both of these things. Example use:
cmd = build_ext(dist)
support.fixup_build_ext(cmd)
cmd.ensure_finalized()
Unlike most other Unix platforms, Mac OS X embeds absolute paths
to shared libraries into executables, so the fixup is not needed there.
r0 z_d.exeZPy_ENABLE_SHAREDZ RUNSHAREDN�.�darwin�=c S s g | ] }|r|�qS r r )r# r6 r r r r% � s z#fixup_build_ext.<locals>.<listcomp>)
r, r5 �sys�
executable�endswith�debugr] r^ Z library_dirs�platform� partition�split�pathsep)�cmdZ runsharedr5 ZequalsrV r r r �fixup_build_ext� s
rn )rH r, re r4 r7 rX r] rY r Z distutilsr Z
distutils.logr r r r r Zdistutils.corer �objectr
r+ rI rP r[ rW rn r r r r �<module> s" #=