Update 2.9

This commit is contained in:
yoshiko2
2020-03-27 02:30:04 +08:00
parent ffe99bf500
commit bc76c4f88f

View File

@@ -35,18 +35,24 @@ def argparse_get_file():
else: else:
return args.file return args.file
def movie_lists(root, escape_folder): def movie_lists(escape_folder):
for folder in escape_folder: escape_folder = re.split('[,]', 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', ]
for entry in os.scandir(root): file_root = os.getcwd()
f = entry.path for root, dirs, files in os.walk(file_root):
if entry.is_dir(): flag_escape = 0
total += movie_lists(f, escape_folder) for folder in escape_folder:
elif os.path.splitext(f)[1] in file_type: if folder in root:
total.append(f) 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)
return total return total
@@ -90,14 +96,13 @@ def getNumber(filepath,absolute_path = False):
if __name__ == '__main__': if __name__ == '__main__':
version = '2.8.3' version = '2.9'
config_file = 'config.ini' config_file = 'config.ini'
config = ConfigParser() config = ConfigParser()
config.read(config_file, encoding='UTF-8') config.read(config_file, encoding='UTF-8')
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('[*]======================================================')
@@ -105,7 +110,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()