summaryrefslogtreecommitdiffstats
path: root/cli/cli.py
diff options
context:
space:
mode:
Diffstat (limited to 'cli/cli.py')
-rw-r--r--cli/cli.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/cli/cli.py b/cli/cli.py
index 1a71e9a..7d3142d 100644
--- a/cli/cli.py
+++ b/cli/cli.py
@@ -12,6 +12,7 @@ from cli.objects.wol import OgWol
from cli.objects.images import OgImage
from cli.objects.disks import OgDisk
from cli.objects.poweroff import OgPoweroff
+from cli.objects.reboot import OgReboot
import argparse
import requests
import sys
@@ -84,7 +85,7 @@ class OgCLI():
OgModes.set_modes(self.rest, args[1:])
def send(self, args):
- choices = ['wol', 'poweroff', 'refresh']
+ choices = ['reboot', 'refresh', 'poweroff', 'wol']
parser = argparse.ArgumentParser(prog='ogcli send')
parser.add_argument('send_obj', choices=choices)
parsed_args = parser.parse_args([args[0]])
@@ -95,6 +96,8 @@ class OgCLI():
OgPoweroff.send_poweroff(self.rest, args[1:])
elif parsed_args.send_obj == 'refresh':
OgClient.send_refresh(self.rest, args[1:])
+ elif parsed_args.send_obj == 'reboot':
+ OgReboot.send_reboot(self.rest, args[1:])
def restore(self, args):
choices = ['image']