mirror of https://git.48k.eu/ogcli/
69 lines
1.6 KiB
Python
69 lines
1.6 KiB
Python
#
|
|
# Copyright (C) 2020 Soleta Networks <info@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, version 3.
|
|
#
|
|
|
|
import argparse
|
|
|
|
class OgModes():
|
|
|
|
@staticmethod
|
|
def list_available_modes(rest):
|
|
r = rest.get('/mode')
|
|
print(r.json())
|
|
|
|
@staticmethod
|
|
def set_modes(rest, args):
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--scope-id',
|
|
nargs=1,
|
|
required=True,
|
|
help='ID of the scope')
|
|
parser.add_argument('--scope-type',
|
|
nargs=1,
|
|
required=True,
|
|
help='Type of the scope')
|
|
parser.add_argument('--mode',
|
|
nargs=1,
|
|
required=True,
|
|
help='Mode for the scope')
|
|
parsed_args = parser.parse_args(args)
|
|
|
|
def scope_lookup(scope_id, scope_type, d):
|
|
if scope_id == d.get('id') and \
|
|
scope_type == d.get('type'):
|
|
return d
|
|
for scope in d['scope']:
|
|
lookup = scope_lookup(scope_id,
|
|
scope_type,
|
|
scope)
|
|
if lookup is not None:
|
|
return lookup
|
|
return None
|
|
|
|
def ips_in_scope(scope):
|
|
if 'ip' in scope:
|
|
return [scope['ip']]
|
|
ips = []
|
|
for child in scope['scope']:
|
|
ips += ips_in_scope(child)
|
|
return ips
|
|
|
|
r = rest.get('/scopes')
|
|
scopes = r.json()
|
|
found_scope = scope_lookup(int(parsed_args.scope_id[0]),
|
|
parsed_args.scope_type[0],
|
|
scopes)
|
|
|
|
if found_scope is None:
|
|
print("Scope not found")
|
|
return None
|
|
|
|
ips = ips_in_scope(found_scope)
|
|
|
|
payload = {'clients': ips, 'mode': parsed_args.mode[0]}
|
|
r = rest.post('/mode', payload=payload)
|