summaryrefslogtreecommitdiffstats
path: root/cli/objects/room.py
blob: da9a872acc06f02e64e6a8e5a6ee6ff036c26d38 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
# 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 sys
import argparse
from cli.utils import check_address

class OgRoom():

    @staticmethod
    def add_room(rest, args):
        parser = argparse.ArgumentParser(prog='ogcli add room')
        parser.add_argument('--name',
                    nargs='?',
                    required=True,
                    help='give a name to the room')
        parser.add_argument('--netmask',
                    nargs='?',
                    required=True,
                    help='provide the netmask for the room')
        parser.add_argument('--center',
                    nargs='?',
                    type=int,
                    required=True,
                    help='provide the id of the center that will contain the room')
        parser.add_argument('--desc',
                    nargs='?',
                    required=False,
                    help='room description')
        parser.add_argument('--gateway',
                    nargs='?',
                    required=True,
                    help='address of the main gateway in the room')
        parser.add_argument('--ntp',
                    nargs='?',
                    required=False,
                    help='address of the ntp server')
        parser.add_argument('--dns',
                    nargs='?',
                    required=False,
                    help='address of the dns server')
        parser.add_argument('--folder',
                    nargs='?',
                    type=int,
                    required=False,
                    help='id of the folder that will contain the room')
        parsed_args = parser.parse_args(args)

        err = False
        if parsed_args.netmask and not check_address(parsed_args.netmask):
            print('invalid netmask address', file=sys.stderr)
            err = True
        if parsed_args.gateway and not check_address(parsed_args.gateway):
            print('invalid gateway address', file=sys.stderr)
            err = True
        if parsed_args.ntp and not check_address(parsed_args.ntp):
            print('invalid ntp address', file=sys.stderr)
            err = True
        if parsed_args.dns and not check_address(parsed_args.dns):
            print('invalid dns address', file=sys.stderr)
            err = True
        if err:
            parser.print_help(file=sys.stderr)
            return 1

        payload = {
            'name': parsed_args.name,
            'netmask': parsed_args.netmask,
            'center': parsed_args.center
        }
        if parsed_args.gateway:
            payload['gateway'] = parsed_args.gateway
        if parsed_args.ntp:
            payload['ntp'] = parsed_args.ntp
        if parsed_args.dns:
            payload['dns'] = parsed_args.dns
        if parsed_args.folder:
            payload['folder_id'] = parsed_args.folder
        if parsed_args.desc:
            payload['location'] = parsed_args.desc
        res = rest.post('/room/add', payload=payload)
        if not res:
            return 1
        return 0

    @staticmethod
    def update_room(rest, args):
        parser = argparse.ArgumentParser(prog='ogcli update room')
        parser.add_argument('--id',
                    type=int,
                    nargs='?',
                    required=True,
                    help='room id in database')
        parser.add_argument('--name',
                    nargs='?',
                    required=True,
                    help='the updated name for the room')
        parser.add_argument('--netmask',
                    nargs='?',
                    required=True,
                    help='the updated netmask for the room')
        parser.add_argument('--gateway',
                    nargs='?',
                    required=True,
                    help='updated address of the main gateway in the room')
        parsed_args = parser.parse_args(args)

        err = False
        if parsed_args.netmask and not check_address(parsed_args.netmask):
            print('invalid netmask address', file=sys.stderr)
            err = True
        if parsed_args.gateway and not check_address(parsed_args.gateway):
            print('invalid gateway address', file=sys.stderr)
            err = True
        if err:
            parser.print_help(file=sys.stderr)
            return 1

        payload = {
            'id': parsed_args.id,
            'name': parsed_args.name,
            'netmask': parsed_args.netmask,
            'gateway': parsed_args.gateway,
        }
        
        res = rest.post('/room/update', payload=payload)
        if not res:
            return 1
        return 0

    @staticmethod
    def delete_room(rest, args):
        parser = argparse.ArgumentParser(prog='ogcli delete room')
        parser.add_argument('--id',
                            type=int,
                            nargs='?',
                            required=True,
                            help='room id in database')
        parsed_args = parser.parse_args(args)
        payload = {'id': parsed_args.id}
        res = rest.post('/room/delete', payload=payload)
        if not res:
            return 1
        return 0