3
�Qg�m � @ s> d ddddddddd d
d d g
Z d
dlZd
dlZd
dlZd
dlZd
dl Z
d
dl Z d
dl
mZ d
dlZy d
dlZW n ek
r� d
dlZY nX ejZejejB ejB Zeed�r�eej O ZeZ eed�r�e ej O Z eed�r�ej Z ndZ dZ e� Z!eed��rej"Z#n eed��rej$Z#ndd� Z#dd� Z%dd� Z&d d � Z'G d d � d �Z(d d!� Z)d"d#� Z*da+d$d%� Z,d&d'� Z-d(d)� Z.d*d+� Z/d,d� Z0d-d � Z1da2d.d
� Z3d/d � Z4dCd1d�Z5dDd2d�Z6d3e dfd4d�Z7G d5d6� d6�Z8G d7d8� d8�Z9dFd<d �Z:ej;d=k�sej<j=d>k�re:Z>need?�a?dHd@d�Z>G dAd� d�Z@G dBd� deA�ZBdS )I�NamedTemporaryFile�
TemporaryFile�SpooledTemporaryFile�TemporaryDirectory�mkstemp�mkdtemp�mktemp�TMP_MAX�
gettempprefix�tempdir�
gettempdir�gettempprefixb� gettempdirb� N)�Random�
O_NOFOLLOW�O_BINARYi' Ztmp�lstat�statc C s t j| t j�}t j|� d S )N)�_os�open�O_RDONLY�close)�fn�fd� r � /usr/lib64/python3.6/tempfile.py�_statT s r c C s* y t | � W n tk
r dS X dS d S )NFT)r �OSError)r r r r �_existsX s
r c G s` d }xJ| D ]B}|d krq
t |t�r8|tkr2td��t}q
|tkrHtd��t}q
W |d kr\tS |S )Nz1Can't mix bytes and non-bytes in path components.)�
isinstance�bytes�str� TypeError)�argsZ return_type�argr r r �_infer_return_typea s
r% c C sd t | ||�}|d kr|� }| d kr:|tkr0t} n
tjt�} |d krX|tkrRt� }nt� }| |||fS )N)r% r! �templater �fsencoder r
)�prefix�suffix�dir� output_typer r r �_sanitize_paramsv s
r, c @ s, e Zd ZdZedd� �Zdd� Zdd� ZdS ) �_RandomNameSequenceZ%abcdefghijklmnopqrstuvwxyz0123456789_c C s, t j� }|t| dd �kr&t� | _|| _| jS )N�_rng_pid)r �getpid�getattr�_RandomZ_rngr. )�selfZcur_pidr r r �rng� s
z_RandomNameSequence.rngc C s | S )Nr )r2 r r r �__iter__� s z _RandomNameSequence.__iter__c s0 | j � | jj�� �fdd�td�D �}dj|�S )Nc s g | ] }�� ��qS r r )�.0Zdummy)�c�chooser r �
<listcomp>� s z0_RandomNameSequence.__next__.<locals>.<listcomp>� � )�
charactersr3 Zchoice�range�join)r2 Zlettersr )r6 r7 r �__next__� s z _RandomNameSequence.__next__N)�__name__�
__module__� __qualname__r; �propertyr3 r4 r> r r r r r- � s r- c C s� g } x$dD ] }t j|�}|r
| j|� q
W t jdkr\| jt jjd�t jjd�ddd d
g� n| jd d d
g� y| jt j� � W n$ t t
fk
r� | jt j � Y nX | S )N�TMPDIR�TEMP�TMP�ntz~\AppData\Local\Tempz%SYSTEMROOT%\Tempzc:\tempzc:\tmpz\tempz\tmpz/tmpz/var/tmpz/usr/tmp)rC rD rE ) r �getenv�append�name�extend�path�
expanduser�
expandvars�getcwd�AttributeErrorr �curdir)�dirlistZenvname�dirnamer r r �_candidate_tempdir_list� s
rS c C s4 t � } t� }�x|D �]}|tjkr0tjj|�}x�td�D ]�}t| �}tjj||�}y\tj |t
d�}z<z*t j |ddd��}|j d� W d Q R X W d tj
|� X W d tj|� X |S tk
r� Y q: tk
�r tjdkr�tjj|�r�tj|tj�r�w:P Y q: tk
�r P Y q:X q:W qW ttjd| ��d S ) N�d i� �wbF)�closefds blatrF z)No usable temporary directory found in %s)r- rS r rP rK �abspathr< �nextr= r �_bin_openflags�_io�writer �unlink�FileExistsError�PermissionErrorrI �isdir�access�W_OKr �FileNotFoundError�_errno�ENOENT)ZnamerrQ r* �seqrI �filenamer �fpr r r �_get_default_tempdir� s:
rh c
C s2 t d kr.tj� zt d kr t� a W d tj� X t S )N)�_name_sequence�
_once_lock�acquirer- �releaser r r r �_get_candidate_names� s
rm c
C s� t � }|tkrttj|�}x�tt�D ]�}t|�}tjj | || | �}ytj
||d�} W nR t k
rn w$Y n> t k
r� tj
dkr�tjj| �r�tj| tj�r�w$n� Y nX | tjj|�fS W t tjd��d S )Ni� rF z#No usable temporary file name found)rm r �mapr r' r< r rX rK r= r r] r^ rI r_ r` ra rW rc �EEXIST)
r* ZpreZsuf�flagsr+ �namesre rI �filer r r r �_mkstemp_inner� s$ rs c G sJ | t jkr | |f|�ddi� n&t jdks8t jj|� rF| |f|�� d S )N�follow_symlinksFrF )r �supports_follow_symlinksrI rK �islink)�funcrK r# r r r �_dont_follow_symlinks s
rx c C s> y
t j}W n tk
r Y nX t|| d� tt j| d� d S )Nr i� )r �chflagsrO rx �chmod)rK ry r r r � _resetperms s
r{ c C s t S )N)r& r r r r r * s c C s t jt� �S )N)r r' r r r r r r . s c
C s2 t d kr.tj� zt d kr t� a W d tj� X t S )N)r
rj rk rh rl r r r r r 4 s
c C s t jt� �S )N)r r' r r r r r r
@ s Fc C s2 t || |�\}} }}|r t}nt}t||| ||�S )N)r, �_text_openflagsrY rs )r) r( r* �textr+ rp r r r r D s
c C s� t || |�\}} }}t� }|tkr.ttj|�}x�tt�D ]�}t|�}tj j
||| | �}ytj |d� W nR t k
r� w8Y n> t
k
r� tjdkr�tj j|�r�tj|tj�r�w8n� Y nX |S W t tjd��d S )Ni� rF z(No usable temporary directory name found)r, rm r rn r r' r< r rX rK r= �mkdirr] r^ rI r_ r` ra rc ro )r) r( r* r+ rq re rI rr r r r r l s&
r: c C s` |d krt � }t� }x:tt�D ].}t|�}tjj||| | �}t|�s |S q W t t
j d��d S )Nz"No usable temporary filename found) r rm r< r rX r rK r= r r] rc ro )r) r( r* rq re rI rr r r r r � s c @ sH e Zd ZdZdZd dd�Zejdkr<ejfdd�Z d d
� Z
nd d� Z dS )
�_TemporaryFileCloserNFTc C s || _ || _|| _d S )N)rr rI �delete)r2 rr rI r� r r r �__init__� s z _TemporaryFileCloser.__init__rF c C s> | j r:| jd k r:d| _ z| jj� W d | jr8|| j� X d S )NT)� close_calledrr r r� rI )r2 r\ r r r r � s z_TemporaryFileCloser.closec C s | j � d S )N)r )r2 r r r �__del__� s z _TemporaryFileCloser.__del__c C s | j sd| _ | jj� d S )NT)r� rr r )r2 r r r r � s )T) r? r@ rA rr r� r� r rI r\ r r� r r r r r � s
r c @ s> e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� ZdS )�_TemporaryFileWrapperTc C s$ || _ || _|| _t|||�| _d S )N)rr rI r� r �_closer)r2 rr rI r� r r r r� � s z _TemporaryFileWrapper.__init__c s^ | j d }t||�}t|d�rD|� tj� �� fdd��}| j|_|}t|t�sZt| ||� |S )Nrr �__call__c s
� | |�S )Nr )r# �kwargs)rw r r � func_wrapper� s z7_TemporaryFileWrapper.__getattr__.<locals>.func_wrapper) �__dict__r0 �hasattr�
_functools�wrapsr� r �int�setattr)r2 rI rr �ar� r )rw r � __getattr__� s
z!_TemporaryFileWrapper.__getattr__c C s | j j� | S )N)rr � __enter__)r2 r r r r� s
z _TemporaryFileWrapper.__enter__c C s | j j|||�}| j� |S )N)rr �__exit__r )r2 �exc�value�tb�resultr r r r� s z _TemporaryFileWrapper.__exit__c C s | j j� d S )N)r� r )r2 r r r r s z_TemporaryFileWrapper.closec c s x| j D ]
}|V qW d S )N)rr )r2 �liner r r r4 s z _TemporaryFileWrapper.__iter__N)T) r? r@ rA r� r� r� r� r r4 r r r r r� � s
r� �w+b� Tc
C s� t |||�\}}}}t} tjdkr0|r0| tjO } t|||| |�\}
} y tj|
| |||d�} t| | |�S t k
r� tj
| � tj |
� � Y nX d S )NrF )� buffering�newline�encoding) r, rY r rI Z O_TEMPORARYrs rZ r r� �
BaseExceptionr\ r )
�moder� r� r� r) r( r* r� r+ rp r rI rr r r r r s
�posix�cygwin� O_TMPFILEc C s� t |||�\}}}}t}tr�y$|tjB tj @ } tj|| d�}
W n* tk
rX daY nB tk
rj Y n0X yt j|
| |||d�S tj
|
� � Y nX t |||||�\}
} y tj | � t j|
| |||d�S tj
|
� � Y nX d S )Ni� F)r� r� r� )
r, rY �_O_TMPFILE_WORKSr r� �O_CREATr �IsADirectoryErrorr rZ r rs r\ ) r� r� r� r� r) r( r* r+ rp Zflags2r rI r r r r N s0
c @ s� e Zd ZdZd7dd�Zdd � Zd
d � Zd d
� Zdd� Zdd� Z dd� Z
e dd� �Z e dd� �Z
dd� Zdd� Zd d � Ze d d � �Ze d d!� �Ze d"d#� �Zd$d%� Zd&d'� Zd(d)� Zd*d+� Ze d,d-� �Zd.d/� Zd8d0d1�Zd2d3� Zd4d5� Z dS )9r Fr �w+br� Nc C sH d|krt j� | _nt jdd�| _|| _d| _|||||||d�| _d S )N�b�
)r� F)r� r� r) r( r� r� r* )rZ �BytesIO�_file�StringIO� _max_size�_rolled�_TemporaryFileArgs) r2 �max_sizer� r� r� r� r) r( r* r r r r� � s z SpooledTemporaryFile.__init__c C s, | j r
d S | j}|r(|j� |kr(| j� d S )N)r� r� �tell�rollover)r2 rr r� r r r �_check� s
zSpooledTemporaryFile._checkc C sN | j r
d S | j}tf | j� }| _| `|j|j� � |j|j� d� d| _ d S )Nr T)r� r� r r� r[ �getvalue�seekr� )r2 rr Znewfiler r r r� � s z SpooledTemporaryFile.rolloverc C s | j jrtd��| S )Nz%Cannot enter context with closed file)r� �closed�
ValueError)r2 r r r r� � s z SpooledTemporaryFile.__enter__c C s | j j� d S )N)r� r )r2 r� r� r� r r r r� � s z SpooledTemporaryFile.__exit__c C s
| j j� S )N)r� r4 )r2 r r r r4 � s z SpooledTemporaryFile.__iter__c C s | j j� d S )N)r� r )r2 r r r r � s zSpooledTemporaryFile.closec C s | j jS )N)r� r� )r2 r r r r� � s zSpooledTemporaryFile.closedc C s8 y| j jS tk
r2 d| jd kr(� | jd S X d S )Nr� r� r� )r� r� rO r� )r2 r r r r� � s z SpooledTemporaryFile.encodingc C s | j � | jj� S )N)r� r� �fileno)r2 r r r r� � s zSpooledTemporaryFile.filenoc C s | j j� d S )N)r� �flush)r2 r r r r� � s zSpooledTemporaryFile.flushc C s
| j j� S )N)r� �isatty)r2 r r r r� � s zSpooledTemporaryFile.isattyc C s( y| j jS tk
r" | jd S X d S )Nr� )r� r� rO r� )r2 r r r r� � s zSpooledTemporaryFile.modec C s" y| j jS tk
r d S X d S )N)r� rI rO )r2 r r r rI � s zSpooledTemporaryFile.namec C s8 y| j jS tk
r2 d| jd kr(� | jd S X d S )Nr� r� r� )r� �newlinesrO r� )r2 r r r r� � s z SpooledTemporaryFile.newlinesc G s | j j|� S )N)r� �read)r2 r# r r r r� � s zSpooledTemporaryFile.readc G s | j j|� S )N)r� �readline)r2 r# r r r r� � s z SpooledTemporaryFile.readlinec G s | j j|� S )N)r� � readlines)r2 r# r r r r� � s z SpooledTemporaryFile.readlinesc G s | j j|� d S )N)r� r� )r2 r# r r r r� � s zSpooledTemporaryFile.seekc C s | j jS )N)r� � softspace)r2 r r r r� s z SpooledTemporaryFile.softspacec C s
| j j� S )N)r� r� )r2 r r r r� s zSpooledTemporaryFile.tellc C s6 |d kr| j j� n || jkr&| j� | j j|� d S )N)r� �truncater� r� )r2 �sizer r r r� s
z SpooledTemporaryFile.truncatec C s | j }|j|�}| j|� |S )N)r� r[ r� )r2 �srr �rvr r r r[ s
zSpooledTemporaryFile.writec C s | j }|j|�}| j|� |S )N)r� �
writelinesr� )r2 �iterablerr r� r r r r� s
z SpooledTemporaryFile.writelines���)r r� r� NNNNN)N) r? r@ rA r� r� r� r� r� r� r4 r rB r� r� r� r� r� r� rI r� r� r� r� r� r� r� r� r[ r� r r r r r � s6
c @ sN e Zd Zddd�Zedd� �Zedd� �Zdd � Zd
d � Zd d
� Z dd� Z
dS )r Nc C s0 t |||�| _tj| | j| jdj| �d�| _d S )NzImplicitly cleaning up {!r})� warn_message)r rI �_weakref�finalize�_cleanup�format�
_finalizer)r2 r) r( r* r r r r� ( s
zTemporaryDirectory.__init__c s � �fdd�}t j�|d� d S )Nc s� t |d t�r|yV|�kr(ttjj|�� t|� ytj|� W n" ttfk
r` � j|� Y nX W q� t k
rx Y q�X nt |d t �r�n� d S )Nr )
�
issubclassr^ r{ r rK rR r\ r� �_rmtreerb )rw rK �exc_info)�clsrI r r �onerror0 s z+TemporaryDirectory._rmtree.<locals>.onerror)r� )�_shutilZrmtree)r� rI r� r )r� rI r r� . s zTemporaryDirectory._rmtreec C s | j |� tj|t� d S )N)r� � _warnings�warn�ResourceWarning)r� rI r� r r r r� E s
zTemporaryDirectory._cleanupc C s dj | jj| j�S )Nz <{} {!r}>)r� � __class__r? rI )r2 r r r �__repr__J s zTemporaryDirectory.__repr__c C s | j S )N)rI )r2 r r r r� M s z TemporaryDirectory.__enter__c C s | j � d S )N)�cleanup)r2 r� r� r� r r r r� P s zTemporaryDirectory.__exit__c C s | j j� r| j| j� d S )N)r� �detachr� rI )r2 r r r r� S s
zTemporaryDirectory.cleanup)NNN) r? r@ rA r� � classmethodr� r� r� r� r� r� r r r r r s
)NNNF)NNNr� )r� r� NNNNNTr� )r� r� NNNNN)C�__all__� functoolsr� �warningsr� �iorZ �osr Zshutilr� �errnorc Zrandomr r1 �weakrefr� �_thread� ImportErrorZ
_dummy_thread�
allocate_lockZ_allocate_lock�O_RDWRr� �O_EXCLr| r� r rY r r r& rj r r r r r% r, r- rS rh ri rm rs rx r{ r r r
r r
r r r r r� r rI �sys�platformr r� r �objectr r r r r �<module> s�
-
(
' +?
$
8