aboutsummaryrefslogtreecommitdiffstats
path: root/moon_manager/moon_manager/api/db/slave.py
blob: f5ac9189d1b21849860fde8a44b284f2fafcb9f5 (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
# Software Name: MOON

# Version: 5.4

# SPDX-FileCopyrightText: Copyright (c) 2018-2020 Orange and its contributors
# SPDX-License-Identifier: Apache-2.0

# This software is distributed under the 'Apache License 2.0',
# the text of which is available at 'http://www.apache.org/licenses/LICENSE-2.0.txt'
# or see the "LICENSE" file for more details.


from uuid import uuid4
import logging
from moon_utilities.security_functions import enforce
from moon_manager.api.db.managers import Managers
from moon_utilities import exceptions

logger = logging.getLogger("moon.db.api.slave")


class SlaveManager(Managers):

    def __init__(self, connector=None):
        self.driver = connector.driver
        Managers.SlaveManager = self

    @enforce(("read", "write"), "slave")
    def update_slave(self, moon_user_id, slave_id, value):
        if slave_id not in self.driver.get_slaves(slave_id=slave_id):
            raise exceptions.SlaveNameUnknown

        return self.driver.update_slave(slave_id=slave_id, value=value)

    @enforce(("read", "write"), "slave")
    def delete_slave(self, moon_user_id, slave_id):
        if slave_id not in self.driver.get_slaves(slave_id=slave_id):
            raise exceptions.SlaveNameUnknown
        return self.driver.delete_slave(slave_id=slave_id)

    @enforce(("read", "write"), "slave")
    def add_slave(self, moon_user_id, slave_id=None, value=None):
        if not value or 'name' not in value or not value['name'].strip():
            raise exceptions.SlaveNameUnknown

        if slave_id in self.driver.get_slaves(slave_id=slave_id):
            raise exceptions.SlaveExisting
        if not slave_id:
            slave_id = uuid4().hex

        return self.driver.add_slave(slave_id=slave_id, value=value)

    @enforce("read", "slave")
    def get_slaves(self, moon_user_id, slave_id=None):
        return self.driver.get_slaves(slave_id=slave_id)