summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorAlejandro Sirgo Rica <asirgo@soleta.eu>2024-11-06 13:19:26 +0100
committerAlejandro Sirgo Rica <asirgo@soleta.eu>2024-11-11 09:14:27 +0100
commita219bc7ff729516aea7ddea8516b50d96e7d9363 (patch)
treecad91b8861fe8b6b31cdfac8e87444ed93e0607a
parent3d0aff3d1d47a053a678ecbec69d585cf1bbb275 (diff)
ogcli: move configuration data into a different file
Move the configuration into its own file so it is easier to access from multiple files.
-rw-r--r--cli/cli.py3
-rwxr-xr-xconfig.py31
-rwxr-xr-xogcli18
3 files changed, 34 insertions, 18 deletions
diff --git a/cli/cli.py b/cli/cli.py
index 2a47ece..11d1521 100644
--- a/cli/cli.py
+++ b/cli/cli.py
@@ -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()
diff --git a/ogcli b/ogcli
index f4b55d5..8ed9903 100755
--- a/ogcli
+++ b/ogcli
@@ -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='?',