U |Ha@sJddlZddlZddlZddlZddlZz ddlZWnek rLdZYnXzddlmZWnek rvdZYnXdZej ddkZ e re ne Z ddZe rdndZejZejd kreed rd d Zd dZddZddZn4ddlmZmZdZdZeZddZddZddZddZddZGddde Z!e!fd d!Z"dS)"N)fspathz1.4.0cCst|ts|tS|SN) isinstance text_typedecodesysgetfilesystemencoding)xr 9/tmp/pip-unpacked-wheel-fm3mek5y/atomicwrites/__init__.py_path_to_unicodes r wbwwin32 F_FULLFSYNCcCst|tjdSr)fcntlr)fdr r r _proper_fsync(src Cs*t|d}z t|W5t|XdS)Nr)osopencloser) directoryrr r r _sync_directory.s  rcCs(t||ttjtj|dSr)rrenamerpathnormpathdirnamesrcdstr r r _replace_atomic6s r!cCsZt||t|tjtj|}tjtj|}t|||krVt|dSr)rlinkunlinkrrrr)rr src_dirZdst_dirr r r _move_atomic:s  r%)windllWinErrorcCs|s tdSr)r')rvr r r _handle_errorsJsr+cCs$ttjt|t|ttBdSr)r+r&kernel32 MoveFileExWr _windows_default_flags_MOVEFILE_REPLACE_EXISTINGrr r r r!Ns cCs ttjt|t|tdSr)r+r&r,r-r r.rr r r r%Ts cCs t||S)z Move ``src`` to ``dst``. If ``dst`` exists, it will be silently overwritten. Both paths must reside on the same filesystem for the operation to be atomic. )r!rr r r replace_atomic[sr0cCs t||S)a Move ``src`` to ``dst``. There might a timewindow where both filesystem entries exist. If ``dst`` already exists, :py:exc:`FileExistsError` will be raised. Both paths must reside on the same filesystem for the operation to be atomic. )r%rr r r move_atomicfs r1c@s`eZdZdZedfddZddZejddZ d e d fd d Z d dZ ddZddZd S) AtomicWriteraY A helper class for performing atomic writes. Usage:: with AtomicWriter(path).open() as f: f.write(...) :param path: The destination filepath. May or may not exist. :param mode: The filemode for the temporary file. This defaults to `wb` in Python 2 and `w` in Python 3. :param overwrite: If set to false, an error is raised if ``path`` exists. Errors are only raised after the file has been written to. Either way, the operation is atomic. If you need further control over the exact behavior, you are encouraged to subclass. FcKs\d|krtdd|kr tdd|kr0tdtdk r@t|}||_||_||_||_dS)NazAppending to an existing file is not supported, because that would involve an expensive `copy`-operation to a temporary file. Open the file in normal `w`-mode and copy explicitly if that's what you're after.r z&Use the `overwrite`-parameter instead.rz%AtomicWriters can only be written to.) ValueErrorr_path_mode _overwrite _open_kwargs)selfrmode overwriteZ open_kwargsr r r __init__szAtomicWriter.__init__cCs ||jS)z* Open the temporary file. )_openget_fileobject)r9r r r rszAtomicWriter.openccsrd}z>d}|f|j}|V||W5QRX||d}W5|slz||Wntk rjYnXXdS)NFT)rollback Exceptionr8synccommit)r9r>fsuccessr r r r=s zAtomicWriter._openNcKsZ|dkrtjtj|j}tj|||d\}}t||j|d<||d<t j f|S)z!Return the temporary file to use.N)suffixprefixdirr:file) rrrrr5tempfilemkstemprr6ior)r9rFrGrHkwargsZ descriptornamer r r r>s   zAtomicWriter.get_fileobjectcCs|t|dS)zNresponsible for clearing as many file caches as possible before commitN)flushrfilenor9rCr r r rAszAtomicWriter.synccCs(|jrt|j|jnt|j|jdS)z/Move the temporary file to the target location.N)r7r0rNr5r1rQr r r rBszAtomicWriter.commitcCst|jdS)z!Clean up all temporary resources.N)rr#rNrQr r r r?szAtomicWriter.rollback)__name__ __module__ __qualname____doc__ DEFAULT_MODEr<r contextlibcontextmanagerr=rJ gettempprefixr>rArBr?r r r r r2rs  r2cKs||f|S)a Simple atomic writes. This wraps :py:class:`AtomicWriter`:: with atomic_write(path) as f: f.write(...) :param path: The target path to write to. :param writer_cls: The writer class to use. This parameter is useful if you subclassed :py:class:`AtomicWriter` to change some behavior and want to use that new subclass. Additional keyword arguments are passed to the writer class. See :py:class:`AtomicWriter`. )r)rZ writer_clsZ cls_kwargsr r r atomic_writesrZ)#rWrLrrrJr ImportErrorr __version__ version_infoPY2unicodestrrr rVfsyncrplatformhasattrrr!r%ctypesr&r'r/Z_MOVEFILE_WRITE_THROUGHr.r+r0r1objectr2rZr r r r sF          a