From 8be21d4ac51943c326d368a70c778fb96fa60b94 Mon Sep 17 00:00:00 2001 From: Javier Hernandez Date: Tue, 17 Oct 2023 13:29:25 +0200 Subject: cli: add center.py command to add center, specifying name and optionally a description ogcli add center --name [--comment ] For example, to add a center with name "foocenter" and comment "foocenter description" type $ ogcli add center --name foocenter --comment "foocenter description" --- cli/cli.py | 5 ++++- cli/objects/center.py | 20 ++++++++++++++++++++ 2 files changed, 24 insertions(+), 1 deletion(-) create mode 100644 cli/objects/center.py diff --git a/cli/cli.py b/cli/cli.py index cfe99f6..9eab3f5 100644 --- a/cli/cli.py +++ b/cli/cli.py @@ -15,6 +15,7 @@ from cli.objects.poweroff import OgPoweroff from cli.objects.reboot import OgReboot from cli.objects.repo import OgRepo from cli.objects.server import OgServer +from cli.objects.center import OgCenter import argparse import requests import sys @@ -209,7 +210,7 @@ class OgCLI(): OgServer.delete_server(self.rest, args[1:]) def add(self, args): - choices = ['server', 'repo'] + choices = ['server', 'repo', 'center'] parser = argparse.ArgumentParser(prog='ogcli add') parser.add_argument('add_obj', choices=choices) @@ -223,3 +224,5 @@ class OgCLI(): OgServer.add_server(self.rest, args[1:]) elif parsed_args.add_obj == 'repo': OgRepo.add_repo(self.rest, args[1:]) + elif parsed_args.add_obj == 'center': + OgCenter.add_center(self.rest, args[1:]) diff --git a/cli/objects/center.py b/cli/objects/center.py new file mode 100644 index 0000000..9275703 --- /dev/null +++ b/cli/objects/center.py @@ -0,0 +1,20 @@ +import argparse + +class OgCenter(): + @staticmethod + def add_center(rest, args): + parser = argparse.ArgumentParser(prog='ogcli add center') + parser.add_argument('--name', + nargs='?', + required=True, + help='Name of the center') + parser.add_argument('--comment', + nargs='?', + required=False, + help='(Optional) Provide a more detailed description of the center') + parsed_args = parser.parse_args(args) + payload = {'name': parsed_args.name} + if parsed_args.comment: + payload['comment'] = parsed_args.comment + rest.post('/center/add', payload=payload) + -- cgit v1.2.3-18-g5258