o i,@sdZddlmZddlZddlZddlZddlmZddlm Z m Z ddl Z ddl m Z ddlmZddlmZdd lmZmZd d Zdd d ZddZddZedddZedddZGdddeZdS)z9 Utilities for file-based Contents/Checkpoints managers. )contextmanagerN) HTTPError) to_api_path to_os_path)str_to_unicode) Configurable)Bool) encodebytes decodebytescCsJttdr t||dStjdkrtj|rt|t||dS)zh replace dst with src switches between os.replace or os.rename based on python 2.7 or python 3 replacentN)hasattrosr namepathexistsremoverename)srcdstr\/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/notebook/services/contents/fileio.py replace_files  rcCsNt||z t||WdSty&|r#|jd|ddYdSYdSw)zZcopy src to dst like shutil.copy2, but log errors in copystat instead of raising zcopystat on %s failedT)exc_infoN)shutilcopyfilecopystatOSErrordebug)rrlogrrr copy2_safe*s  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.~rrsplitjoinrdirnamebasenamerrrpath_to_intermediate6sr'cCs"tj|\}}tj||dS)zEName of invalid file after a failed atomic write and subsequent read.z.invalidr!r$rrrpath_to_invalid=sr(Tutf-8ckstj|rtjtj|t|}t|}tj|r&t|||d|r:| ddt |dfd|i|}n t |dfi|}z|VWn | t ||| t|| tj|rrt|dSdS)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)rrislinkr#r%readlinkr'isfiler setdefaultopencloserflushfsyncfilenor)rtextr.rkwargstmp_pathfileobjrrratomic_writingBs,      r=ckstj|rtjtj|t|}|r)|ddt|dfd|i|}n t|dfi|}z|VWn||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.r/N) rrr0r#r%r1r3r4r5)rr9r.rr:r<rrr_simple_writings    r>c@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 both notebooks and generic files. Shared by FileContentsManager and FileCheckpoints. Note ---- Classes using this mixin must provide the following attributes: root_dir : unicode A directory against which API-style paths are to be resolved. log : logging.Logger TaBy default notebooks are saved on disk on a temporary file and then if successfully written, it replaces the old ones. This procedure, namely 'atomic_writing', causes some bugs on file system without 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 ))confighelpc os~||/t|g|Ri| }|VWdn1s wYWddSWddS1s8wYdS)zrCrrrr=s "zFileManagerMixin.atomic_writingc csnzdVWdSty6}z#|jtjtjhvr1|s!t|jpd}t||jd}tdd||d}~ww)z7context manager for turning permission errors into 403.Nz unknown file)rootizPermission denied: ) rerrnoEPERMEACCESrfilenamerroot_dirr)rDrEerrrrrBs zFileManagerMixin.perm_to_403cCst|||jddS)zPcopy src to dest like shutil.copy2, but log errors in copystat r*N)r r)rDrdestrrr_copyszFileManagerMixin._copycCsFtj|j}t||}tj|tjj|s!td|d|S)aYGiven an API path, return its file system path. Parameters ---------- path : string The relative API path to the named file. Returns ------- path : string Native, absolute OS path to for a file. Raises ------ 404: if path is outside root iz# is outside root contents directory)rrabspathrOrsep startswithr)rDrrJrErrr _get_os_paths  zFileManagerMixin._get_os_pathc Cs|j|dddX}ztj||dWWdSty,}z|}WYd}~nd}~wwt|}|jr:tj|sEt dd|d|t |}t ||t ||| ||WdS1scwYdS) z Read a notebook from an os path.rr)r.) as_versionNzUnreadable Notebook:  ) r4nbformatread Exceptionr'rHrrrrr(r_read_notebook)rDrErZrGrPe_origr; invalid_filerrrr`s&   $zFileManagerMixin._read_notebookcCsF|j|dd}tj||tjdWddS1swYdS)zSave a notebook to an os_path.r)rY)versionN)r=r]write NO_CONVERT)rDrEnbrGrrr_save_notebooks"zFileManagerMixin._save_notebookc Cstj|stdd|||d }|}Wdn1s#wY|dus0|dkrZz|ddfWStyY}z|dkrOtd|ddd |WYd}~nd}~wwt|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 rbNr9utf8z is not UTF-8 encodedz bad format)reasonasciibase64) rrr2rr4r^decode UnicodeErrorr )rDrEformatrGbcontentrPrrr _read_file$s*   zFileManagerMixin._read_filec Cs|dvr tddz|dkr|d}n |d}t|}Wnty6}z tdd|d||d }~ww|j|d d }||Wd d S1sOwYd S) zSave content of a generic file.>r9rlr[z:Must specify format of file contents as 'text' or 'base64'r9rirkzEncoding error saving z: NF)r9)rencoder r_r=rd)rDrEcontentrorp b64_bytesrPrGrrr _save_fileAs,   "zFileManagerMixin._save_fileN)rI)rW)__name__ __module__ __qualname____doc__rrHrr4r=rBrRrVr`rgrqrurrrrr?s"     r?)N)Tr)N)ry contextlibrrKrr tornado.webrnotebook.utilsrrr]ipython_genutils.py3compatrtraitlets.configr traitletsrrlr r rr r'r(r=r>r?rrrrs*        < +