mirror of https://git.48k.eu/ogcli/
131 lines
4.6 KiB
Python
131 lines
4.6 KiB
Python
# Copyright (C) 2020-2024 Soleta Networks <opengnsys@soleta.eu>
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify it under
|
|
# the terms of the GNU Affero General Public License as published by the
|
|
# Free Software Foundation; either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
|
|
import argparse
|
|
import requests
|
|
|
|
from cli.utils import print_json, check_address, check_mac_address
|
|
|
|
class OgClient():
|
|
|
|
@staticmethod
|
|
def list_clients(rest):
|
|
r = rest.get('/clients')
|
|
print_json(r.text)
|
|
|
|
@staticmethod
|
|
def list_client_hardware(rest, args):
|
|
parser = argparse.ArgumentParser(prog='ogcli list hardware')
|
|
parser.add_argument('--client-ip',
|
|
nargs=1,
|
|
type=str,
|
|
required=True,
|
|
help='client IP')
|
|
parsed_args = parser.parse_args(args)
|
|
|
|
payload = {'client': parsed_args.client_ip}
|
|
r = rest.get('/hardware', payload=payload)
|
|
print_json(r.text)
|
|
|
|
@staticmethod
|
|
def get_client_properties(rest, args):
|
|
parser = argparse.ArgumentParser(prog='ogcli list client')
|
|
parser.add_argument('--client-ip',
|
|
nargs=1,
|
|
required=True,
|
|
help='client IP')
|
|
parsed_args = parser.parse_args(args)
|
|
|
|
payload = {'client': parsed_args.client_ip}
|
|
r = rest.get('/client/info', payload=payload)
|
|
print_json(r.text)
|
|
|
|
@staticmethod
|
|
def request_refresh(rest, args):
|
|
parser = argparse.ArgumentParser(prog='ogcli request refresh')
|
|
parser.add_argument('--client-ip',
|
|
action='append',
|
|
default=[],
|
|
required=True,
|
|
help='Client IP')
|
|
parsed_args = parser.parse_args(args)
|
|
|
|
payload = {'clients': parsed_args.client_ip}
|
|
rest.post('/refresh', payload=payload)
|
|
|
|
@staticmethod
|
|
def add_client(rest, args):
|
|
parser = argparse.ArgumentParser(prog='ogcli add client')
|
|
parser.add_argument('--repo-id',
|
|
type=int,
|
|
nargs='?',
|
|
required=True,
|
|
help='specify the id of the repo')
|
|
parser.add_argument('--room-id',
|
|
nargs='?',
|
|
type=int,
|
|
required=True,
|
|
help='specify the id of the room where the client is')
|
|
parser.add_argument('--hostname',
|
|
nargs='?',
|
|
required=True,
|
|
help='give a name to the client')
|
|
parser.add_argument('--mac',
|
|
nargs='?',
|
|
required=True,
|
|
help='mac address should have this format xx:xx:xx:xx:xx:xx')
|
|
parser.add_argument('--ip',
|
|
nargs='?',
|
|
required=True,
|
|
help='specify the ip address of the client')
|
|
r = rest.get('/mode')
|
|
boot_choices = r.json()['modes']
|
|
parser.add_argument('--boot-mode',
|
|
nargs='?',
|
|
required=True,
|
|
choices=boot_choices,
|
|
help='set a valid boot mode for the client')
|
|
parsed_args = parser.parse_args(args)
|
|
|
|
err = False
|
|
if not check_address(parsed_args.ip):
|
|
print('client\'s ip address is not valid', file=sys.stderr)
|
|
err = True
|
|
if not check_mac_address(parsed_args.mac):
|
|
print('client\'s mac address is not valid', file=sys.stderr)
|
|
err = True
|
|
if err:
|
|
parser.print_help(file=sys.stderr)
|
|
sys.exit(1)
|
|
payload = {
|
|
'name': parsed_args.hostname,
|
|
'netmask': '0',
|
|
'ip': parsed_args.ip,
|
|
'mac': parsed_args.mac.replace(":", ""),
|
|
'repo_id': parsed_args.repo_id,
|
|
'room': parsed_args.room_id,
|
|
'boot': parsed_args.boot_mode,
|
|
'netiface': 'eth0',
|
|
'netdriver': 'generic',
|
|
'livedir': 'ogLive'
|
|
}
|
|
rest.post('/client/add', payload=payload)
|
|
|
|
@staticmethod
|
|
def delete_client(rest, args):
|
|
parser = argparse.ArgumentParser(prog='ogcli delete client')
|
|
parser.add_argument('--ip',
|
|
action='append',
|
|
default=[],
|
|
required=True,
|
|
help='Client IP')
|
|
parsed_args = parser.parse_args(args)
|
|
payload = {
|
|
'clients': parsed_args.ip,
|
|
}
|
|
rest.post('/client/delete', payload=payload)
|