.PhcrdZddlZddlZddlZddlZddlZddlZddlmZm Z m Z gdZ dZ dZ dZdZGd d ZejejzejzZeed r eejzZeed r eejzZeZeed r eejzZ ddlZdZn#e$rdZYnwxYwdZdZejdkr8ddl Z ddl m!Z!ddl"m#Z#m$Z$e j%j&j'Z(e!e!e!e#e$e$ge(_)dZ*d%dZ+ndZ*d%dZ+e+Z,de*_GddZ-d&dZ.d'dZ/e/Z0Gdd Z1d!d"d#e2fd$Z3dS)(zVirtually every Python programmer has used Python for wrangling disk contents, and ``fileutils`` collects solutions to some of the most commonly-found gaps in the standard library. N)copy2copystatError)mkdir_p atomic_save AtomicSaver FilePermsiter_find_filescopytreeiic tj|nM#t$r@}|jtjkr%tj|rYd}~dSd}~wwxYwdS)a!Creates a directory and any parent directories that may need to be created along the way, without raising errors for any existing directories. This function mimics the behavior of the ``mkdir -p`` command available in Linux/BSD environments, but also works on Windows. N)osmakedirsOSErrorerrnoEEXISTpathisdir)rexcs Q/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/boltons/fileutils.pyrr7sr D  9 $ $t)<)< $ FFFFF  Fs A!4AAA!ceZdZdZGddZddZedZedZdZ ed d Z ed d Z ed dZ dZ dS)r asThe :class:`FilePerms` type is used to represent standard POSIX filesystem permissions: * Read * Write * Execute Across three classes of user: * Owning (u)ser * Owner's (g)roup * Any (o)ther user This class assists with computing new permissions, as well as working with numeric octal ``777``-style and ``rwx``-style permissions. Currently it only considers the bottom 9 permission bits; it does not support sticky bits or more advanced permission systems. Args: user (str): A string in the 'rwx' format, omitting characters for which owning user's permissions are not provided. group (str): A string in the 'rwx' format, omitting characters for which owning group permissions are not provided. other (str): A string in the 'rwx' format, omitting characters for which owning other/world permissions are not provided. There are many ways to use :class:`FilePerms`: >>> FilePerms(user='rwx', group='xrw', other='wxr') # note character order FilePerms(user='rwx', group='rwx', other='rwx') >>> int(FilePerms('r', 'r', '')) 288 >>> oct(288)[-3:] # XXX Py3k '440' See also the :meth:`FilePerms.from_int` and :meth:`FilePerms.from_path` classmethods for useful alternative ways to construct :class:`FilePerms` objects. cNeZdZdZedZddddZdZd dZd Z d Z dS) FilePerms._FilePermPropertyrwx)rwxc"||_||_dSN) attributeoffset)selfr#r$s r__init__z$FilePerms._FilePermProperty.__init__vs&DN DKKKNc4||St||jSr")getattrr#)r%fp_objtype_s r__get__z#FilePerms._FilePermProperty.__get__zs~ 64>22 2r'ct|j}||krdS tt|jz }n #t $rt d|zwxYw|rt d|d|djfd}dtt||d}t|j| ||dS) Nzexpected string, not %rzgot invalid chars z in permission specification z*, expected empty string or one or more of cj|Sr") _perm_val)cr%s rsort_keyz5FilePerms._FilePermProperty.__set__..sort_keysDN1$55r'T)keyreverse) r)r#setstr _perm_set TypeError ValueError _perm_charsjoinsortedsetattr_update_integer)r%r*valuecur invalid_charsr1 new_values` r__set__z#FilePerms._FilePermProperty.__set__s)&$.11Ce|| C #CJJ$. @  C C C 9E ABBB C M j%2MM555$:J:J"LMMM 6 5 5 5 5s5zz+3T!C!C!CDDI FDNI 6 6 6   3 3 3 3 3s $AA"cd}d}|D]*}d||z}|||jdzzz}+|xj|zc_dS)Nrxwrr)indexr$_integer)r%r*r?moder3symbolbits rr>z+FilePerms._FilePermProperty._update_integers\DC 3 3399V,,,q12 OOt #OOOOr'r") __name__ __module__ __qualname__r: frozensetr7r/r&r,rCr>r'r_FilePermPropertyrqsx Ie$$ !!,,  ! ! ! 3 3 3 3  4 4 4( $ $ $ $ $r'rQr2chd\|_|_|_d|_||_||_||_dS)Nr2r2r2r)_user_group_otherrHusergroupother)r%rWrXrYs rr&zFilePerms.__init__s4/9, DK    r'c|tz}d}g}|r*|||tz|dz}|*|||S)zCreate a :class:`FilePerms` object from an integer. >>> FilePerms.from_int(0o644) # note the leading zero-oh for octal FilePerms(user='rw', group='r', other='r') )r2r rxwrrxrwrrF) FULL_PERMSappend_SINGLE_FULL_PERMr4)clsir3partss rfrom_intzFilePerms.from_intsj Z:  LLQ!223 4 4 4 !GA   sE{r'ctj|}|tj|jS)aoMake a new :class:`FilePerms` object based on the permissions assigned to the file or directory at *path*. Args: path (str): Filesystem path of the target file. Here's an example that holds true on most systems: >>> import tempfile >>> 'r' in FilePerms.from_path(tempfile.gettempdir()).user True )rstatrdS_IMODEst_mode)rarstat_ress r from_pathzFilePerms.from_paths074==||DL)9::;;;r'c|jSr")rHr%s r__int__zFilePerms.__int__s }r'rTrrUrrVrcT|jj}|d|jd|jd|jdS)Nz(user=z, group=z, other=)) __class__rLrWrXrY)r%cns r__repr__zFilePerms.__repr__s5 ^ $22tyyy$***djjj: ;r'NrS)rLrMrN__doc__rQr& classmethodrdrjrmrWrXrYrrrPr'rr r Gs''R($($($($($($($($T  [ <<[<   Wa ( (D5  h * *E6  h * *E6;;;;;r'r O_NOINHERIT O_NOFOLLOWO_BINARYc tj|tjd}|tjz}tj|tj|n#t$rYnwxYwdS)zDoes a best-effort :func:`fcntl.fcntl` call to set a fd to be automatically closed by any future child processes. Implementation from the :mod:`tempfile` module. rN)fcntlF_GETFD FD_CLOEXECF_SETFDr)fdflagss r set_cloexecrsm  2KEM155E U% %E KEM5 1 1 1 1     D  s A AAcdS)z5Dummy set_cloexec for platforms without fcntl supportNrP)r}s rrrs r'c t|fi|S)aA convenient interface to the :class:`AtomicSaver` type. Example: >>> try: ... with atomic_save("file.txt", text_mode=True) as fo: ... _ = fo.write('bye') ... 1/0 # will error ... fo.write('bye') ... except ZeroDivisionError: ... pass # at least our file.txt didn't get overwritten See the :class:`AtomicSaver` documentation for details. )r) dest_pathkwargss rrrs y + +F + ++r'ct|tr|Stjptj}||Sr") isinstancer6sysgetfilesystemencodinggetdefaultencodingdecode)rencodings rpath_to_unicodersG$ (**Fc.D.F.FH ;;x  r'nt) c_wchar_p)DWORDLPVOIDcX tj||dS#t$r!}|jtjkrnYd}~nd}~wwxYwt |}t |}t t|t|dddd}|std|d|dS)Nrzfailed to replace z with )rrenamerrrr _ReplaceFiler)srcdstweress rreplacers  Ic3    F   x5<''  c""c""9S>>9S>>D$00 ECsCCCCCDD Ds A?AFcV|rt||ntj||dSz22!\(*:;;  ! ! !!5   ! WT^T_j A AB2ty$.AA   4444   $$&&&  s2AAA0C %C0cLtj|jr'|js t t jd|j|jr=tj|j rtj |j | dS)aSCalled on context manager entry (the :keyword:`with` statement), the ``setup()`` method creates the temporary file in the same directory as the destination file. ``setup()`` tests for a writable directory with rename permissions early, as the part file may not be written to immediately (not using :func:`os.access` because of the potential issues of effective vs. real privileges). If the caller is not using the :class:`AtomicSaver` as a context manager, this method should be called explicitly before writing. z*Overwrite disabled and file already existsN) rrlexistsrrrrrrrrrrls rsetupzAtomicSaver.setups 7??4> * * .> .elJ"n...   &27??4>#B#B & Idn % % % r'c8||jSr")rrrls r __enter__zAtomicSaver.__enter__s ~r'c|jr]|jtj|j|j|r4|jr+ tj|jn#t$rYnwxYwdS t|j|j |j nA#t$r4|jr+ tj|jn#t$rYnwxYwwxYwdS)N)r)rflushrfsyncfilenorrrr Exceptionrrrrr%exc_typeexc_valexc_tbs r__exit__zAtomicSaver.__exit__s< > # N " " " HT^**,, - - - N " " "  " Idn---- D F  $.$.$(N 4 4 4 4 4   " Idn---- D    sB/B BB!B>>C<C*)C<* C74C<6C77C<N) rLrMrNrsRW_PERMSrr&rrrrrPr'rrrOsr00b#.60r'rc#Kt|tr|g}tjdd|D}|sg}nt|tr|g}tjdd|D}t |tjj }tj |D]\}} } |9t |tjj |z |krA|rV| D]S} | | r<|r| | r/tj|| } | VT| D]S} | | r<|r| | r/tj|| } | VTdS)aReturns a generator that yields file paths under a *directory*, matching *patterns* using `glob`_ syntax (e.g., ``*.txt``). Also supports *ignored* patterns. Args: directory (str): Path that serves as the root of the search. Yielded paths will include this as a prefix. patterns (str or list): A single pattern or list of glob-formatted patterns to find under *directory*. ignored (str or list): A single pattern or list of glob-formatted patterns to ignore. include_dirs (bool): Whether to include directories that match patterns, as well. Defaults to ``False``. max_depth (int): traverse up to this level of subdirectory. I.e., 0 for the specified *directory* only, 1 for *directory* and one level of subdirectory. For example, finding Python files in the current directory: >>> _CUR_DIR = os.path.dirname(os.path.abspath(__file__)) >>> filenames = sorted(iter_find_files(_CUR_DIR, '*.py')) >>> os.path.basename(filenames[-1]) 'urlutils.py' Or, Python files while ignoring emacs lockfiles: >>> filenames = iter_find_files(_CUR_DIR, '*.py', ignored='.#*') .. _glob: https://en.wikipedia.org/wiki/Glob_%28programming%29 |c6g|]}tj|SrPfnmatch translate.0ps r z#iter_find_files..s#"J"J"JA7#4Q#7#7"J"J"Jr'c6g|]}tj|SrPrrs rrz#iter_find_files..s#!H!H!H1'"3A"6"6!H!H!Hr'N) rr6recompiler;lensplitrrsepwalkmatch) directorypatternsignored include_dirs max_depthpats_reign_re start_depthrootdirsfilesbasenamefilenames rr r s@(C  :j"J"J"J"J"JKKLLG  GS ! !) Z!H!H!H!H!HII J JFioobgk2233KWY//dE  c$**RW[*A*A&B&B[&PT]%]%]   #  # #==**#!6<<#9#9! !w||D(;;H"NNN  H}}X&& v||H557<<h77    Fr'c tj|}| |||}nt}t|g}|D]E}||vrtj||}tj||} |rItj|r*tj|} tj| | nBtj |rt|| ||nt|| #t$r+} | | jdYd} ~ d} ~ wt$r0} ||| t#| fYd} ~ ?d} ~ wwxYw t%||n<#t$r/} |||t#| fYd} ~ nd} ~ wwxYw|rt|dS)a3The ``copy_tree`` function is an exact copy of the built-in :func:`shutil.copytree`, with one key difference: it will not raise an exception if part of the tree already exists. It achieves this by using :func:`mkdir_p`. As of Python 3.8, you may pass :func:`shutil.copytree` the `dirs_exist_ok=True` flag to achieve the same effect. Args: src (str): Path of the source directory to copy. dst (str): Destination path. Existing directories accepted. symlinks (bool): If ``True``, copy symlinks rather than their contents. ignore (callable): A callable that takes a path and directory listing, returning the files within the listing to be ignored. For more details, check out :func:`shutil.copytree` and :func:`shutil.copy2`. Nr)rlistdirr5rrr;islinkreadlinksymlinkrr rrextendargsrr_r6r) rrsymlinksignorenames ignored_nameserrorsnamesrcnamedstnamelinktoewhys r copy_treer*s* JsOOE sE**   CLLL F88 = ',,sD))',,sD)) 8 (BGNN733 (W-- 67++++w'' ('8V<<<<gw''' % % % MM!&) $ $ $ $ $ $ $ $ 8 8 8 MM7GSXX6 7 7 7 7 7 7 7 7 8,c ,,, sCS*++++++++, Fmms=B D F & E  F %FF  F G(%GGcveZdZddZdZdZdZdZdd Zdd Z dd Z d Z d Z dZ dZdZdZdZdZdS) DummyFilerNcv||_||_d|_d|_d|_d|_d|_d|_dS)NFr)rrIclosedrisattyrnewlines softspace)r%rrIrs rr&zDummyFile.__init__js>       r'cd|_dS)NT)rrls rrzDummyFile.closets  r'cdS)NrrPrls rrzDummyFile.filenowsrr'c2|jrtddSNI/O operation on a closed filerr9rls rrzDummyFile.flushz ; ?=>> >r'ctr" StopIterationrls rnextzDummyFile.next oor'rc2|jrtddSNr r2r r%sizes rreadzDummyFile.read ; ?=>> >rr'c2|jrtddSrr rs rreadlinezDummyFile.readlinerr'c2|jrtdgSrr rs r readlineszDummyFile.readliness ; ?=>> > r'c2|jrtddSrr rls rseekzDummyFile.seekr r'c2|jrtddS)Nr rr rls rtellzDummyFile.tells ; ?=>> >qr'c2|jrtddSrr rls rtruncatezDummyFile.truncater r'c2|jrtddSrr )r%strings rwritezDummyFile.writer r'c2|jrtddSrr )r%list_of_stringss r writelineszDummyFile.writelinesr r'ctr"r rls r__next__zDummyFile.__next__rr'c2|jrtddSrr rls rrzDummyFile.__enter__r r'cdSr"rPrs rrzDummyFile.__exit__sr')rN)r)rLrMrNr&rrrrrrrrrr r#r&r(rrrPr'rrrgs          r'r)keepr,c &|dkrtd|tj|sdStj|std|dtj|\}}g}t d|dzD];}|r||d||!||d|<|g|z}ttt||ddD]:\}}tj|s%tj ||;tj|drtj |ddS)a4 If *filename.ext* exists, it will be moved to *filename.1.ext*, with all conflicting filenames being moved up by one, dropping any files beyond *keep*. After rotation, *filename* will be available for creation as a new file. Fails if *filename* is not a file or if *keep* is not > 0. rzexpected "keep" to be >=1, not Nz expected z to be a file.r) r9rrexistsisfilesplitextranger_reversedlistziprremove) rr,fn_rootfn_ext kept_namesrbfns orig_name kept_names r rotate_filer=s axxA4AABBB 7>>( # # 7>>( # #>rMsD>  )))))))))) * * *     J;J;J;J;J;J;J;J;^)bj(294 72}&r~%O 72|%r}$O  72z"bk!N           , , , , !!!7d??MMM      --------=)5L& 9"FF4L*###ZZZZZZZZz; ; ; ; |6666r LLLLLLLL^*+! ! ! 3! ! ! ! ! ! sB##B.-B.