Speed up by skipping scan in escape folders

Try again using older Python APIs. Let me know the detailed error if this doesn't work.
This commit is contained in:
mtxcab
2020-03-27 17:20:55 -07:00
committed by GitHub
parent ccee4ebb6b
commit ea95bed85a

View File

@@ -35,24 +35,19 @@ 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) for folder in escape_folder:
if folder in root:
return []
total = [] total = []
file_type = ['.mp4', '.avi', '.rmvb', '.wmv', '.mov', '.mkv', '.flv', '.ts', '.webm', '.MP4', '.AVI', '.RMVB', '.WMV','.MOV', '.MKV', '.FLV', '.TS', '.WEBM', ] file_type = ['.mp4', '.avi', '.rmvb', '.wmv', '.mov', '.mkv', '.flv', '.ts', '.webm', '.MP4', '.AVI', '.RMVB', '.WMV','.MOV', '.MKV', '.FLV', '.TS', '.WEBM', ]
file_root = os.getcwd() dirs = os.listdir(root)
for root, dirs, files in os.walk(file_root): for entry in dirs:
flag_escape = 0 f = os.path.join(root, entry)
for folder in escape_folder: if os.path.isdir(f):
if folder in root: total += movie_lists(f, escape_folder)
flag_escape = 1 elif os.path.splitext(f)[1] in file_type:
break total.append(f)
if flag_escape == 1:
continue
for f in files:
if os.path.splitext(f)[1] in file_type:
path = os.path.join(root, f)
path = path.replace(file_root, '.')
total.append(path)
return total return total
@@ -103,6 +98,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 +106,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()
@@ -160,4 +156,4 @@ if __name__ == '__main__':
CEF(success_folder) CEF(success_folder)
CEF(failed_folder) CEF(failed_folder)
print("[+]All finished!!!") print("[+]All finished!!!")
input("[+][+]Press enter key exit, you can check the error messge before you exit.") input("[+][+]Press enter key exit, you can check the error messge before you exit.")