^j ddlmZddlZddlmZddlmZddlZddl Z GddejZ d ddZ dddZ dd Zdd Z d dd Zdd Zdd Zy)) annotationsN)Path)Anyc eZdZdfd ZxZS)NumpyJsonEncoderc t|tjr t|St|tjr t |St|tj r|jSt|%|SN) isinstancenpintegerintfloatingfloatndarraytolistsuperdefault)selfobj __class__s a/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/supervision/utils/file.pyrzNumpyJsonEncoder.default s[ c2:: &s8O c2;; ':  c2:: &::< ws##)rrreturnr)__name__ __module__ __qualname__r __classcell__)rs@rrr s $$rrc t|}g}||jdDcgc]}|js|}}i}|D]o}|jDcgc]!}|j j d#}}t t|D chc]} dj|| dc} ||<qt} |D]h} | j j d} | s%|jD]1\}} || vs | | vs|j|| j|3j|S|j|jd|Scc}wcc}wcc} w)a List files in a directory with specified extensions or all files if no extensions are provided. Args: directory: The directory path as a string or Path object. extensions: A list of file extensions to filter. Extensions may be supplied with or without a leading dot (e.g. ``'jpg'`` and ``'.jpg'`` are equivalent). Matching is case-insensitive. Multi-part extensions are supported (e.g. ``'tar.gz'``). Pass ``None`` (default) to list all files; pass an empty list to return no files. Returns: A list of Path objects for the matching files. Examples: ```pycon >>> import supervision as sv >>> from pathlib import Path >>> import tempfile >>> # Keep a reference to the directory object >>> tmp_dir_obj = tempfile.TemporaryDirectory() >>> tmpdir = tmp_dir_obj.name >>> # Create test files >>> (Path(tmpdir) / "test1.txt").touch() >>> (Path(tmpdir) / "test2.md").touch() >>> (Path(tmpdir) / "test3.py").touch() >>> # List all files in the directory >>> files = sv.list_files_with_extensions(directory=tmpdir) >>> len(files) 3 >>> # Leading dot accepted; matching is case-insensitive >>> files = sv.list_files_with_extensions( ... directory=tmpdir, extensions=['.txt', 'md']) >>> len(files) 2 ``` N*.)rglobis_filesuffixeslowerlstriprangelenjoinsetitemsappendaddextend) directory extensionsfiles_with_extensionsp candidates path_indexpathsuffixr#index seen_pathsextnormalized_extensionpath_extensionss rlist_files_with_extensionsr;saXYI(*!*!4DA aD D+-  DAEOv --c2OHO8=c(m8L /4%&)* Jt   !$  )C#&99;#5#5c#: ')3)9)9); )%oz).Bo.U)006NN4( )  ) !  $$Y^^C%89  'EP sEE&EEctt|5}|r0|Dcgc]$}|js|jd&}}n|Dcgc]}|jd}}ddd|Scc}wcc}w#1swYSxYw)a Read a text file and return a list of strings without newline characters. Optionally skip empty lines. Args: file_path: The file path as a string or Path object. skip_empty: If True, skip lines that are empty or contain only whitespace. Default is False. Returns: A list of strings representing the lines in the text file. Examples: ```pycon >>> import tempfile >>> from pathlib import Path >>> from supervision.utils.file import read_txt_file, save_text_file >>> with tempfile.TemporaryDirectory() as tmpdir: ... file_path = Path(tmpdir) / "test.txt" ... save_text_file(["line1", " ", "line3"], file_path) ... print(read_txt_file(file_path)) ... print(read_txt_file(file_path, skip_empty=True)) ['line1', ' ', 'line3'] ['line1', 'line3'] ```  N)openstrstriprstrip) file_path skip_emptyfilelineliness r read_txt_filerG\s|8 c)n 9 37H44::r?write)rFrBrDrEs rsave_text_filerKsH c)nc "$d $D JJtd{ # $$$$s <Actt|5}tj|}ddd|S#1swYSxYw)a Read a json file and return a dict. Args: file_path: The file path as a string or Path object. Returns: A dict of annotations information Examples: ```pycon >>> import tempfile >>> from pathlib import Path >>> from supervision.utils.file import read_json_file, save_json_file >>> data = {"key": "value", "list": [1, 2, 3]} >>> with tempfile.TemporaryDirectory() as tmpdir: ... file_path = Path(tmpdir) / "test.json" ... save_json_file(data, file_path) ... print(read_json_file(file_path)) {'key': 'value', 'list': [1, 2, 3]} ``` N)r>r?jsonloadrBrDdatas rread_json_filerQs90 c)n yy K K5?ctt|d5}tj||t|dddy#1swYyxYw)z Write a dict to a json file. Args: data: dict with unique keys and value as pair. file_path: The file path as a string or Path object. indent: rI)clsindentN)r>r?rMdumpr)rPrBrUfps rsave_json_filerXs@ c)nc "Ab $ 0@AAAs =Actt|5}tj|}ddd|S#1swYSxYw)z Read a yaml file and return a dict. Args: file_path: The file path as a string or Path object. Returns: A dict of content information N)r>r?yaml safe_loadrOs rread_yaml_filer\s: c)n $~~d#$ K$ KrRctt|d5}tj||dddddy#1swYyxYw)z Save a dict to a yaml file. Args: data: dict with unique keys and value as pair. file_path: The file path as a string or Path object. rIFN) sort_keysdefault_flow_style)r>r?rZrV)rPrBoutfiles rsave_yaml_fileras? c)nc "Kg $5TJKKKs 9Ar )r. str | Pathr/zlist[str] | Nonerz list[Path])F)rBrbrCboolr list[str])rFrdrBrbrNone)rBrbrdict[str, Any]))rPrfrBrbrUr rre)rPrfrBrbrre) __future__rrMpathlibrtypingrnumpyr rZ JSONEncoderrr;rGrKrQrXr\rarrrns"  $t''$;?C!C!'7C!C!L"J $<@A A  A%/ A9< A  A  Kr