a yf @sddlZddlZddlZddlZddlZddlmZddlmZddlmZGdddej Z eddZ dd d Z e fd dZe fddZddZdddZdedd fddZdS)N)contextmanager)datetime)Pathc@s(eZdZdZddZddZddZdS) WorkingDirectorya A context manager and decorator for temporarily changing the working directory. This class allows for the temporary change of the working directory using a context manager or decorator. It ensures that the original working directory is restored after the context or decorated function completes. Attributes: dir (Path): The new directory to switch to. cwd (Path): The original current working directory before the switch. Methods: __enter__: Changes the current directory to the specified directory. __exit__: Restores the original working directory on context exit. Examples: Using as a context manager: >>> with WorkingDirectory('/path/to/new/dir'): >>> # Perform operations in the new directory >>> pass Using as a decorator: >>> @WorkingDirectory('/path/to/new/dir') >>> def some_function(): >>> # Perform operations in the new directory >>> pass cCs||_t|_dS)zgSets the working directory to 'new_dir' upon instantiation for use with context managers or decorators.N)dirrcwdresolve)selfZnew_dirr S/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/utils/files.py__init__)szWorkingDirectory.__init__cCst|jdS)z[Changes the current working directory to the specified directory upon entering the context.N)oschdirr)r r r r __enter__.szWorkingDirectory.__enter__cCst|jdS)zARestores the original working directory when exiting the context.N)r rr)r exc_typeexc_valexc_tbr r r __exit__2szWorkingDirectory.__exit__N)__name__ __module__ __qualname____doc__r rrr r r r r src csdt|vrt|t}t|}t}t||jdd}|rVt ||n$| rz|j j dddt ||zD|rt|n|VW|rtj ||ddq| rt ||n0|rtj ||ddn| rt ||0Wdn1s0Yn|VdS)a Context manager to handle paths with spaces in their names. If a path contains spaces, it replaces them with underscores, copies the file/directory to the new path, executes the context code block, then copies the file/directory back to its original location. Args: path (str | Path): The original path that may contain spaces. Yields: (Path): Temporary path with spaces replaced by underscores if spaces were present, otherwise the original path. Examples: Use the context manager to handle paths with spaces: >>> from ultralytics.utils.files import spaces_in_path >>> with spaces_in_path('/path/with spaces') as new_path: >>> # Your code here  _Tparentsexist_ok) dirs_exist_okN)str isinstancertempfileTemporaryDirectorynamereplaceis_dirshutilcopytreeis_fileparentmkdircopy2)pathstringZtmp_dirZtmp_pathr r r spaces_in_path7s*   0r-FcCst|}|rt|st|r,|d|jfn|df\}}tddD](}||||}tj|sBqlqBt|}|r|jddd|S)a Increments a file or directory path, i.e., runs/exp --> runs/exp{sep}2, runs/exp{sep}3, ... etc. If the path exists and `exist_ok` is not True, the path will be incremented by appending a number and `sep` to the end of the path. If the path is a file, the file extension will be preserved. If the path is a directory, the number will be appended directly to the end of the path. If `mkdir` is set to True, the path will be created as a directory if it does not already exist. Args: path (str | pathlib.Path): Path to increment. exist_ok (bool): If True, the path will not be incremented and returned as-is. sep (str): Separator to use between the path and the incrementation number. mkdir (bool): Create a directory if it does not exist. Returns: (pathlib.Path): Incremented path. Examples: Increment a directory path: >>> from pathlib import Path >>> path = Path("runs/exp") >>> new_path = increment_path(path) >>> print(new_path) runs/exp2 Increment a file path: >>> path = Path("runs/exp/results.txt") >>> new_path = increment_path(path) >>> print(new_path) runs/exp/results2.txt r.i'Tr) rexistsr' with_suffixsuffixranger r+r))r+rsepr)r2npr r r increment_pathks  $ r7cCs"ttt|j}|jS)z>Return days since the last modification of the specified file.)rnow fromtimestamprstatst_mtimedays)r+dtr r r file_agesr>cCs.tt|j}|jd|jd|jS)z8Returns the file modification date in 'YYYY-M-D' format.-)rr9rr:r;yearmonthday)r+tr r r file_datesrDcCsXt|ttfrTd}t|}|r0|j|S|rTtdd|dD|SdS)z:Returns the size of a file or directory in megabytes (MB).icss |]}|r|jVqdS)N)r'r:st_size).0fr r r zfile_size..z**/*g) rrrr'r:rEr$sumglob)r+mbr r r file_sizesrM.cCs,tj|ddd}|r(t|tjjdSdS)zdReturns the path to the most recent 'last.pt' file in the specified directory for resuming training.z /**/last*.ptT) recursive)keyr.)rKmaxr r+getctime)Z search_dirZ last_listr r r get_latest_runsrS)z yolov8n.ptc Csddlm}ddlm}|d}|jddd|D]`}||}td|||}||rl|d|j_||} td |d | | | q2d S) ag Updates and re-saves specified YOLO models in an 'updated_models' subdirectory. Args: model_names (Tuple[str, ...]): Model filenames to update. source_dir (Path): Directory containing models and target subdirectory. update_names (bool): Update model names from a data YAML. Examples: Update specified YOLO models and save them in 'updated_models' subdirectory: >>> from ultralytics.utils.files import update_models >>> model_names = ("yolov8n.pt", "yolov8s.pt") >>> update_models(model_names, source_dir=Path("/models"), update_names=True) r)YOLO)default_class_namesZupdated_modelsTrzLoading model from z coco8.yamlz Re-saving z model to N) Z ultralyticsrTZultralytics.nn.autobackendrUr)printZhalfmodelnamessave) Z model_names source_dirZ update_namesrTrU target_dirZ model_nameZ model_pathrWZ save_pathr r r update_modelss   r\)Fr.F)rN) contextlibrKr r%r rrpathlibrContextDecoratorrr-r7__file__r>rDrMrSr\r r r r s    * 3 1