fix list_files_with_extensions

This commit is contained in:
jxxghp 2023-06-11 11:31:36 +08:00
parent 5334a408f2
commit 81abd0ee49

View File

@ -88,6 +88,12 @@ class SystemUtils:
@staticmethod @staticmethod
def list_files_with_extensions(directory: Path, extensions: list) -> List[Path]: def list_files_with_extensions(directory: Path, extensions: list) -> List[Path]:
"""
获取目录下所有指定扩展名的文件
"""
if directory.is_file():
return [directory]
files = [] files = []
pattern = r".*(" + "|".join(extensions) + ")$" pattern = r".*(" + "|".join(extensions) + ")$"