bacow/AV_Data_Capture
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #175 from mtxcab/patch-3

Browse Source 
Speed up by avoiding scan in escape folders
This commit is contained in:
Yoshiko
2020-03-27 02:02:16 +08:00
committed by GitHub
parent e6d5c20afb 3a267ecc87
commit 16cd10e0d7
1 changed files with 12 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
AV_Data_Capture.py
View File

@@ -35,24 +35,18 @@ def argparse_get_file():
else: else:
return args.file return args.file
def movie_lists(escape_folder): def movie_lists(root, escape_folder):
escape_folder = re.split('[,，]', escape_folder)
total = []
file_type = ['.mp4', '.avi', '.rmvb', '.wmv', '.mov', '.mkv', '.flv', '.ts', '.webm', '.MP4', '.AVI', '.RMVB', '.WMV','.MOV', '.MKV', '.FLV', '.TS', '.WEBM', ]
file_root = os.getcwd()
for root, dirs, files in os.walk(file_root):
flag_escape = 0
for folder in escape_folder: for folder in escape_folder:
if folder in root: if folder in root:
flag_escape = 1 return []
break total = []
if flag_escape == 1: file_type = ['.mp4', '.avi', '.rmvb', '.wmv', '.mov', '.mkv', '.flv', '.ts', '.webm', '.MP4', '.AVI', '.RMVB', '.WMV','.MOV', '.MKV', '.FLV', '.TS', '.WEBM', ]
continue for entry in os.scandir(root):
for f in files: f = entry.path
if os.path.splitext(f)[1] in file_type: if entry.is_dir():
path = os.path.join(root, f) total += movie_lists(f, escape_folder)
path = path.replace(file_root, '.') elif os.path.splitext(f)[1] in file_type:
total.append(path) total.append(f)
return total return total
@@ -103,6 +97,7 @@ if __name__ == '__main__':
success_folder = config['common']['success_output_folder'] success_folder = config['common']['success_output_folder']
failed_folder = config['common']['failed_output_folder'] # 失败输出目录 failed_folder = config['common']['failed_output_folder'] # 失败输出目录
escape_folder = config['escape']['folders'] # 多级目录刮削需要排除的目录 escape_folder = config['escape']['folders'] # 多级目录刮削需要排除的目录
escape_folder = re.split('[,，]', escape_folder)
print('[*]================== AV Data Capture ===================') print('[*]================== AV Data Capture ===================')
print('[*] Version ' + version) print('[*] Version ' + version)
print('[*]======================================================') print('[*]======================================================')
@@ -110,7 +105,7 @@ if __name__ == '__main__':
UpdateCheck(version) UpdateCheck(version)
CreatFailedFolder(failed_folder) CreatFailedFolder(failed_folder)
os.chdir(os.getcwd()) os.chdir(os.getcwd())
movie_list = movie_lists(escape_folder) movie_list = movie_lists('.', escape_folder)
#========== 野鸡番号拖动 ========== #========== 野鸡番号拖动 ==========
number_argparse=argparse_get_file() number_argparse=argparse_get_file()