nojhan/weboob-devel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

BackendsConfig methods: edit_backend() and get_backend()

Browse source
This commit is contained in:
Romain Bignon 2010-06-21 22:59:44 +02:00
commit 39f5deb7db
1 changed files with 22 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

27
weboob/modules.py
View file

@ -1,16 +1,16 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# Copyright(C) 2010 Romain Bignon # Copyright(C) 2010 Romain Bignon
# #
# This program is free software; you can redistribute it and/or modify # This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by # it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 3 of the License. # the Free Software Foundation, version 3 of the License.
# #
# This program is distributed in the hope that it will be useful, # This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of # but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details. # GNU General Public License for more details.
# #
# You should have received a copy of the GNU General Public License # You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software # along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
@ -105,18 +105,35 @@ class BackendsConfig(object):
warning('Missing field "_type" for backend "%s"', name) warning('Missing field "_type" for backend "%s"', name)
continue continue
def add_backend(self, name, _type, params): def add_backend(self, name, _type, params, edit=False):
if not name: if not name:
raise ValueError(u'Please give a name to the backend.') raise ValueError(u'Please give a name to the backend.')
config = SafeConfigParser() config = SafeConfigParser()
config.read(self.confpath) config.read(self.confpath)
config.add_section(name) if not edit:
config.add_section(name)
config.set(name, '_type', _type) config.set(name, '_type', _type)
for key, value in params.iteritems(): for key, value in params.iteritems():
config.set(name, key, value) config.set(name, key, value)
with open(self.confpath, 'wb') as f: with open(self.confpath, 'wb') as f:
config.write(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): def remove_backend(self, name):
config = SafeConfigParser() config = SafeConfigParser()
config.read(self.confpath) config.read(self.confpath)