a mb&;@sdZddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl Z ddl mZddlmZdd lmZdd lmZdd lmZdd lmZd dZddZd#ddZd$ddZddZddZe d%ddZe d&ddZGdd d eZ Gd!d"d"e Z!dS)'z9 Utilities for file-based Contents/Checkpoints managers. N) decodebytes) encodebytes)contextmanager)partial)run_sync) HTTPError)Bool) Configurable) to_api_path) to_os_pathcCst||dS)zreplace dst with srcN)osreplacesrcdstrFlib/python3.9/site-packages/jupyter_server/services/contents/fileio.py replace_filesrcsttj||IdHdS)z#replace dst with src asynchronouslyN)rr r rrrrasync_replace_filesrcCsHt||zt||Wn&tyB|r>|jd|ddYn0dS)zZcopy src to dst like shutil.copy2, but log errors in copystat instead of raising copystat on %s failedTexc_infoN)shutilcopyfilecopystatOSErrordebugrrlogrrr copy2_safe#s   rcsXttj||IdHzttj||IdHWn&tyR|rN|jd|ddYn0dS)zicopy src to dst asynchronously like shutil.copy2, but log errors in copystat instead of raising NrTr)rrrrrrrrrrasync_copy2_safe0s  r cCs"tj|\}}tj|d|S)ztName of the intermediate file used in atomic writes. The .~ prefix will make Dropbox ignore the temporary file.z.~r pathsplitjoinr"dirnamebasenamerrrpath_to_intermediate=sr(cCs"tj|\}}tj||dS)zEName of invalid file after a failed atomic write and subsequent read.z.invalidr!r%rrrpath_to_invalidEsr)Tutf-8ckstj|r(tjtj|t|}t|}tj|rJt|||d|rt| ddt j |dfd|i|}nt j |dfi|}z |VWn | t ||Yn0|t|| tj|rt|dS)aContext manager to write to a file only if the entire write is successful. This works by copying the previous file contents to a temporary file in the same directory, and renaming that file back to the target if the context exits with an error. If the context is successful, the new data is synced to disk and the temporary file is removed. Parameters ---------- path : str The target file to write to. text : bool, optional Whether to open the file in text mode (i.e. to write unicode). Default is True. encoding : str, optional The encoding to use for files opened in text mode. Default is UTF-8. **kwargs Passed to :func:`io.open`. rnewline wencodingwbN)r r"islinkr$r&readlinkr(isfiler setdefaultioopencloserflushfsyncfilenoremove)r"textr/rkwargstmp_pathfileobjrrratomic_writingKs(      r@ckstj|r(tjtj|t|}|rR|ddtj|dfd|i|}ntj|dfi|}z |VWn| Yn0| dS)aContext manager to write file without doing atomic writing (for weird filesystem eg: nfs). Parameters ---------- path : str The target file to write to. text : bool, optional Whether to open the file in text mode (i.e. to write unicode). Default is True. encoding : str, optional The encoding to use for files opened in text mode. Default is UTF-8. **kwargs Passed to :func:`io.open`. r,r-r.r/r0N) r r"r1r$r&r2r4r5r6r7)r"r<r/rr=r?rrr_simple_writings   rAc@sveZdZdZeddddZeddZeddZedd d Z d d Z ddZ dddZ ddZ ddZddZdS)FileManagerMixina Mixin for ContentsAPI classes that interact with the filesystem. Provides facilities for reading, writing, and copying files. Shared by FileContentsManager and FileCheckpoints. Note ---- Classes using this mixin must provide the following attributes: root_dir : unicode A directory against against which API-style paths are to be resolved. log : logging.Logger Ta}By default notebooks are saved on disk on a temporary file and then if succefully written, it replaces the old ones. This procedure, namely 'atomic_writing', causes some bugs on file system whitout operation order enforcement (like some networked fs). If set to False, the new notebook is written directly on the old one which could fail (eg: full filesystem or quota ))Zconfighelpc osl||Ntj|g|Ri|}|VWdn1s@0YWdn1s^0YdS)z|jpZ invalid_filerrrres   zFileManagerMixin._read_notebookcCsD|j|dd"}tj||tjdWdn1s60YdSzSave a notebook to an os_path.r*r^)versionN)r@rawrite NO_CONVERTrFrGnbrIrrr_save_notebook!szFileManagerMixin._save_notebookc Cstj|stdd|||d}|}Wdn1sD0Y|dus^|dkrz|ddfWSty}z(|dkrtdd|dd |WYd}~n d}~00t|d d fS a*Read a non-notebook file. os_path: The path to be read. format: If 'text', the contents will be decoded as UTF-8. If 'base64', the raw bytes contents will be encoded as base64. If not specified, try to decode as UTF-8, and fall back to base64 r`zCannot read non-file %srbNr<utf8z%s is not UTF-8 encodedz bad format)reasonasciibase64) r r"r3rr6rbdecode UnicodeErrorrrFrGformatrIbcontentrRrrr _read_file&s" &zFileManagerMixin._read_filec Cs|dvrtddz*|dkr(|d}n|d}t|}Wn8tyt}z tdd||f|WYd}~n d}~00|j|d d }||Wdn1s0YdS zSave content of a generic file.>r<rsr`z:Must specify format of file contents as 'text' or 'base64'r<rprrzEncoding error saving %s: %sNF)r<)rencoderrcr@rirFrGZcontentrwrxZ b64_bytesrRrIrrr _save_fileCs   *zFileManagerMixin._save_fileN)rK)r[)__name__ __module__ __qualname____doc__rrJrr6r@rDrVrZrermryr}rrrrrBs$    rBc@s:eZdZdZddZdddZddZd d Zd d Zd S)AsyncFileManagerMixinzY Mixin for ContentsAPI classes that interact with the filesystem asynchronously. cst|||jdIdHdSrS)r rrTrrrrV\szAsyncFileManagerMixin._copyr[c s|j|ddd}z*tttj|d|IdHWWdStyd}z|}WYd}~n d}~00t|}|jrtj |st dd||ft |}t ||IdHt ||IdH|||IdHWdS1s0YdSr\)r6rrrarbrcr(rJr r"rdrr)rrerfrrrrecs* z$AsyncFileManagerMixin._read_notebookcsP|j|dd.}tttjtjd||IdHWdn1sB0YdSrg)r@rrrarirjrkrrrrm|sz$AsyncFileManagerMixin._save_notebookc stj|stdd|||d }t|jIdH}Wdn1sL0Y|dusf|dkrz|ddfWSty}z(|dkrtdd|dd |WYd}~n d}~00t |d d fSrn) r r"r3rr6rrbrtrurrvrrrrys" .z AsyncFileManagerMixin._read_filec s|dvrtddz*|dkr(|d}n|d}t|}Wn8tyt}z tdd||f|WYd}~n d}~00|j|d d "}t|j|IdHWdn1s0YdSrz)rr{rrcr@rrir|rrrr}s   *z AsyncFileManagerMixin._save_fileN)r[) r~rrrrVrermryr}rrrrrWs  r)N)N)Tr*N)Tr*N)"rrNr5r rrsrr contextlibr functoolsrraZanyio.to_threadrZ tornado.webrZ traitletsrZtraitlets.configr Zjupyter_server.utilsr r rrrr r(r)r@rArBrrrrrs8            8 &-