From 39b6aabd213d18c9ce1348e240acd3de23ccac2b Mon Sep 17 00:00:00 2001 From: yoshiko2 Date: Tue, 4 May 2021 03:06:51 +0800 Subject: [PATCH] Fix Sigle File mode --- AV_Data_Capture.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/AV_Data_Capture.py b/AV_Data_Capture.py index 952dc87..d52fe11 100755 --- a/AV_Data_Capture.py +++ b/AV_Data_Capture.py @@ -113,7 +113,7 @@ def create_data_and_move(file_path: str, c: config.Config, debug): print('[!]', err) -def create_data_and_move_with_custom_number(file_path: str, c: config.Config, custom_number=None): +def create_data_and_move_with_custom_number(file_path: str, c: config.Config, custom_number): try: print("[!]Making Data for [{}], the number is [{}]".format(file_path, custom_number)) core_main(file_path, custom_number, c) @@ -159,7 +159,10 @@ if __name__ == '__main__': if not single_file_path == '': #Single File print('[+]==================== Single File =====================') - create_data_and_move_with_custom_number(single_file_path, conf, custom_number) + if custom_number == '': + create_data_and_move_with_custom_number(single_file_path, conf, get_number(conf.debug(), os.path.basename(single_file_path))) + else: + create_data_and_move_with_custom_number(single_file_path, conf, custom_number) else: if folder_path == '': folder_path = os.path.abspath(".")