|h. ^ddlZddlZddlZddlZddlZddlmZddlmZddlmZddl m Z GddejZ ede e effd Zdde e efd ed e d edef dZefde e efdefdZefde e efde fdZde e efdefdZdde de fdZdedd fdededefdZy)N)contextmanager)datetime)Path)Unionc2eZdZdZdeeeffdZdZdZ y)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 | str): 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 new_dirc`||_tjj|_y)zJInitialize the WorkingDirectory context manager with the target directory.N)dirrcwdresolve)selfr s V/var/www/html/test/engine/venv/lib/python3.12/site-packages/ultralytics/utils/files.py__init__zWorkingDirectory.__init__*s88:%%'cBtj|jy)zZChange the current working directory to the specified directory upon entering the context.N)oschdirr )rs r __enter__zWorkingDirectory.__enter__/ rcBtj|jy)z@Restore the original working directory when exiting the context.N)rrr )rexc_typeexc_valexc_tbs r__exit__zWorkingDirectory.__exit__3rrN) __name__ __module__ __qualname____doc__rstrrrrrrrrrs&6(c4i 0( rrpathc#LKdt|vr-t|t}t|}tj5}t||j j ddz }|jrtj||nC|jr3|jjddtj|| |r t|n||jrtj||dn'|jrtj|| dddy|y#|jrtj||dw|jrtj||wwxYw#1swYyxYww)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 | str): Temporary path with spaces replaced by underscores if spaces were present, otherwise the original path. Examples: >>> with spaces_in_path('/path/with spaces') as new_path: >>> # Your code here >>> pass  _Tparentsexist_ok) dirs_exist_okN)r isinstancertempfileTemporaryDirectorynamereplaceis_dirshutilcopytreeis_fileparentmkdircopy2)r"stringtmp_dirtmp_paths rspaces_in_pathr98sE* c$iD#&Dz ( ( * 1gG}tyy'8'8c'BBH{{}h/%%dT%B T8, 1'-c(m8;??$OOHd$G%%'LL40' 1 1. ??$OOHd$G%%'LL40(% 1 1s8>F$BFE%AF5F$AFFF!F$Fr(sepr4returnc`t|}|jr}|s{|jr|jd|jfn|df\}}t ddD]-}||||}t jj|r-nt}|r|jdd|S)a Increment 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. Args: path (str | Path): Path to increment. exist_ok (bool, optional): If True, the path will not be incremented and returned as-is. sep (str, optional): Separator to use between the path and the incrementation number. mkdir (bool, optional): Create a directory if it does not exist. Returns: (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 i'Tr&) rexistsr2 with_suffixsuffixrangerr"r4)r"r(r:r4rAnps rincrement_pathrEls> :D {{}X>Blln((,dkk:SWY[R\ fq$ A&aS)A77>>!$ Aw  4$ / Krctjtjt|j j z }|j S)z>Return days since the last modification of the specified file.)rnow fromtimestamprstatst_mtimedays)r"dts rfile_agerMs7 (00d1B1K1KL LB 77Nrctjt|jj}|j d|j d|jS)z7Return the file modification date in 'YYYY-M-D' format.-)rrHrrIrJyearmonthday)r"ts r file_daterTsFtDz099:AffXQqwwiq ((rc t|ttfrnd}t|}|jr|j j |z S|j r$td|jdD|z Sy)z9Return the size of a file or directory in megabytes (MB).ic3rK|]/}|js|jj1yw)N)r2rIst_size).0fs r zfile_size..s$RAaiikqvvx''Rs77z**/*g) r*r rr2rIrWr/sumglob)r"mbs r file_sizer^sj$d $ Dz <<>99;&&+ + [[]R61BRRUWW W r. search_dirctj|dd}|r%t|tjjSdS)zcReturn the path to the most recent 'last.pt' file in the specified directory for resuming training.z /**/last*.ptT) recursive)keyr=)r\maxrr"getctime)r` last_lists rget_latest_runrgs7 ZL 5FI3<3ybgg.. /D"Dr)z yolo11n.pt model_names source_dir update_namesc0ddlm}ddlm}|dz }|j dd|D]m}||z }t d|||}|j |r|d|j_||z } t d |d | |j| oy ) ay Update and re-save specified YOLO models in an 'updated_models' subdirectory. Args: model_names (tuple, optional): Model filenames to update. source_dir (Path, optional): Directory containing models and target subdirectory. update_names (bool, optional): 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 = ("yolo11n.pt", "yolov8s.pt") >>> update_models(model_names, source_dir=Path("/models"), update_names=True) r)YOLO)default_class_namesupdated_modelsTr&zLoading model from z coco8.yamlz Re-saving z model to N) ultralyticsrlultralytics.nn.autobackendrmr4printhalfmodelnamessave) rhrirjrlrm target_dir model_name model_pathrs save_paths r update_modelsrzs!>..JTD1! *,  #J<01Z   3L AEKK +   :,j <= 9r)Fr=F)r_) contextlibr\rr0r+rrpathlibrtypingrContextDecoratorrr r9boolrE__file__intrMrTfloatr^rgtuplerzr!rrrs, %'z22'T0sDy)00f-sDy)-T--Y]-jn-`'/5d#3 (0)E#t)$)C) E#t)$  EsESE (74PS9kp$u$D$dh$r