mirror of
https://github.com/iperov/DeepFaceLive.git
synced 2024-10-18 01:20:20 -07:00
120 lines
3.4 KiB
Python
120 lines
3.4 KiB
Python
from os import scandir
|
|
from pathlib import Path
|
|
from typing import List
|
|
|
|
# image_extensions = [".jpg", ".jpeg", ".png", ".tif", ".tiff"]
|
|
|
|
# def write_bytes_safe(p, bytes_data):
|
|
# """
|
|
# writes to .tmp first and then rename to target filename
|
|
# """
|
|
# p_tmp = p.parent / (p.name + '.tmp')
|
|
# p_tmp.write_bytes(bytes_data)
|
|
# if p.exists():
|
|
# p.unlink()
|
|
# p_tmp.rename (p)
|
|
|
|
def scantree(path):
|
|
"""Recursively yield DirEntry objects for given directory."""
|
|
for entry in scandir(path):
|
|
if entry.is_dir(follow_symlinks=False):
|
|
yield from scantree(entry.path)
|
|
else:
|
|
yield entry
|
|
|
|
|
|
def get_files_paths(dir_path, extensions=None, subdirs=False) -> List[Path]:
|
|
"""
|
|
returns array of Path() of files
|
|
"""
|
|
dir_path = Path (dir_path)
|
|
|
|
result = []
|
|
if dir_path.exists():
|
|
|
|
if subdirs:
|
|
gen = scantree(str(dir_path))
|
|
else:
|
|
gen = scandir(str(dir_path))
|
|
|
|
for x in list(gen):
|
|
if extensions is None or any([x.name.lower().endswith(ext) for ext in extensions]):
|
|
result.append( Path(x.path) )
|
|
return sorted(result)
|
|
|
|
# def get_image_unique_filestem_paths(dir_path, verbose_print_func=None):
|
|
# result = get_image_paths(dir_path)
|
|
# result_dup = set()
|
|
|
|
# for f in result[:]:
|
|
# f_stem = Path(f).stem
|
|
# if f_stem in result_dup:
|
|
# result.remove(f)
|
|
# if verbose_print_func is not None:
|
|
# verbose_print_func ("Duplicate filenames are not allowed, skipping: %s" % Path(f).name )
|
|
# continue
|
|
# result_dup.add(f_stem)
|
|
|
|
# return sorted(result)
|
|
|
|
# def get_paths(dir_path):
|
|
# dir_path = Path (dir_path)
|
|
|
|
# if dir_path.exists():
|
|
# return [ Path(x) for x in sorted([ x.path for x in list(scandir(str(dir_path))) ]) ]
|
|
# else:
|
|
# return []
|
|
|
|
# def get_file_paths(dir_path):
|
|
# dir_path = Path (dir_path)
|
|
|
|
# if dir_path.exists():
|
|
# return [ Path(x) for x in sorted([ x.path for x in list(scandir(str(dir_path))) if x.is_file() ]) ]
|
|
# else:
|
|
# return []
|
|
|
|
# def get_all_dir_names (dir_path):
|
|
# dir_path = Path (dir_path)
|
|
|
|
# if dir_path.exists():
|
|
# return sorted([ x.name for x in list(scandir(str(dir_path))) if x.is_dir() ])
|
|
# else:
|
|
# return []
|
|
|
|
# def get_all_dir_names_startswith (dir_path, startswith):
|
|
# dir_path = Path (dir_path)
|
|
# startswith = startswith.lower()
|
|
|
|
# result = []
|
|
# if dir_path.exists():
|
|
# for x in list(scandir(str(dir_path))):
|
|
# if x.name.lower().startswith(startswith):
|
|
# result.append ( x.name[len(startswith):] )
|
|
# return sorted(result)
|
|
|
|
# def get_first_file_by_stem (dir_path, stem, exts=None):
|
|
# dir_path = Path (dir_path)
|
|
# stem = stem.lower()
|
|
|
|
# if dir_path.exists():
|
|
# for x in sorted(list(scandir(str(dir_path))), key=lambda x: x.name):
|
|
# if not x.is_file():
|
|
# continue
|
|
# xp = Path(x.path)
|
|
# if xp.stem.lower() == stem and (exts is None or xp.suffix.lower() in exts):
|
|
# return xp
|
|
|
|
# return None
|
|
|
|
# def move_all_files (src_dir_path, dst_dir_path):
|
|
# paths = get_file_paths(src_dir_path)
|
|
# for p in paths:
|
|
# p = Path(p)
|
|
# p.rename ( Path(dst_dir_path) / p.name )
|
|
|
|
# def delete_all_files (dir_path):
|
|
# paths = get_file_paths(dir_path)
|
|
# for p in paths:
|
|
# p = Path(p)
|
|
# p.unlink()
|