feat: thread manager

This commit is contained in:
RedDeadDepresso
2024-07-25 01:03:06 +01:00
parent dfa783a699
commit 758d77fe4b
8 changed files with 84 additions and 82 deletions

View File

@@ -1,66 +1,5 @@
import sys
import traceback
try:
with open('traceback.log', 'w') as f:
pass
from app.modules.request import Request
from app.common.logger import logger
from app.modules.install_chara import InstallChara
from app.modules.remove_chara import RemoveChara
from app.modules.fc_kks import FilterConvertKKS
from app.modules.create_backup import CreateBackup
class Script:
def __init__(self, config, file_manager):
"""Initializes the primary azurlane-auto instance with the passed in
Config instance;
Args:
config (Config): BAAuto Config instance
"""
self.config = config
fileManager = file_manager
self.modules = {
'InstallChara': None,
'RemoveChara': None,
'CreateBackup': None,
'FCKKS': None,
}
if self.config.install_chara['Enable']:
self.modules['InstallChara'] = InstallChara(self.config, fileManager)
if self.config.remove_chara['Enable']:
self.modules['RemoveChara'] = RemoveChara(self.config, fileManager)
if self.config.create_backup['Enable']:
self.modules['CreateBackup'] = CreateBackup(self.config, fileManager)
if self.config.fc_kks["Enable"]:
self.modules['FCKKS'] = FilterConvertKKS(self.config, fileManager)
def run(self):
for task in self.config.tasks:
if self.modules[task]:
logger.info("SCRIPT", f'Start Task: {task}')
try:
self.modules[task].logic_wrapper()
except:
logger.error("SCRIPT", f'Task error: {task}. For more info, check the traceback.log file.')
with open('traceback.log', 'a') as f:
f.write(f'[{task}]\n')
traceback.print_exc(None, f, True)
f.write('\n')
sys.exit(1)
sys.exit(0)
except:
print(f'[ERROR] Script Initialisation Error. For more info, check the traceback.log file.')
with open('traceback.log', 'w') as f:
f.write(f'Script Initialisation Error\n')
traceback.print_exc(None, f, True)
f.write('\n')
sys.exit(1)
if __name__ == "__main__":
config = Config('config.json')
file_manager = FileManager(config)
script = Script(config, file_manager)
script.run()
request = Request()
request.process()