3
�L�j�} � @ s� d ddddddddd d
d d g
Z d
dlZd
dlZd
dlZd
dlZd
dl Z
d
dl Z d
dl
mZ d
dlmZ d
dlZy d
dlZW n ek
r� d
dlZY nX ejZejejB ej B Z e ed�r�e ej O Z e Z e ed�r�e ej!O Z e ed�r�ej"Z"ndZ"dZ#e� Z$e ed��rej%Z&n e ed��r$ejZ&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*� Z0d+d,� Z1d-d.� Z2ej3ej4ejhej5k�o�ej3ej6kZ7d/d0� Z8yej9ej:B ej B Z;W n ed4d � Z?da@d5d
� ZAd6d � ZBdJd8d�ZCdKd9d�ZDd:e#dfd;d�ZEG dd?� d?�ZGdMdCd �ZHejIdDk�s|ejJjKdEk�r�eHZLne edF�aMdOdGd�ZLG dHd� d�ZNG dId� deO�ZPdS )P�NamedTemporaryFile�
TemporaryFile�SpooledTemporaryFile�TemporaryDirectory�mkstemp�mkdtemp�mktemp�TMP_MAX�
gettempprefix�tempdir�
gettempdir�gettempprefixb� gettempdirb� N)�Random)�S_ISLNK�
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�_statU s r c C s* y t | � W n tk
r dS X dS d S )NFT)r �OSError)r r r r �_existsY 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_typeb 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_paramsw 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 )r3 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 �
� s z0_RandomNameSequence.__next__..� � )�
charactersr4 Zchoice�range�join)r3 Zlettersr )r7 r8 r �__next__� s z _RandomNameSequence.__next__N)�__name__�
__module__� __qualname__r< �propertyr4 r5 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)rD rE rF ) r �getenv�append�name�extend�path�
expanduser�
expandvars�getcwd�AttributeErrorr �curdir)�dirlistZenvname�dirnamer r r �_candidate_tempdir_list� s
rT c C s4 t � } t� }�x|D �]}|tjkr0tjj|�}x�td�D ]�}t| �}tjj||�}y\tj |t
d�}z r �_bin_openflags�_io�writer �unlink�FileExistsError�PermissionErrorrJ �isdir�access�W_OKr �FileNotFoundError�_errno�ENOENT)ZnamerrR r+ �seqrJ �filenamer �fpr r r �_get_default_tempdir� s:
ri 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
rn 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� rG z#No usable temporary file name found)rn r! �mapr r( r= r rY rL r> r r^ r_ rJ r` ra rb rX rd �EEXIST)
r+ ZpreZsuf�flagsr, �namesrf rJ �filer r r r �_mkstemp_inner� s$ rt c G sJ | t jkr | |f|�ddi� n&t jdks8t jj|� rF| |f|�� d S )N�follow_symlinksFrG )r �supports_follow_symlinksrJ rL �islink)�funcrL r$ r r r �_dont_follow_symlinks s
ry c C s0 y
t j}W n tk
r Y nX t|| d� d S )Nr )r �chflagsrP ry )rL rz r r r � _resetflags s
r{ c C s t | � ttj| d� d S )Ni� )r{ ry r �chmod)rL r r r � _resetperms( s r} c C s, | d krt |� d S t|� tj| d� d S )Ni� )r} r{ r r| )�dir_fdrL r r r �_resetperms_fd3 s
r c C s� |d krt |� d S t|� tjtjkr t
k
r� tjdkr�tj j|�r�tj|tj�r�w8n� Y nX |S W t tjd��d S )Ni� rG z(No usable temporary directory name found)r- rn r! ro r r( r= r rY rL r> �mkdirr^ r_ rJ r` ra rb rd rp )r* r) r+ r, rr rf rJ rs r r r r � 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 rn r= r rY r rL r> r r^ rd rp )r* r) r+ rr rf rJ rs r r r r � s c @ sH e Zd ZdZdZd dd�Zejdkr | j r:| jd k r:d| _ z| jj� W d | jr8|| j� X d S )NT)� close_calledrs r r� rJ )r3 r] r r r r s z_TemporaryFileCloser.closec C s | j � d S )N)r )r3 r r r �__del__ s z _TemporaryFileCloser.__del__c C s | j sd| _ | jj� d S )NT)r� rs r )r3 r r r r s )T) r@ rA rB rs r� r� r rJ 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)rs rJ r� r� �_closer)r3 rs rJ r� r r r r� s z _TemporaryFileWrapper.__init__c s^ | j d }t||�}t|d�rD|� tj� �� fdd��}| j|_|}t|t�sZt| ||� |S )Nrs �__call__c s
� | |�S )Nr )r$ �kwargs)rx r r � func_wrapper+ s z7_TemporaryFileWrapper.__getattr__..func_wrapper) �__dict__r1 �hasattr�
_functools�wrapsr� r �int�setattr)r3 rJ rs �ar� r )rx r � __getattr__# s
z!_TemporaryFileWrapper.__getattr__c C s | j j� | S )N)rs � __enter__)r3 r r r r� 8 s
z _TemporaryFileWrapper.__enter__c C s | j j|||�}| j� |S )N)rs �__exit__r )r3 �exc�value�tb�resultr r r r� > s z _TemporaryFileWrapper.__exit__c C s | j j� d S )N)r� r )r3 r r r r C s z_TemporaryFileWrapper.closec c s x| j D ]
}|V qW d S )N)rs )r3 �liner r r r5 J s z _TemporaryFileWrapper.__iter__N)T) r@ rA rB r� r� r� r� r r5 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 )NrG )� buffering�newline�encoding) r- rZ r rJ Z O_TEMPORARYrt r[ r r� �
BaseExceptionr] r )
�moder� r� r� r* r) r+ r� r, rq r rJ rs r r r r T 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- rZ �_O_TMPFILE_WORKSr r� �O_CREATr �IsADirectoryErrorr r[ r rt r] ) r� r� r� r� r* r) r+ r, rq Zflags2r rJ r r r r � 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+ )r[ �BytesIO�_file�StringIO� _max_size�_rolled�_TemporaryFileArgs) r3 �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)r3 rs 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� )r3 rs Znewfiler r r r� � s z SpooledTemporaryFile.rolloverc C s | j jrtd��| S )Nz%Cannot enter context with closed file)r� �closed�
ValueError)r3 r r r r� � s z SpooledTemporaryFile.__enter__c C s | j j� d S )N)r� r )r3 r� r� r� r r r r� � s z SpooledTemporaryFile.__exit__c C s
| j j� S )N)r� r5 )r3 r r r r5 � s z SpooledTemporaryFile.__iter__c C s | j j� d S )N)r� r )r3 r r r r � s zSpooledTemporaryFile.closec C s | j jS )N)r� r� )r3 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� rP r� )r3 r r r r� s z SpooledTemporaryFile.encodingc C s | j � | jj� S )N)r� r� �fileno)r3 r r r r�
s zSpooledTemporaryFile.filenoc C s | j j� d S )N)r� �flush)r3 r r r r� s zSpooledTemporaryFile.flushc C s
| j j� S )N)r� �isatty)r3 r r r r� s zSpooledTemporaryFile.isattyc C s( y| j jS tk
r" | jd S X d S )Nr� )r� r� rP r� )r3 r r r r� s zSpooledTemporaryFile.modec C s" y| j jS tk
r d S X d S )N)r� rJ rP )r3 r r r rJ s zSpooledTemporaryFile.namec C s8 y| j jS tk
r2 d| jd kr(� | jd S X d S )Nr� r� r� )r� �newlinesrP r� )r3 r r r r� " s z SpooledTemporaryFile.newlinesc G s | j j|� S )N)r� �read)r3 r$ r r r r� + s zSpooledTemporaryFile.readc G s | j j|� S )N)r� �readline)r3 r$ r r r r� . s z SpooledTemporaryFile.readlinec G s | j j|� S )N)r� � readlines)r3 r$ r r r r� 1 s z SpooledTemporaryFile.readlinesc G s | j j|� d S )N)r� r� )r3 r$ r r r r� 4 s zSpooledTemporaryFile.seekc C s | j jS )N)r� � softspace)r3 r r r r� 7 s z SpooledTemporaryFile.softspacec C s
| j j� S )N)r� r� )r3 r r r r� ; s zSpooledTemporaryFile.tellc C s6 |d kr| j j� n || jkr&| j� | j j|� d S )N)r� �truncater� r� )r3 �sizer r r r� > s
z SpooledTemporaryFile.truncatec C s | j }|j|�}| j|� |S )N)r� r\ r� )r3 �srs �rvr r r r\ F s
zSpooledTemporaryFile.writec C s | j }|j|�}| j|� |S )N)r� �
writelinesr� )r3 �iterablers r� r r r r� L s
z SpooledTemporaryFile.writelines�����)r r� r� NNNNN)N) r@ rA rB r� r� r� r� r� r� r5 r rC r� r� r� r� r� r� rJ r� r� r� r� r� r� r� r� r\ r� r r r r r � s6
c @ sP e Zd Zddd�Zeddd��Zedd� �Zd d
� 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 rJ �_weakref�finalize�_cleanup�format�
_finalizer)r3 r* r) r+ r r r r� _ s
zTemporaryDirectory.__init__Fc s8 �d kr ��d� ����fdd� }t j�||dd� d S )Nc sj t |d t��rL�r&|�kr&�r$d S � �|t��d � }|d ksHt rT|d }}n|d krb|}n|}y�|�kr�t|tjj|�� t|||� ytj ||d� W nn t
k
r� � j |�|�k||d� Y nD tk
�r tjj |�s�r�d S � � j |�|�k||d� Y nX W n6 t
k
�r( Y n" tk
�rH ��rBd S � Y nX nt |d t
��r^n��sf� d S )Nr )r~ )�
ignore_errors�repeatedr~ �fullname)�
issubclassr_ �len�_rmtree_use_dir_fdr r rL rS r� r] r� �_rmtreer` rc r )rx rL �exc_infoZ
entry_namer~ Zfullpath�base)�clsr� r� rJ r� r r �onerrork sN
z+TemporaryDirectory._rmtree..onerrorT)r� r~ Z
_onexc_kwargs)NN)�_shutilZrmtree)r� rJ r� r� r~ r� r� r )r� r� r� rJ r� r r� e s
;
zTemporaryDirectory._rmtreec C s | j |� tj|t� d S )N)r� � _warnings�warn�ResourceWarning)r� rJ r� r r r r� � s
zTemporaryDirectory._cleanupc C s dj | jj| j�S )Nz <{} {!r}>)r� � __class__r@ rJ )r3 r r r �__repr__� s zTemporaryDirectory.__repr__c C s | j S )N)rJ )r3 r r r r� � s z TemporaryDirectory.__enter__c C s | j � d S )N)�cleanup)r3 r� r� r� r r r r� � s zTemporaryDirectory.__exit__c C s | j j� r| j| j� d S )N)r� �detachr� rJ )r3 r r r r� � s
zTemporaryDirectory.cleanup)NNN)FFNN) r@ rA rB r� � classmethodr� r� r� r� r� r� r r r r r S s
B )NNNF)NNNr� )r� r� NNNNNTr� )r� r� NNNNN)Q�__all__� functoolsr� �warningsr� �ior[ �osr Zshutilr� �errnord Zrandomr r2 r r r� �weakrefr� �_thread� ImportErrorZ
_dummy_thread�
allocate_lockZ_allocate_lock�O_RDWRr� �O_EXCLr� r� r rZ r r r' rk r r r r& r- r. rT ri rj rn rt ry r{ r} r| r] �supports_dir_fd� supports_fdr� r r �
O_NONBLOCKr� rP r� r r r
r r
r r r r� r� r rJ �sys�platformr r� r �objectr r r r r � s�
-
!
(
' +?
$
8