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

Change the way to reading movie types and sub types to config file

Browse Source
This commit is contained in:
yoshiko2
2021-01-02 00:31:55 +08:00
parent 7135b29abb
commit 13844b9934
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
AV_Data_Capture.py
View File

@@ -39,19 +39,18 @@ def argparse_function(ver: str) -> [str, str, bool]:
return args.file, args.config, args.number, args.autoexit return args.file, args.config, args.number, args.autoexit
def movie_lists(root, escape_folder): def movie_lists(root, escape_folder, c: config.Config):
for folder in escape_folder: for folder in escape_folder:
if folder in root: if folder in root:
return [] return []
total = [] total = []
file_type = ['.mp4', '.avi', '.rmvb', '.wmv', '.mov', '.mkv', '.flv', '.ts', '.webm', '.MP4', '.AVI', '.RMVB', file_type = c.media_type().upper()
'.WMV', '.MOV', '.MKV', '.FLV', '.TS', '.WEBM', '.iso', '.ISO']
dirs = os.listdir(root) dirs = os.listdir(root)
for entry in dirs: for entry in dirs:
f = os.path.join(root, entry) f = os.path.join(root, entry)
if os.path.isdir(f): if os.path.isdir(f):
total += movie_lists(f, escape_folder) total += movie_lists(f, escape_folder)
elif os.path.splitext(f)[1] in file_type: elif os.path.splitext(f)[1].upper() in file_type:
total.append(f) total.append(f)
return total return total