3
_Vjm @ 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 Zeedreej O ZeZ eedre ej O Z eedrej Z ndZ dZ e Z!eedrej"Z#n eedrej$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 Z4dCd1dZ5dDd2dZ6d3e dfd4dZ7G d5d6 d6Z8G d7d8 d8Z9dFdkre:Z>need?a?dHd@dZ>G dAd dZ@G dBd deAZBdS )INamedTemporaryFile
TemporaryFileSpooledTemporaryFileTemporaryDirectorymkstempmkdtempmktempTMP_MAX
gettempprefixtempdir
gettempdirgettempprefixb gettempdirb N)Random
O_NOFOLLOWO_BINARYi' Ztmplstatstatc C s t j| t j}t j| d S )N)_osopenO_RDONLYclose)fnfd 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 |tr8|tkr2tdt}q
|tkrHtdt}q
W |d kr\tS |S )Nz1Can't mix bytes and non-bytes in path components.)
isinstancebytesstr TypeError)argsZ return_typeargr 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
)prefixsuffixdir 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 getpidgetattr_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 fddtdD }dj|S )Nc s g | ] } qS r r ).0Zdummy)cchooser r
s z0_RandomNameSequence.__next__.. )
charactersr3 Zchoicerangejoin)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 jjdt jjdddd d
g n| jd d d
g y| jt j W n$ t t
fk
r | jt j Y nX | S )NTMPDIRTEMPTMPntz~\AppData\Local\Tempz%SYSTEMROOT%\Tempzc:\tempzc:\tmpz\tempz\tmpz/tmpz/var/tmpz/usr/tmp)rC rD rE ) r getenvappendnameextendpath
expanduser
expandvarsgetcwdAttributeErrorr curdir)dirlistZenvnamedirnamer r r _candidate_tempdir_list s
rS c C s4 t } t }x|D ]}|tjkr0tjj|}xtdD ]}t| }tjj||}y\tj |t
d}z t k
r tj
dkrtjj| rtj| tjrw$n Y nX | tjj|fS W t tjdd 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* ZpreZsufflagsr+ 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 )Nfollow_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|}xttD ]}t|}tj j
||| | }ytj |d W nR t k
r w8Y n> t
k
r tjdkrtj j|rtj|tjrw8n Y nX |S W t tjdd 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:ttD ].}t|}tjj||| | }t|s |S q W t t
j dd 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 ddZejdkr | 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 ZdddZdd 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|drD| tj fdd}| j|_|}t|tsZt| || |S )Nrr __call__c s
| |S )Nr )r# kwargs)rw r r func_wrapper s z7_TemporaryFileWrapper.__getattr__..func_wrapper) __dict__r0 hasattr
_functoolswrapsr r intsetattr)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 excvaluetbresultr 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 ) bufferingnewlineencoding) 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
posixcygwin O_TMPFILEc C s t |||\}}}}t}try$|tjB tj @ } tj|| d}
W n* tk
rX daY nB tk
rj Y n0X yt j|
| |||dS tj
|
Y nX t |||||\}
} y tj | t j|
| |||dS 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 ZdZd7ddZdd 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/ Zd8d0d1Zd2d3 Zd4d5 Z dS )9r Fr w+br Nc C sH d|krt j | _nt jdd| _|| _d| _|||||||d| _d S )Nb
)r F)r r r) r( r r r* )rZ BytesIO_fileStringIO _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 tellrollover)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[ getvalueseekr )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 ZdddZedd 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 _weakreffinalize_cleanupformat
_finalizer)r2 r) r( r* r r r r ( s
zTemporaryDirectory.__init__c s fdd}t j|d d S )Nc s t |d tr|yV|kr(ttjj| t| ytj| W n" ttfk
r` j| Y nX W q t k
rx Y qX nt |d t rn d S )Nr )
issubclassr^ r{ r rK rR r\ r _rmtreerb )rw rK exc_info)clsrI r r onerror0 s z+TemporaryDirectory._rmtree..onerror)r )_shutilZrmtree)r rI r r )r rI r r . s zTemporaryDirectory._rmtreec C s | j | tj|t d S )N)r _warningswarnResourceWarning)r rI r r r r r E s
zTemporaryDirectory._cleanupc C s dj | jj| jS )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_lockO_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 sysplatformr r r objectr r r r r s
-
(
' +?
$
8