youqu/manage.py

109 lines
3.4 KiB
Python

#!/usr/bin/env python3
# _*_ coding:utf-8 _*_
# SPDX-FileCopyrightText: 2023 UnionTech Software Technology Co., Ltd.
# SPDX-License-Identifier: GPL-2.0-only
# pylint: disable=C0114
# pylint: disable=wrong-import-position
import os
import sys
import traceback
from argparse import ArgumentParser
from setting.globalconfig import GlobalConfig
from setting.globalconfig import SystemPath
os.environ["DISPLAY"] = ":0"
os.environ["PIPENV_VERBOSITY"] = "-1"
os.environ["XAUTHORITY"] = f"{GlobalConfig.HOME}/.Xauthority"
for i in SystemPath:
if i.value in sys.path:
continue
sys.path.append(i.value)
from funnylog import logger
from funnylog.conf import setting as log_setting
log_setting.LOG_FILE_PATH = GlobalConfig.REPORT_PATH
class Manage:
__author__ = "mikigo<huangmingqiang@uniontech.com>"
def __init__(self):
from src.plugins.mng import trim
from src.plugins.mng import help_tip
from src.plugins.mng import SubCmd
trim()
logger(GlobalConfig.LOG_LEVEL)
self.cmd_args = sys.argv[1:]
if not self.cmd_args:
print(help_tip())
sys.exit(1)
parser = ArgumentParser(epilog=self.__author__)
subparsers = parser.add_subparsers(help="子命令")
sub_parser_remote = subparsers.add_parser(SubCmd.remote.value)
sub_parser_run = subparsers.add_parser(SubCmd.run.value)
sub_parser_pms = subparsers.add_parser(SubCmd.pmsctl.value)
sub_parser_csv = subparsers.add_parser(SubCmd.csvctl.value)
sub_parser_git = subparsers.add_parser(SubCmd.git.value)
if self.cmd_args[0] == SubCmd.remote.value:
from src.rtk._cargo import remote_runner
from src.rtk.remote_runner import RemoteRunner
remote_kwargs = remote_runner(parser, sub_parser_remote)
RemoteRunner(**remote_kwargs).remote_run()
elif self.cmd_args[0] == SubCmd.run.value:
from src.rtk._cargo import local_runner
from src.rtk.local_runner import LocalRunner
_local_kwargs, _ = local_runner(parser, sub_parser_run, self.cmd_args)
LocalRunner(**_local_kwargs).local_run()
elif self.cmd_args[0] == SubCmd.pmsctl.value:
from src.pms._cargo import pms_control
pms_control(parser, sub_parser_pms)
elif self.cmd_args[0] == SubCmd.csvctl.value:
from src.rtk._cargo import csv_control
csv_control(parser, sub_parser_csv)
elif self.cmd_args[0] == SubCmd.startapp.value:
start_config_log = f"{SubCmd.startapp.value} 后面直接加工程名称"
try:
if self.cmd_args[1] in ("-h", "--help"):
print(start_config_log)
sys.exit(0)
from src.plugins.mng import start_app
start_app(self.cmd_args[1])
except IndexError:
logger.error(f"参数异常: {start_config_log}")
elif self.cmd_args[0] == SubCmd.git.value:
from src.git._cargo import git_control
git_control(parser, sub_parser_git)
elif self.cmd_args[0] in ["-h", "--help"]:
print(help_tip())
else:
print(f"参数异常 \033[0;31m{self.cmd_args}\033[0m!\n{help_tip()}")
if __name__ == "__main__":
try:
Manage()
except Exception as exc:
traceback.print_exception(*sys.exc_info())