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
|
#!/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 yaml import load, dump
try:
from yaml import CLoader as Loader, CDumper as Dumper
except ImportError:
from yaml import Loader, Dumper
class OpenSteakConfig:
"""OpenSteak config class
Use this object as a dict
"""
def __init__(self,
config_file="/usr/local/opensteak/infra/config/common.yaml",
autosave=False):
""" Function __init__
Load saved opensteak config.
@param PARAM: DESCRIPTION
@return RETURN: DESCRIPTION
@param config_file: the yaml config file to read.
default is '/usr/local/opensteak/infra/config/common.yaml'
@param autosave: save automaticly the config at destroy
default is False
"""
self.config_file = config_file
self.autosave = autosave
with open(self.config_file, 'r') as stream:
self._data = load(stream, Loader=Loader)
def __getitem__(self, index):
"""Get an item of the configuration"""
return self._data[index]
def __setitem__(self, index, value):
"""Set an item of the configuration"""
self._data[index] = value
def list(self):
"""Set an item of the configuration"""
return self._data.keys()
def dump(self):
"""Dump the configuration"""
return dump(self._data, Dumper=Dumper)
def save(self):
"""Save the configuration to the file"""
with open(self.config_file, 'w') as f:
f.write(dump(self._data, Dumper=Dumper))
def __del__(self):
if self.autosave:
self.save()
|