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:
return args.file
def movie_lists(escape_folder):
escape_folder = re.split('[,]', escape_folder)
def movie_lists(root, escape_folder):
for folder in escape_folder:
if folder in root:
return []
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:
if folder in root:
flag_escape = 1
break
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)
dirs = os.listdir(root)
for entry in dirs:
f = os.path.join(root, entry)
if os.path.isdir(f):
total += movie_lists(f, escape_folder)
elif os.path.splitext(f)[1] in file_type:
total.append(f)
return total
@@ -103,6 +98,7 @@ if __name__ == '__main__':
success_folder = config['common']['success_output_folder']
failed_folder = config['common']['failed_output_folder'] # 失败输出目录
escape_folder = config['escape']['folders'] # 多级目录刮削需要排除的目录
escape_folder = re.split('[,]', escape_folder)
print('[*]================== AV Data Capture ===================')
print('[*] Version ' + version)
print('[*]======================================================')
@@ -110,7 +106,7 @@ if __name__ == '__main__':
UpdateCheck(version)
CreatFailedFolder(failed_folder)
os.chdir(os.getcwd())
movie_list = movie_lists(escape_folder)
movie_list = movie_lists('.', escape_folder)
#========== 野鸡番号拖动 ==========
number_argparse=argparse_get_file()
@@ -160,4 +156,4 @@ if __name__ == '__main__':
CEF(success_folder)
CEF(failed_folder)
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.")