diff options
Diffstat (limited to 'ogcli')
-rw-r--r-- | ogcli/objects/__init__.py | 0 | ||||
-rw-r--r-- | ogcli/objects/og_client.py | 6 | ||||
-rw-r--r-- | ogcli/ogcli.py | 32 |
3 files changed, 29 insertions, 9 deletions
diff --git a/ogcli/objects/__init__.py b/ogcli/objects/__init__.py new file mode 100644 index 0000000..e69de29 --- /dev/null +++ b/ogcli/objects/__init__.py diff --git a/ogcli/objects/og_client.py b/ogcli/objects/og_client.py new file mode 100644 index 0000000..ca91036 --- /dev/null +++ b/ogcli/objects/og_client.py @@ -0,0 +1,6 @@ +class OgClient(): + + @staticmethod + def list_clients(rest): + r = rest.get('/clients') + print(r.json()) diff --git a/ogcli/ogcli.py b/ogcli/ogcli.py index d383e2d..37fb2e0 100644 --- a/ogcli/ogcli.py +++ b/ogcli/ogcli.py @@ -1,18 +1,32 @@ +from ogcli.objects.og_client import OgClient +import argparse import requests +import sys -class OgCLI(): - def __init__(self, cfg): - self.api_token = cfg['api_token'] +class OgREST(): + def __init__(self, ip, port, api_token): + self.URL = f'http://{ip}:{port}' + self.HEADERS = {'Authorization' : api_token} - def client_list(self): - headers = {'Authorization' : self.api_token} + def get(self, path): try: - r = requests.get('http://127.0.0.1:8888/clients', - headers=headers) + r = requests.get(f'{self.URL}/clients', + headers=self.HEADERS) if r.status_code != 200: sys.exit(f"Cannot connect to ogServer: " - f"{r.status_code} HTTP status code") + f"{r.status_code} HTTP status code") except IOError as e: sys.exit(f"Cannot connect to ogServer: {e}") + return r + +class OgCLI(): + def __init__(self, cfg): + self.rest = OgREST(cfg['ip'], cfg['port'], cfg['api_token']) + + def list(self, args): + parser = argparse.ArgumentParser() + parser.add_argument('item', choices=['clients']) + args = parser.parse_args(args) - print(r.json()) + if args.item == 'clients': + OgClient.list_clients(self.rest) |