summaryrefslogtreecommitdiffstats
path: root/opensteak/tools/opensteak/foreman_objects/operatingsystems.py
blob: 8cce606e62e46d70f3f9e85fe4eb437d50485a50 (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
#!/usr/bin/python3
# -*- coding: utf-8 -*-
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
#
# Authors:
# @author: David Blaisonneau <david.blaisonneau@orange.com>
# @author: Arnaud Morin <arnaud1.morin@orange.com>

from opensteak.foreman_objects.objects import ForemanObjects
from opensteak.foreman_objects.item import ForemanItem


class OperatingSystems(ForemanObjects):
    """
    OperatingSystems class
    """
    objName = 'operatingsystems'
    payloadObj = 'operatingsystem'

    def __getitem__(self, key):
        """ Function __getitem__

        @param key: The operating system id/name
        @return RETURN: The item
        """
        ret = self.api.list(self.objName,
                            filter='title = "{}"'.format(key))
        if len(ret):
            return ForemanItem(self.api, key,
                               self.objName, self.payloadObj,
                               ret[0])
        else:
            return None

    def __setitem__(self, key, attributes):
        """ Function __getitem__

        @param key: The operating system id/name
        @param attributes: The content of the operating system to create
        @return RETURN: The API result
        """
        if key not in self:
            payload = {self.payloadObj: {'title': key}}
            payload[self.payloadObj].update(attributes)
            return self.api.create(self.objName, payload)
        return False

    def listName(self):
        """ Function listName
        Get the list of all objects name with Ids

        @param key: The targeted object
        @return RETURN: A dict of obejct name:id
        """
        return { x['title']: x['id'] for x in self.api.list(self.objName,
                                                            limit=999999)}