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

add param -q regex query filepath

Browse Source
This commit is contained in:
lededev
2021-09-26 19:20:20 +08:00
parent 948a3d20b0
commit 620affc987
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
AV_Data_Capture.py
View File

@@ -48,10 +48,11 @@ Use --log-dir= to turn off logging feature.""")
parser.add_argument("-n", "--number", default='', nargs='?', help="Custom file number") parser.add_argument("-n", "--number", default='', nargs='?', help="Custom file number")
parser.add_argument("-a", "--auto-exit", dest='autoexit', action="store_true", parser.add_argument("-a", "--auto-exit", dest='autoexit', action="store_true",
help="Auto exit after program complete") help="Auto exit after program complete")
parser.add_argument("-q","--regex-query",dest='regexstr',default='',nargs='?',help="python re module regex filepath filtering.")
parser.add_argument("-v", "--version", action="version", version=ver) parser.add_argument("-v", "--version", action="version", version=ver)
args = parser.parse_args() args = parser.parse_args()
return args.file, args.path, args.number, args.autoexit, args.logdir return args.file, args.path, args.number, args.autoexit, args.logdir, args.regexstr
class OutLogger(object): class OutLogger(object):
@@ -127,7 +128,7 @@ def close_logfile(logdir: str):
# 重写视频文件扫描，消除递归，取消全局变量，新增失败文件列表跳过处理 # 重写视频文件扫描，消除递归，取消全局变量，新增失败文件列表跳过处理
def movie_lists(root, conf): def movie_lists(root, conf, regexstr):
escape_folder = re.split("[,，]", conf.escape_folder()) escape_folder = re.split("[,，]", conf.escape_folder())
main_mode = conf.main_mode() main_mode = conf.main_mode()
debug = conf.debug() debug = conf.debug()
@@ -136,6 +137,12 @@ def movie_lists(root, conf):
total = [] total = []
file_type = conf.media_type().upper().split(",") file_type = conf.media_type().upper().split(",")
trailerRE = re.compile(r'-trailer\.', re.IGNORECASE) trailerRE = re.compile(r'-trailer\.', re.IGNORECASE)
cliRE = None
if len(regexstr):
try:
cliRE = re.compile(regexstr, re.IGNORECASE)
except:
pass
failed_list = [] failed_list = []
if main_mode == 3 or soft_link: if main_mode == 3 or soft_link:
try: try:
@@ -155,6 +162,8 @@ def movie_lists(root, conf):
if debug: if debug:
print('[!]Skip failed file:', absf) print('[!]Skip failed file:', absf)
continue continue
if cliRE and not cliRE.search(absf):
continue
if main_mode == 3 and nfo_skip_days > 0: if main_mode == 3 and nfo_skip_days > 0:
nfo = Path(absf).with_suffix('.nfo') nfo = Path(absf).with_suffix('.nfo')
if file_modification_days(nfo) <= nfo_skip_days: if file_modification_days(nfo) <= nfo_skip_days:
@@ -275,7 +284,7 @@ if __name__ == '__main__':
version = '4.7.2' version = '4.7.2'
urllib3.disable_warnings() #Ignore http proxy warning urllib3.disable_warnings() #Ignore http proxy warning
# Parse command line args # Parse command line args
single_file_path, folder_path, custom_number, auto_exit, logdir = argparse_function(version) single_file_path, folder_path, custom_number, auto_exit, logdir, regexstr = argparse_function(version)
dupe_stdout_to_logfile(logdir) dupe_stdout_to_logfile(logdir)
@@ -309,7 +318,7 @@ if __name__ == '__main__':
if folder_path == '': if folder_path == '':
folder_path = os.path.abspath(".") folder_path = os.path.abspath(".")
movie_list = movie_lists(folder_path, conf) movie_list = movie_lists(folder_path, conf, regexstr)
count = 0 count = 0
count_all = str(len(movie_list)) count_all = str(len(movie_list))