diff options
Diffstat (limited to 'cli/cli.py')
| -rw-r--r-- | cli/cli.py | 9 |
1 files changed, 7 insertions, 2 deletions
@@ -16,6 +16,7 @@ from cli.objects.reboot import OgReboot from cli.objects.repo import OgRepo from cli.objects.server import OgServer from cli.objects.center import OgCenter +from cli.objects.room import OgRoom import argparse import requests import sys @@ -196,7 +197,7 @@ class OgCLI(): OgImage.update_image(self.rest, args[1:]) def delete(self, args): - choices = ['server', 'center'] + choices = ['server', 'center', 'room'] parser = argparse.ArgumentParser(prog='ogcli delete') parser.add_argument('delete_obj', choices=choices) @@ -210,9 +211,11 @@ class OgCLI(): OgServer.delete_server(self.rest, args[1:]) elif parsed_args.delete_obj == 'center': OgCenter.delete_center(self.rest, args[1:]) + elif parsed_args.delete_obj == 'room': + OgRoom.delete_room(self.rest, args[1:]) def add(self, args): - choices = ['server', 'repo', 'center'] + choices = ['server', 'repo', 'center', 'room'] parser = argparse.ArgumentParser(prog='ogcli add') parser.add_argument('add_obj', choices=choices) @@ -228,3 +231,5 @@ class OgCLI(): OgRepo.add_repo(self.rest, args[1:]) elif parsed_args.add_obj == 'center': OgCenter.add_center(self.rest, args[1:]) + elif parsed_args.add_obj == 'room': + OgRoom.add_room(self.rest, args[1:]) |