45842d3c89281b4f969e0a0ecbe4560ce96e1661
[ta/config-manager.git] / cmdsshandler.py
1 # Copyright 2019 Nokia
2
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
6 #
7 #     http://www.apache.org/licenses/LICENSE-2.0
8 #
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 # See the License for the specific language governing permissions and
13 # limitations under the License.
14 from __future__ import print_function
15 import logging
16 from urlparse import urlparse
17
18 from cmframework.apis import cmerror
19 from cmframework.apis import cmstate
20 from dss.client import dss_client
21 from dss.api import dss_error
22
23
24 class CMDSSHandler(cmstate.CMState):
25     def __init__(self, **kw):
26         uridata = urlparse(kw['uri'])
27         socket = uridata.path
28         self.client = dss_client.Client(socket)
29
30     def _domain_exists(self, domain):
31         return domain in self.get_domains()
32
33     def _name_exists_in_domain(self, domain, name):
34         return name in self.get_domain(domain)
35
36     def get(self, domain, name):
37         logging.debug('get called for %s %s', domain, name)
38
39         if self._domain_exists(domain):
40             try:
41                 domain_data = self.client.get_domain(domain)
42                 return domain_data.get(name, None)
43             except dss_error.Error as ex:
44                 raise cmerror.CMError(str(ex))
45
46         return None
47
48     def get_domain(self, domain):
49         logging.debug('get_domain called for %s', domain)
50
51         if self._domain_exists(domain):
52             try:
53                 return self.client.get_domain(domain)
54             except dss_error.Error as ex:
55                 raise cmerror.CMError(str(ex))
56
57         return None
58
59     def set(self, domain, name, value):
60         logging.debug('set called for setting %s %s=%s', domain, name, value)
61         try:
62             return self.client.set(domain, name, value)
63         except dss_error.Error as ex:
64             raise cmerror.CMError(str(ex))
65
66     def get_domains(self):
67         logging.debug('get_domains called')
68         try:
69             return self.client.get_domains()
70         except dss_error.Error as ex:
71             raise cmerror.CMError(str(ex))
72
73     def delete(self, domain, name):
74         logging.debug('delete called for %s %s', domain, name)
75
76         if self._domain_exists(domain):
77             if self._name_exists_in_domain(domain, name):
78                 try:
79                     self.client.delete(domain, name)
80                 except dss_error.Error as ex:
81                     raise cmerror.CMError(str(ex))
82
83     def delete_domain(self, domain):
84         logging.debug('delete_domain called for %s', domain)
85
86         if self._domain_exists(domain):
87             try:
88                 self.client.delete_domain(domain)
89             except dss_error.Error as ex:
90                 raise cmerror.CMError(str(ex))