summaryrefslogtreecommitdiffstats
path: root/cli/cli.py
diff options
context:
space:
mode:
authorJose M. Guisado <jguisado@soleta.eu>2022-05-20 10:22:42 +0200
committerJose M. Guisado <jguisado@soleta.eu>2022-05-20 10:22:42 +0200
commitd90ab82cec328004e743b2bfbb7c25b32eb299ea (patch)
treefcfbfd01f3517631119b7f41ae8363e1259177bd /cli/cli.py
parentf88bcf78ac9ffefcedfb4e108b703dac40ca5839 (diff)
format: use autopep8
Use autopep8 for coding format, and only for whitespace changes. This change drops use of tabs in favor of spaces. Doesn't use autopep8 --aggresive option. Format command: $ autopep8 --inline --recursive . When using git-blame, use --ignore-rev in order to ignore this reformatting commit.
Diffstat (limited to 'cli/cli.py')
-rw-r--r--cli/cli.py204
1 files changed, 103 insertions, 101 deletions
diff --git a/cli/cli.py b/cli/cli.py
index de05f55..1a71e9a 100644
--- a/cli/cli.py
+++ b/cli/cli.py
@@ -16,107 +16,109 @@ import argparse
import requests
import sys
+
class OgREST():
- def __init__(self, ip, port, api_token):
- self.URL = f'http://{ip}:{port}'
- self.HEADERS = {'Authorization' : api_token}
-
- def get(self, path, payload=None):
- try:
- r = requests.get(f'{self.URL}{path}',
- headers=self.HEADERS,
- json=payload)
- if r.status_code != 200:
- sys.exit(f"Cannot connect to ogServer: "
- f"{r.status_code} HTTP status code")
- except IOError as e:
- sys.exit(f"Cannot connect to ogServer: {e}")
- return r
-
- def post(self, path, payload):
- try:
- r = requests.post(f'{self.URL}{path}',
- headers=self.HEADERS,
- json=payload)
- print(r.text)
- if r.status_code not in {200, 202}:
- sys.exit(f"Cannot connect to ogServer: "
- f"{r.status_code} HTTP status code")
- except IOError as e:
- sys.exit(f"Cannot connect to ogServer: {e}")
- return r
+ def __init__(self, ip, port, api_token):
+ self.URL = f'http://{ip}:{port}'
+ self.HEADERS = {'Authorization': api_token}
+
+ def get(self, path, payload=None):
+ try:
+ r = requests.get(f'{self.URL}{path}',
+ headers=self.HEADERS,
+ json=payload)
+ if r.status_code != 200:
+ sys.exit(f"Cannot connect to ogServer: "
+ f"{r.status_code} HTTP status code")
+ except IOError as e:
+ sys.exit(f"Cannot connect to ogServer: {e}")
+ return r
+
+ def post(self, path, payload):
+ try:
+ r = requests.post(f'{self.URL}{path}',
+ headers=self.HEADERS,
+ json=payload)
+ print(r.text)
+ if r.status_code not in {200, 202}:
+ sys.exit(f"Cannot connect to ogServer: "
+ 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):
- choices = ['clients', 'scopes', 'modes', 'hardware',
- 'client', 'images', 'disks']
- parser = argparse.ArgumentParser(prog='ogcli list')
- parser.add_argument('item', choices=choices)
- parsed_args = parser.parse_args([args[0]])
-
- if parsed_args.item == 'clients':
- OgClient.list_clients(self.rest)
- elif parsed_args.item == 'client':
- OgClient.get_client_properties(self.rest, args[1:])
- elif parsed_args.item == 'hardware':
- OgClient.list_client_hardware(self.rest, args[1:])
- elif parsed_args.item == 'modes':
- OgModes.list_available_modes(self.rest)
- elif parsed_args.item == 'scopes':
- OgScope.list_scopes(self.rest)
- elif parsed_args.item == 'images':
- OgImage.list_images(self.rest)
- elif parsed_args.item == 'disks':
- OgDisk.list_disks(self.rest, args[1:])
-
- def set(self, args):
- choices = ['modes', 'mode']
- parser = argparse.ArgumentParser(prog='ogcli set')
- parser.add_argument('item', choices=choices)
- parsed_args = parser.parse_args([args[0]])
-
- if parsed_args.item in choices:
- OgModes.set_modes(self.rest, args[1:])
-
- def send(self, args):
- choices = ['wol', 'poweroff', 'refresh']
- parser = argparse.ArgumentParser(prog='ogcli send')
- parser.add_argument('send_obj', choices=choices)
- parsed_args = parser.parse_args([args[0]])
-
- if parsed_args.send_obj == 'wol':
- OgWol.send_wol(self.rest, args[1:])
- elif parsed_args.send_obj == 'poweroff':
- OgPoweroff.send_poweroff(self.rest, args[1:])
- elif parsed_args.send_obj == 'refresh':
- OgClient.send_refresh(self.rest, args[1:])
-
- def restore(self, args):
- choices = ['image']
- parser = argparse.ArgumentParser(prog='ogcli restore')
- parser.add_argument('send_obj', choices=choices)
- parsed_args = parser.parse_args([args[0]])
-
- if parsed_args.send_obj == 'image':
- OgImage.restore_image(self.rest, args[1:])
-
- def create(self, args):
- choices = ['image']
- parser = argparse.ArgumentParser(prog='ogcli create')
- parser.add_argument('create_obj', choices=choices)
- parsed_args = parser.parse_args([args[0]])
-
- if parsed_args.create_obj == 'image':
- OgImage.create_image(self.rest, args[1:])
-
- def setup(self, args):
- choices = ['disk']
- parser = argparse.ArgumentParser(prog='ogcli setup')
- parser.add_argument('setup_obj', choices=choices)
- parsed_args = parser.parse_args([args[0]])
-
- if parsed_args.setup_obj == 'disk':
- OgDisk.setup_disk(self.rest, args[1:])
+ def __init__(self, cfg):
+ self.rest = OgREST(cfg['ip'], cfg['port'], cfg['api_token'])
+
+ def list(self, args):
+ choices = ['clients', 'scopes', 'modes', 'hardware',
+ 'client', 'images', 'disks']
+ parser = argparse.ArgumentParser(prog='ogcli list')
+ parser.add_argument('item', choices=choices)
+ parsed_args = parser.parse_args([args[0]])
+
+ if parsed_args.item == 'clients':
+ OgClient.list_clients(self.rest)
+ elif parsed_args.item == 'client':
+ OgClient.get_client_properties(self.rest, args[1:])
+ elif parsed_args.item == 'hardware':
+ OgClient.list_client_hardware(self.rest, args[1:])
+ elif parsed_args.item == 'modes':
+ OgModes.list_available_modes(self.rest)
+ elif parsed_args.item == 'scopes':
+ OgScope.list_scopes(self.rest)
+ elif parsed_args.item == 'images':
+ OgImage.list_images(self.rest)
+ elif parsed_args.item == 'disks':
+ OgDisk.list_disks(self.rest, args[1:])
+
+ def set(self, args):
+ choices = ['modes', 'mode']
+ parser = argparse.ArgumentParser(prog='ogcli set')
+ parser.add_argument('item', choices=choices)
+ parsed_args = parser.parse_args([args[0]])
+
+ if parsed_args.item in choices:
+ OgModes.set_modes(self.rest, args[1:])
+
+ def send(self, args):
+ choices = ['wol', 'poweroff', 'refresh']
+ parser = argparse.ArgumentParser(prog='ogcli send')
+ parser.add_argument('send_obj', choices=choices)
+ parsed_args = parser.parse_args([args[0]])
+
+ if parsed_args.send_obj == 'wol':
+ OgWol.send_wol(self.rest, args[1:])
+ elif parsed_args.send_obj == 'poweroff':
+ OgPoweroff.send_poweroff(self.rest, args[1:])
+ elif parsed_args.send_obj == 'refresh':
+ OgClient.send_refresh(self.rest, args[1:])
+
+ def restore(self, args):
+ choices = ['image']
+ parser = argparse.ArgumentParser(prog='ogcli restore')
+ parser.add_argument('send_obj', choices=choices)
+ parsed_args = parser.parse_args([args[0]])
+
+ if parsed_args.send_obj == 'image':
+ OgImage.restore_image(self.rest, args[1:])
+
+ def create(self, args):
+ choices = ['image']
+ parser = argparse.ArgumentParser(prog='ogcli create')
+ parser.add_argument('create_obj', choices=choices)
+ parsed_args = parser.parse_args([args[0]])
+
+ if parsed_args.create_obj == 'image':
+ OgImage.create_image(self.rest, args[1:])
+
+ def setup(self, args):
+ choices = ['disk']
+ parser = argparse.ArgumentParser(prog='ogcli setup')
+ parser.add_argument('setup_obj', choices=choices)
+ parsed_args = parser.parse_args([args[0]])
+
+ if parsed_args.setup_obj == 'disk':
+ OgDisk.setup_disk(self.rest, args[1:])