diff options
| -rw-r--r-- | cli/cli.py | 3 | ||||
| -rwxr-xr-x | config.py | 31 | ||||
| -rwxr-xr-x | ogcli | 18 |
3 files changed, 34 insertions, 18 deletions
@@ -20,6 +20,7 @@ from cli.objects.center import OgCenter from cli.objects.room import OgRoom from cli.objects.folder import OgFolder from cli.objects.session import OgSession +from config import cfg import argparse import requests import sys @@ -77,7 +78,7 @@ class OgREST(): class OgCLI(): - def __init__(self, cfg): + def __init__(self): self.rest = OgREST(cfg['ip'], cfg['port'], cfg['api_token']) def list(self, args): diff --git a/config.py b/config.py new file mode 100755 index 0000000..5a79698 --- /dev/null +++ b/config.py @@ -0,0 +1,31 @@ +#!/usr/bin/env python3 + +# Copyright (C) 2020-2024 Soleta Networks <opengnsys@soleta.eu> +# +# This program is free software: you can redistribute it and/or modify it under +# the terms of the GNU Affero General Public License as published by the +# Free Software Foundation; either version 3 of the License, or +# (at your option) any later version. + +import json + +OG_CLI_CFG_PATH = "/opt/opengnsys/etc/ogcli.json" + +def load_config(): + data = None + try: + with open(OG_CLI_CFG_PATH, 'r') as json_file: + data = json.load(json_file) + except json.JSONDecodeError: + sys.exit(f'ERROR: Failed parse malformed JSON file {OG_CLI_CFG_PATH}') + except Exception as e: + sys.exit(f'ERROR: cannot open {OG_CLI_CFG_PATH}: {e}') + + required_cfg_params = {'api_token', 'ip', 'port'} + difference_cfg_params = required_cfg_params - data.keys() + if len(difference_cfg_params) > 0: + sys.exit(f'Missing {difference_cfg_params} key in ' + f'json config file') + return data + +cfg = load_config() @@ -15,8 +15,6 @@ import argparse import json import sys -OG_CLI_CFG_PATH = "/opt/opengnsys/etc/ogcli.json" - def sigint_handler(signum, frame): print("User has pressed ctrl-C, interrupting...") sys.exit(1) @@ -25,22 +23,8 @@ class CLI(): def __init__(self): signal.signal(signal.SIGPIPE, signal.SIG_DFL) signal.signal(signal.SIGINT, sigint_handler) - try: - with open(OG_CLI_CFG_PATH, 'r') as json_file: - self.cfg = json.load(json_file) - except json.JSONDecodeError: - sys.exit(f'ERROR: Failed parse malformed JSON file ' - f'{OG_CLI_CFG_PATH}') - except: - sys.exit(f'ERROR: cannot open {OG_CLI_CFG_PATH}') - - required_cfg_params = {'api_token', 'ip', 'port'} - difference_cfg_params = required_cfg_params - self.cfg.keys() - if len(difference_cfg_params) > 0: - sys.exit(f'Missing {difference_cfg_params} key in ' - f'json config file') - self.ogcli = OgCLI(self.cfg) + self.ogcli = OgCLI() parser = argparse.ArgumentParser(prog='ogcli') parser.add_argument('command', help='Subcommand to run', nargs='?', |