aboutsummaryrefslogtreecommitdiffstats
path: root/storm/config.py
blob: 469f0f5087fbf702a7074d0108543ba5b68653dd (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
# Copyright (C) 2007 Free Software Foundation, Inc.
# This file is part of Ajenti.
# Written by Eugeny Pankov, 2010-2011.
#
# Ajenti is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; only
# version 3 of the License.
#
# Ajenti is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with Ajenti; if not, see <http://www.gnu.org/licenses/>

from ConfigParser import ConfigParser
import os

class Config(ConfigParser):
    internal = {}
    filename = ''

    def __init__(self):
        ConfigParser.__init__(self)

    def load(self, fn):
        self.filename = fn
        self.read(fn)

    def save(self):
        with open(self.filename, 'w') as f:
            self.write(f)

    def get(self, section, val=None, default=None):
        if val is None:
            return self.internal[section]
        else:
            try:
                return ConfigParser.get(self, section, val)
            except:
                if default is not None:
                    return default
                raise

    def set(self, section, val, value=None):
        if value is None:
            self.internal[section] = val
        else:
            if not self.has_section(section):
                self.add_section(section)
            ConfigParser.set(self, section, val, value)

    def has_option(self, section, name):
        try:
            return ConfigParser.has_option(self, section, name)
        except:
            return False

    def getlist(self, section, option, separator=","):
        s = self.get(section, option).strip()
        if s:
            return [i.strip() for i in s.split(separator)]
        else:
            return []