FileUtils.py 664 B

123456789101112131415161718192021222324
  1. from glob import glob
  2. import os
  3. def list_folders(path: str) -> list:
  4. """Returns the names of all immediate child folders of path.
  5. Args:
  6. path (str): path to search
  7. Returns:
  8. list: list of all child folder names
  9. """
  10. return [name for name in os.listdir(path) if os.path.isdir(os.path.join(path, name))]
  11. def list_jpegs_recursive(path: str) -> list:
  12. """Recursively lists all jpeg files in path.
  13. Args:
  14. path (str): path to search
  15. Returns:
  16. list: list of all jpeg files
  17. """
  18. return [name for name in glob(os.path.join(path, "**/*.jpg"), recursive=True) if os.path.isfile(os.path.join(path, name))]