Fi ddlmZddlZddlZddlZddlZddlZddlmZddlmZddl m Z GddejZ eddZ ddd Zefdd Zefdd Zdd ZdddZde dd fddZy)) annotationsN)contextmanager)datetime)Pathc$eZdZdZddZdZdZy)WorkingDirectoryaA 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 c`||_tjj|_y)zJInitialize the WorkingDirectory context manager with the target directory.N)dirrcwdresolve)selfnew_dirs Y/var/www/pod-logistic/pod-ai/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 )r s r __enter__zWorkingDirectory.__enter__/ rcBtj|jy)z@Restore the original working directory when exiting the context.N)rrr )r exc_typeexc_valexc_tbs r__exit__zWorkingDirectory.__exit__3rrN)r str | Path)__name__ __module__ __qualname____doc__rrrrrrrs4( rrc#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)aDContext 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 any spaces replaced by underscores. Examples: >>> with spaces_in_path("/path/with spaces") as new_path: ... # Your code here ... pass  _Tparentsexist_ok) dirs_exist_okN)str isinstancertempfileTemporaryDirectorynamereplaceis_dirshutilcopytreeis_fileparentmkdircopy2)pathstringtmp_dirtmp_paths rspaces_in_pathr:8sE& 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$Fc`t|}|jr}|s{|jr|jd|jfn|df\}}t ddD]-}||||}t jj|r-nt}|r|jdd|S)aIncrement 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/exp-2 Increment a file path: >>> path = Path("runs/exp/results.txt") >>> new_path = increment_path(path) >>> print(new_path) runs/exp/results-2.txt i'Tr%) rexistsr2 with_suffixsuffixrangerr6r4)r6r'sepr4r@nps rincrement_pathrEjs< :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)r6dts rfile_agerMs7 (00d1B1K1KL LB 77Nrctjt|jj}|j d|j d|jS)z7Return the file modification date in 'YYYY-M-D' format.-)rrHrrIrJyearmonthday)r6ts r file_daterTsFtDz099:AffXQqwwiq ((rc t|ttfrnd}t|}|jr|j j |z S|j r$td|jdD|z Sy)z:Return the size of a file or directory in mebibytes (MiB).ic3rK|]/}|js|jj1yw)N)r2rIst_size).0fs r zfile_size..s$RAaiikqvvx''Rs77z**/*g) r*r)rr2rIrWr/sumglob)r6mbs r file_sizer^sj$d $ Dz <<>99;&&+ + [[]R61BRRUWW W r.ctj|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\maxrr6getctime) search_dir last_lists rget_latest_runrgs7 ZL 5FI3<3ybgg.. /D"Dr)z yolo26n.ptcTddlm}ddlm}ddlm}|dz }|j dd|D]y}||z }|jd|||} | j|r|d | j_ ||z } |jd |d | | j| {y ) atUpdate 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 = ("yolo26n.pt", "yolo11s.pt") >>> update_models(model_names, source_dir=Path("/models"), update_names=True) r)YOLO)default_class_names)LOGGERupdated_modelsTr%zLoading model from z coco8.yamlz Re-saving z model to N) ultralyticsriultralytics.nn.autobackendrjultralytics.utilsrkr4infohalfmodelnamessave) model_names source_dir update_namesrirjrk target_dir model_name model_pathrr save_paths r update_modelsr|s!>(..JTD1! *,  )*67Z   3L AEKK +   j JykBC 9r)r6r)FrOF) r6rr'boolrBr)r4r}returnr)r6rr~int)r6rr~r))r6rr~float)r_)rer)r~r))rutuplervrrwr}) __future__r contextlibr\rr0r+rrpathlibrContextDecoratorrr:rE__file__rMrTr^rgr|r!rrrs# %&z22&R..b,^!) "*) E (74PS9kp$r