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

Update to 4.1.1, add auto exit argparse function, change os._exit(0) and exit() to sys.exit()

Browse Source
This commit is contained in:
yoshiko2
2020-12-13 02:44:33 +08:00
parent 77b4431bde
commit d79f6ce009
1 changed files with 15 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
AV_Data_Capture.py
View File

@@ -1,8 +1,9 @@
import argparse import argparse
from core import *
import os import os
from number_parser import get_number import sys
from number_parser import get_number
from core import *
def check_update(local_version): def check_update(local_version):
data = json.loads(get_html("https://api.github.com/repos/yoshiko2/AV_Data_Capture/releases/latest")) data = json.loads(get_html("https://api.github.com/repos/yoshiko2/AV_Data_Capture/releases/latest"))
@@ -23,10 +24,11 @@ def argparse_function(ver: str) -> [str, str, bool]:
parser.add_argument("file", default='', nargs='?', help="Single Movie file path.") parser.add_argument("file", default='', nargs='?', help="Single Movie file path.")
parser.add_argument("-c", "--config", default='config.ini', nargs='?', help="The config file Path.") parser.add_argument("-c", "--config", default='config.ini', nargs='?', help="The config file Path.")
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("--version", action="version", version=ver) parser.add_argument("-a", "--auto-exit", dest='autoexit', action="store_true", help="Auto exit after program complete")
parser.add_argument("-v", "--version", action="version", version=ver)
args = parser.parse_args() args = parser.parse_args()
return args.file, args.config, args.number return args.file, args.config, args.number, args.autoexit
def movie_lists(root, escape_folder): def movie_lists(root, escape_folder):
for folder in escape_folder: for folder in escape_folder:
@@ -50,7 +52,7 @@ def create_failed_folder(failed_folder):
os.makedirs(failed_folder + '/') os.makedirs(failed_folder + '/')
except: except:
print("[-]failed!can not be make folder 'failed'\n[-](Please run as Administrator)") print("[-]failed!can not be make folder 'failed'\n[-](Please run as Administrator)")
os._exit(0) sys.exit(0)
def CEF(path): def CEF(path):
@@ -117,10 +119,10 @@ def create_data_and_move_with_custom_number(file_path: str, c: config.Config, cu
if __name__ == '__main__': if __name__ == '__main__':
version = '4.0.3' version = '4.1.1'
# Parse command line args # Parse command line args
single_file_path, config_file, custom_number = argparse_function(version) single_file_path, config_file, custom_number, auto_exit = argparse_function(version)
# Read config.ini # Read config.ini
conf = config.Config(path=config_file) conf = config.Config(path=config_file)
@@ -144,7 +146,7 @@ if __name__ == '__main__':
CEF(conf.failed_folder()) CEF(conf.failed_folder())
print("[+]All finished!!!") 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.")
exit() sys.exit(0)
# ========== Single File ========== # ========== Single File ==========
movie_list = movie_lists(".", re.split("[,，]", conf.escape_folder())) movie_list = movie_lists(".", re.split("[,，]", conf.escape_folder()))
@@ -166,5 +168,8 @@ if __name__ == '__main__':
CEF(conf.failed_folder()) CEF(conf.failed_folder())
print("[+]All finished!!!") print("[+]All finished!!!")
if conf.auto_exit(): if conf.auto_exit():
os._exit(0) sys.exit(0)
input("[+][+]Press enter key exit, you can check the error message before you exit.") if auto_exit:
sys.exit(0)
input("Press enter key exit, you can check the error message before you exit...")
sys.exit(0)