Mh*RLdZddlmZddlZddlZddlZddlZddlmZm Z ddl m Z ddl m Z ddlZddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZmZmZd ZdZddZ ddZ!dZ"dZ#e ddZ$e ddZ%GddeeZ&Gdde&Z'dS)z9 Utilities for file-based Contents/Checkpoints managers. ) annotationsN) decodebytes encodebytes)contextmanager)partial)run_sync) HTTPError)BoolEnum) Configurable)LoggingConfigurable)ApiPath to_api_path to_os_pathc0tj||dS)zreplace dst with srcN)osreplacesrcdsts g/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/jupyter_server/services/contents/fileio.py replace_filersJsCcLKttj||d{VdS)z#replace dst with src asynchronouslyN)rrrrs rasync_replace_filer s2 2:sC ( ((((((((((rcdtj|tjs7|r|d|dt t jd|tj|| tj ||dS#t$r!|r|d|dYdSYdSwxYw)zZcopy src to dst like shutil.copy2, but log errors in copystat instead of raising Source file, %s, is not writableTexc_infoFile is not writable: copystat on %s failedN) raccessW_OKdebugPermissionErrorerrnoEACCESshutilcopyfilecopystatOSErrorrrlogs r copy2_safer.%s 9S"' " "L  N II8#I M M Mel,JS,J,JKKK OCCS!!!!! CCC  C II-sTI B B B B B B C C CCs-B$B/.B/cKtj|tjs7|r|d|dt t jd|ttj ||d{V ttj ||d{VdS#t$r!|r|d|dYdSYdSwxYw)zicopy src to dst asynchronously like shutil.copy2, but log errors in copystat instead of raising rTrr Nr!) rr"r#r$r%r&r'rr(r)r*r+r,s rasync_copy2_safer08s 9S"' " "L  N II8#I M M Mel,JS,J,JKKK 6?C - --------CvS11111111111 CCC  C II-sTI B B B B B B C C CCs;!B$C C ctj|\}}tj|d|zS)ztName of the intermediate file used in atomic writes. The .~ prefix will make Dropbox ignore the temporary file.z.~rpathsplitjoinr3dirnamebasenames rpath_to_intermediater9Js6 d++GX 7<< 1 11rctj|\}}tj||dzS)zEName of invalid file after a failed atomic write and subsequent read.z.invalidr2r6s rpath_to_invalidr;Rs5 d++GX 7<<J!6 7 77rTutf-8c+"Ktj|rOtjtj|tj|}tj|ptj}tj|rtj|tj s{tj|tj r\|rdnd}|rt||fd|i|}nt||fi|} |V| n#| wxYwdSt|}tj|rt||||r'|ddt|dfd|i|}nt|dfi|} |Vn3#t$r&| t!||wxYw|tj|| tj|rtj|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`. wwbencodingNr-newline )rr3islinkr5r7readlinkgetcwdisfiler"r#opencloser9r. setdefault BaseExceptionrflushfsyncfilenoremove) r3textr@r-kwargsdirpathmodefileobjtmp_paths ratomic_writingrVXsq0 w~~dFw||BGOOD112;t3D3DEEgood##2ry{{G w~~d BIgrw$?$? BIdTVT[D\D\ $ss  14CCCFCCGG40000G MMM MMOOOOGMMOOOO#D))H w~~d,4s++++ -)T***tS>>8>v>>tT,,V,,  Xt$$$  MMOOOHW^^   MMOOO w~~h (s(EEG0H c+Ktj|rOtjtj|tj|}|r'|ddt|dfd|i|}nt|dfi|} |Vn##t$r| wxYw| 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`. rBrCr>r@r?N) rr3rDr5r7rErJrHrKrI)r3rPr@r-rQrTs r_simple_writingrXs( w~~dFw||BGOOD112;t3D3DEE -)T***tS>>8>v>>tT,,V,,    MMOOOOOs )B.. CceZdZdZedddZeejdddZ e dZ e d Z e d!d Z d Zd Z d"d#dZd$dZd%dZ d&d'dZd ZdS)(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 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 ))confighelpsha256z:Hash algorithm to use for file content, support by hashlib) default_valuer[r\c/K||5t|g|Ri|5}|Vdddn #1swxYwYddddS#1swxYwYdS)z A> AAAc/:K||5|j|d<|jr-t|g|Ri|5}|Vdddn #1swxYwYn,t |g|Ri|5}|Vdddn #1swxYwYddddS#1swxYwYdS)zwrapper around atomic_writing that turns permission errors to 403. Depending on flag 'use_atomic_writing', the wrapper perform an actual atomic writing or simply writes the file (whatever an old exists or not)r-N)r`r-use_atomic_writingrVrXras rrVzFileManagerMixin.atomic_writings   g & &   HF5M& #G=d===f==GGG%W>t>>>v>>!GGG                   sW"BA  B A BA B'A8, B8A< <B?A< BBBc#K dVdS#t$rZ}|jtjtjhvr5|s |jpd}t ||j}tdd|z|d}~wwxYw)z7context manager for turning permission errors into 403.Nz unknown file)rootizPermission denied: %s)r+r&EPERMr'filenamerroot_dirr )rbrcer3s rr`zFileManagerMixin.perm_to_403s  EEEEE   w5; 555;j:NG"7???%!>!>??T I J((27;6BB4HH QC!H4!OPP Ps;B B:B:NFrawboolc||d|} tj|d||}|r ||dfn|S#t$r }|}Yd}~nd}~wwxYwt |} |jrt j| stdd|d |t|} t|| t| || |||| S) Read a notebook from an os path.rP)r}r as_versioncapture_validation_errorNUnreadable Notebook:  rr}) _read_filenbformatreads Exceptionr9rgrr3existsr r;r_read_notebook rbrcrrr}answernbrne_origrU invalid_files rrzFileManagerMixin._read_notebook;s8&c:: q %)AB '*1Bq ??r 1   FFFFFF  (00& bgnnX.F.F <<<&<<  'w// Wl+++Xw'''"" Z:RX[#   s*A AAAc||d5}tj||tj|ddddS#1swxYwYdSzSave a notebook to an os_path.r<)r@)versionrN)rVrwrite NO_CONVERTrbrcrrres r_save_notebookzFileManagerMixin._save_notebook_s  7 ; ; q N +)A                       s#AA A  byte_contentbytesreturndict[str, str]c|j}tj|}||||dS)a^Compute the hash hexdigest for the provided bytes. The hash algorithm is provided by the `hash_algorithm` attribute. Parameters ---------- byte_content : bytes The bytes to hash Returns ------- A dictionary to be appended to a model {"hash": str, "hash_algorithm": str}. )hashhash_algorithm)rhashlibnewupdate hexdigest)rbr algorithmhs r _get_hashzFileManagerMixin._get_hashisC' K " "  CCCrrcstrformat str | None8tuple[str | bytes, str] | tuple[str | bytes, str, bytes]c|tj|stdd|z||d5}|}dddn #1swxYwY|dkr |r|d|fn|dfS||dkre |r|dd|fn|ddfS#t$r&}|dkrtdd|zd |Yd}~nd}~wwxYw|r%t|d d |fn#t|d d fS aRead a non-notebook file. Parameters ---------- os_path: str The path to be read. format: str If 'text', the contents will be decoded as UTF-8. If 'base64', the raw bytes contents will be encoded as base64. If 'byte', the raw bytes contents will be returned. If not specified, try to decode as UTF-8, and fall back to base64 raw: bool [Optional] If True, will return as third argument the raw bytes content Returns ------- (content, format, byte_content) It returns the content in the given format as well as the raw byte content. rzCannot read non-file %srbNbyterPutf8z%s is not UTF-8 encodedz bad format)reasonasciibase64) rr3rGr rHreaddecode UnicodeErrorrrbrcrr}rebcontentrns rrzFileManagerMixin._read_file|s,w~~g&& FC!:W!DEE E YYw % % vvxxH                V  36NHfh//XvVv-- X__V,,fh??!//    V###1G;+ $####  [ " " ) )' 2 2Hh G GH%%,,W55 s* A**A.1A.0C C0 C++C0c|dvrtdd |dkr|d}n$|d}t|}n)#t$r}tdd|d||d }~wwxYw||d 5}||d d d d S#1swxYwYd S zSave content of a generic file.>rPrrz:Must specify format of file contents as 'text' or 'base64'rPrrzEncoding error saving z: NF)rP)r encoderrrVrrbrccontentrr b64_bytesrnres r _save_filezFileManagerMixin._save_files= + + +L  Q">>&11#NN733 &y11 Q Q QC!H'!H!HQ!H!HIIq P Q u 5 5  GGH                     s*AA A=!A88A=B::B>B>)rhr|NFr}r~N)rrrrFrcrrrr}r~rr)__name__ __module__ __qualname____doc__r rgr ralgorithms_availablerrrHrVr`rsr{rrrrrrrrZrZs[" T$ I N^   ^ ^ ,,,%%%PQV" " " " " HDDDD(=B: : : : : xrrZc>eZdZdZdZ dddZdd Z dddZdZdS)AsyncFileManagerMixinzY Mixin for ContentsAPI classes that interact with the filesystem asynchronously. cDKt|||jd{VdSrp)r0r-rqs rrszAsyncFileManagerMixin._copys7 sDdh777777777777rr|NFr}r~cJK||d|d{V} tttj|||dd{V}|r ||dfn|S#t $r }|}Yd}~nd}~wwxYwt |} |jrtj | stdd|d|t|} t|| d{Vt| |d{V||||| d{V}|S) rrPNrrrrrrr)rrrrrrr9rgrr3rr r;rrrs rrz$AsyncFileManagerMixin._read_notebooksw<<<<<<<< N)-E q B'*1Bq ??r 1   FFFFFF  (00& bgnnX.F.F <<<&<<  'w//  ,777777777 7333333333** Z:RX[+         sAA%% A;/A66A;cK||d5}tttjtj|||d{VddddS#1swxYwYdSr)rVrrrrrrs rrz$AsyncFileManagerMixin._save_notebooks  7 ; ; qN$/-E                          sVv-- X__V,,fh??!//    V###1G;+ $####  C[ " " ) )' 2 2Hh G Gh''..w77B s* A22A69A60C C8C33C8cK|dvrtdd |dkr|d}n$|d}t|}n)#t$r}tdd|d||d }~wwxYw||d 5}t |j|d {Vd d d d S#1swxYwYd Sr)r rrrrVrrrs rrz AsyncFileManagerMixin._save_fileFsS + + +L  Q">>&11#NN733 &y11 Q Q QC!H'!H!HQ!H!HIIq P Q u 5 5 .17H-- - - - - - - - . . . . . . . . . . . . . . . . . .s*AA A?#A::A?CC Crrrrr) rrrrrsrrrrrrrrrs888QV&&&&&P    =B7 7 7 7 7 r.....rrr)Tr<N)(r __future__rr&rrr(rrr contextlibr functoolsrranyio.to_threadr tornado.webr traitletsr r traitlets.configr traitlets.config.configurabler jupyter_server.utilsrrrrrr.r0r9r;rVrXrZrrrrrs1 #"""""  ++++++++%%%%%%$$$$$$!!!!!! ))))))======AAAAAAAAAA ))) CCCC&CCCC$222888 DDDDN####LBBBBB*LBBBJK.K.K.K.K.,K.K.K.K.K.r