nas/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-02-14 07:55:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #2763 from Cabbagec/dev

Browse Source
This commit is contained in:
jxxghp
2024-09-22 07:13:24 +08:00
committed by GitHub
parent 17aa6c674f f0100e6dbc
commit 8e8a10f04e
1 changed files with 18 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
app/utils/system.py
View File

@@ -5,6 +5,7 @@ import re
import shutil
import subprocess
import sys
from glob import glob
from pathlib import Path
from typing import List, Union, Tuple, Optional
@@ -162,9 +163,14 @@ class SystemUtils:
return -1, str(err)
@staticmethod
def list_files(directory: Path, extensions: list, min_filesize: int = 0) -> List[Path]:
def list_files(directory: Path, extensions: list, min_filesize: int = 0, recursive: bool = True) -> List[Path]:
"""
获取目录下所有指定扩展名的文件（包括子目录）
:param directory: 指定的父目录
:param extensions: 需要包含的扩展名列表，例如 ['mkv', 'mp4']
:param min_filesize: 文件最低大小，单位 MB
:param recursive: 是否递归查找，可选参数，默认 True
:return: 文件 Path 列表
"""
if not min_filesize:
@@ -183,7 +189,8 @@ class SystemUtils:
pattern = r".*(" + "|".join(extensions) + ")$"
# 遍历目录及子目录
for path in directory.rglob('**/*'):
for matched_glob in glob(str(directory / '**'), recursive=recursive, include_hidden=True):
path = Path(matched_glob)
if path.is_file() \
and re.match(pattern, path.name, re.IGNORECASE) \
and path.stat().st_size >= min_filesize * 1024 * 1024:
@@ -192,10 +199,15 @@ class SystemUtils:
return files
@staticmethod
def exits_files(directory: Path, extensions: list, min_filesize: int = 0) -> bool:
def exits_files(directory: Path, extensions: list, min_filesize: int = 0, recursive: bool = True) -> bool:
"""
判断目录下是否存在指定扩展名的文件
:return True存在 False不存在
:param directory: 指定的父目录
:param extensions: 需要包含的扩展名列表，例如 ['mkv', 'mp4']
:param min_filesize: 文件最低大小，单位 MB
:param recursive: 是否递归查找，可选参数，默认 True
:return: True存在 False不存在
"""
if not min_filesize:
@@ -213,7 +225,8 @@ class SystemUtils:
pattern = r".*(" + "|".join(extensions) + ")$"
# 遍历目录及子目录
for path in directory.rglob('**/*'):
for matched_glob in glob(str(directory / '**'), recursive=recursive, include_hidden=True):
path = Path(matched_glob)
if path.is_file() \
and re.match(pattern, path.name, re.IGNORECASE) \
and path.stat().st_size >= min_filesize * 1024 * 1024: