diff options
Diffstat (limited to 'cli/cli.py')
| -rw-r--r-- | cli/cli.py | 9 |
1 files changed, 7 insertions, 2 deletions
@@ -13,6 +13,7 @@ from cli.objects.images import OgImage from cli.objects.disks import OgDisk from cli.objects.poweroff import OgPoweroff from cli.objects.reboot import OgReboot +from cli.objects.repo import OgRepo from cli.objects.server import OgServer import argparse import requests @@ -70,7 +71,7 @@ class OgCLI(): def list(self, args): choices = ['clients', 'scopes', 'modes', 'hardware', 'client', 'images', 'disks', 'server', - 'servers'] + 'servers', 'repos'] parser = argparse.ArgumentParser(prog='ogcli list') parser.add_argument('item', choices=choices) @@ -94,11 +95,13 @@ class OgCLI(): OgImage.list_images(self.rest) elif parsed_args.item == 'disks': OgDisk.list_disks(self.rest, args[1:]) + elif parsed_args.item == 'repos': + OgRepo.list_repos(self.rest) elif parsed_args.item in ['server', 'servers']: OgServer.list_servers(self.rest) def set(self, args): - choices = ['modes', 'mode', 'server'] + choices = ['modes', 'mode', 'server', 'repo'] parser = argparse.ArgumentParser(prog='ogcli set') parser.add_argument('item', choices=choices) @@ -110,6 +113,8 @@ class OgCLI(): parsed_args = parser.parse_args([args[0]]) if parsed_args.item in ['modes', 'mode']: OgModes.set_modes(self.rest, args[1:]) + elif parsed_args.item == 'repo': + OgRepo.set_repo(self.rest, args[1:]) elif parsed_args.item == 'server': OgServer.set_server(self.rest, args[1:]) |