diff options
Diffstat (limited to 'cli/cli.py')
| -rw-r--r-- | cli/cli.py | 9 |
1 files changed, 7 insertions, 2 deletions
@@ -18,6 +18,7 @@ from cli.objects.server import OgServer from cli.objects.live import OgLive from cli.objects.center import OgCenter from cli.objects.room import OgRoom +from cli.objects.folder import OgFolder import argparse import requests import sys @@ -199,7 +200,7 @@ class OgCLI(): OgImage.update_image(self.rest, args[1:]) def delete(self, args): - choices = ['server', 'repo', 'center', 'room', 'client'] + choices = ['server', 'repo', 'center', 'room', 'client', 'folder'] parser = argparse.ArgumentParser(prog='ogcli delete') parser.add_argument('delete_obj', choices=choices) @@ -219,9 +220,11 @@ class OgCLI(): OgRoom.delete_room(self.rest, args[1:]) elif parsed_args.delete_obj == 'client': OgClient.delete_client(self.rest, args[1:]) + elif parsed_args.add_obj == 'folder': + OgFolder.delete_folder(self.rest, args[1:]) def add(self, args): - choices = ['server', 'repo', 'center', 'room', 'client'] + choices = ['server', 'repo', 'center', 'room', 'client', 'folder'] parser = argparse.ArgumentParser(prog='ogcli add') parser.add_argument('add_obj', choices=choices) @@ -241,3 +244,5 @@ class OgCLI(): OgRoom.add_room(self.rest, args[1:]) elif parsed_args.add_obj == 'client': OgClient.add_client(self.rest, args[1:]) + elif parsed_args.add_obj == 'folder': + OgFolder.add_folder(self.rest, args[1:]) |