3
Lj} @ 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 edre ej O Z e Z e edre ej!O Z e edrej"Z"ndZ"dZ#e Z$e edrej%Z&n e edr$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. Z2ej3ej4ejhej5koej3ej6kZ7d/d0 Z8yej9ej:B ej B Z;W n ed4d Z?da@d5d
ZAd6d ZBdJd8dZCdKd9dZDd:e#dfd;dZEG dd? d?ZGdMdCd ZHejIdDks|ejJjKdEkreHZLne edFaMdOdGdZLG dHd dZNG dId deOZPdS )PNamedTemporaryFile
TemporaryFileSpooledTemporaryFileTemporaryDirectorymkstempmkdtempmktempTMP_MAX
gettempprefixtempdir
gettempdirgettempprefixb gettempdirb N)Random)S_ISLNK
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_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 |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_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
)prefixsuffixdir 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 getpidgetattr_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 fddtdD }dj|S )Nc s g | ] } qS r r ).0Zdummy)cchooser r
s z0_RandomNameSequence.__next__.. )
charactersr4 Zchoicerangejoin)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 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)rD rE rF ) r getenvappendnameextendpath
expanduser
expandvarsgetcwdAttributeErrorr curdir)dirlistZenvnamedirnamer r r _candidate_tempdir_list s
rT c C s4 t } t }x|D ]}|tjkr0tjj|}xtdD ]}t| }tjj||}y\tj |t
d}z r _bin_openflags_iowriter unlinkFileExistsErrorPermissionErrorrJ isdiraccessW_OKr FileNotFoundError_errnoENOENT)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_lockacquirer. releaser r r r _get_candidate_names s
rn c
C s t }|tkrttj|}xttD ]}t|}tjj | || | }ytj
||d} W nR t k
rn w$Y n> t k
r tj
dkrtjj| rtj| tjrw$n Y nX | tjj|fS W t tjdd 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+ ZpreZsufflagsr, 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 )Nfollow_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 tjdkrtj j|rtj|tjrw8n Y nX |S W t tjdd 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:ttD ].}t|}tjj||| | }t|s |S q W t t
j dd 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 ddZejdkr | 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 ZdddZdd 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|drD| tj fdd}| j|_|}t|tsZt| || |S )Nrs __call__c s
| |S )Nr )r$ kwargs)rx r r func_wrapper+ s z7_TemporaryFileWrapper.__getattr__..func_wrapper) __dict__r1 hasattr
_functoolswrapsr r intsetattr)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 excvaluetbresultr 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 ) bufferingnewlineencoding) 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
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- 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 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+ )r[ BytesIO_fileStringIO _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 tellrollover)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\ getvalueseekr )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 ZdddZedddZedd 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 _weakreffinalize_cleanupformat
_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 trLr&|kr&r$d S |td }|d ksHt rT|d }}n|d krb|}n|}y|krt|tjj| t||| ytj ||d W nn t
k
r j ||k||d Y nD tk
r tjj |srd S j ||k||d Y nX W n6 t
k
r( Y n" tk
rH rBd S Y nX nt |d t
r^nsf d S )Nr )r~ )
ignore_errorsrepeatedr~ fullname)
issubclassr_ len_rmtree_use_dir_fdr r rL rS r r] r _rmtreer` rc r )rx rL exc_infoZ
entry_namer~ Zfullpathbase)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 _warningswarnResourceWarning)r rJ r r r r r s
zTemporaryDirectory._cleanupc C s dj | jj| jS )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_lockO_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 sysplatformr r r objectr r r r r s
-
!
(
' +?
$
8