BackendsConfig methods: edit_backend() and get_backend()
This commit is contained in:
parent
4deb1398a8
commit
39f5deb7db
1 changed files with 22 additions and 5 deletions
|
|
@ -105,18 +105,35 @@ class BackendsConfig(object):
|
|||
warning('Missing field "_type" for backend "%s"', name)
|
||||
continue
|
||||
|
||||
def add_backend(self, name, _type, params):
|
||||
def add_backend(self, name, _type, params, edit=False):
|
||||
if not name:
|
||||
raise ValueError(u'Please give a name to the backend.')
|
||||
config = SafeConfigParser()
|
||||
config.read(self.confpath)
|
||||
config.add_section(name)
|
||||
if not edit:
|
||||
config.add_section(name)
|
||||
config.set(name, '_type', _type)
|
||||
for key, value in params.iteritems():
|
||||
config.set(name, key, value)
|
||||
with open(self.confpath, 'wb') as f:
|
||||
config.write(f)
|
||||
|
||||
def edit_backend(self, name, _type, params):
|
||||
return self.add_backend(name, _type, params, True)
|
||||
|
||||
def get_backend(self, name):
|
||||
config = SafeConfigParser()
|
||||
config.read(self.confpath)
|
||||
if not config.has_section(name):
|
||||
raise KeyError(u'Backend "%s" not found' % name)
|
||||
|
||||
items = dict(config.items(name, raw=True))
|
||||
try:
|
||||
return items.pop('_type'), items
|
||||
except KeyError:
|
||||
warning('Missing field "_type" for backend "%s"', name)
|
||||
raise KeyError(u'Backend "%s" not found' % name)
|
||||
|
||||
def remove_backend(self, name):
|
||||
config = SafeConfigParser()
|
||||
config.read(self.confpath)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue