File: //proc/self/root/lib/python3/dist-packages/keyring/tests/__pycache__/util.cpython-38.pyc
U
ܦ�\� � @ sl d dl Z d dlZd dlZd dlZd dlZeZG dd� d�Ze jdd� �Z dd� Z
ej ej Z
e
fdd �ZdS )
� Nc @ s: e Zd ZdZdd� Zd
dd�Zdd� Zd d
� Zd d � ZdS )� ImportKillerz@Context manager to make an import of a given name or names fail.c G s
|| _ d S �N��names)�selfr � r �4/usr/lib/python3/dist-packages/keyring/tests/util.py�__init__
s zImportKiller.__init__Nc C s || j kr| S d S r r )r �fullname�pathr r r � find_module s
zImportKiller.find_modulec C s || j kst�t|��d S r )r �AssertionError� ImportError)r r
r r r � load_module s zImportKiller.load_modulec C s8 i | _ | jD ]}tj�|d �| j |< q tj�d| � d S )Nr )�originalr �sys�modules�pop� meta_path�insert)r �namer r r � __enter__ s
zImportKiller.__enter__c G s6 t j�| � | j�� D ]\}}|d k r|t j|< qd S r )r r �remover �itemsr )r �args�key�valuer r r �__exit__ s zImportKiller.__exit__)N) �__name__�
__module__� __qualname__�__doc__r r r r r r r r r r
s
r c k s� i }|� � D ]6\}}| �|�||< |dkr:|| krB| |= q || |< q dV |� � D ](\}}|dkrr|| krz| |= qR|| |< qRdS )z�Helper context manager to make and unmake changes to a dict.
A None is not a valid value for the destination, and so means that the
associated name should be removed.N)r �get)Z destination�changesr r r r r r �NoNoneDictMutator% s
r$ c K s t tjf| �S )z6A context manager to temporarily change the os.environ)r$ �os�environ)r# r r r �Environ< s r' c C s& d}t d| �D ]}|t�|�7 }q|S )z2Generate a random string with length <i>k</i>
� r )�range�randomZchoice)�k�source�result�ir r r �
random_stringD s r/ )�
contextlibr% r r* �string�typeZ
__metaclass__r �contextmanagerr$ r' Z
ascii_lettersZdigitsZALPHABETr/ r r r r �<module> s