--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/VirtualMailManager/cli/config.py Thu Jun 28 19:26:50 2012 +0000
@@ -0,0 +1,96 @@
+# -*- coding: UTF-8 -*-
+# Copyright (c) 2010 - 2012, Pascal Volk
+# See COPYING for distribution information.
+"""
+ VirtualMailManager.cli.config
+ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+ Adds some interactive stuff to the Config class.
+"""
+
+from ConfigParser import RawConfigParser
+from shutil import copy2
+
+from VirtualMailManager import ENCODING
+from VirtualMailManager.config import Config, ConfigValueError, LazyConfig
+from VirtualMailManager.errors import ConfigError, VMMError
+from VirtualMailManager.cli import w_err, w_std
+from VirtualMailManager.constants import CONF_ERROR, VMM_TOO_MANY_FAILURES
+
+_ = lambda msg: msg
+
+
+class CliConfig(Config):
+ """Adds the interactive ``configure`` method to the `Config` class
+ and overwrites `LazyConfig.set(), in order to update a single option
+ in the configuration file with a single command line command.
+ """
+
+ def configure(self, sections):
+ """Interactive method for configuring all options of the given
+ iterable ``sections`` object."""
+ input_fmt = _(u'Enter new value for option %(option)s '
+ u'[%(current_value)s]: ')
+ failures = 0
+
+ w_std(_(u'Using configuration file: %s\n') % self._cfg_filename)
+ for section in sections:
+ w_std(_(u"* Configuration section: '%s'") % section)
+ for opt, val in self.items(section):
+ failures = 0
+ while True:
+ newval = raw_input(input_fmt.encode(ENCODING, 'replace') %
+ {'option': opt, 'current_value': val})
+ if newval and newval != val:
+ try:
+ LazyConfig.set(self, '%s.%s' % (section, opt),
+ newval)
+ break
+ except (ValueError, ConfigValueError, VMMError), err:
+ w_err(0, _(u'Warning: %s') % err)
+ failures += 1
+ if failures > 2:
+ raise ConfigError(_(u'Too many failures - try '
+ u'again later.'),
+ VMM_TOO_MANY_FAILURES)
+ else:
+ break
+ print
+ if self._modified:
+ self._save_changes()
+
+ def set(self, option, value):
+ """Set the value of an option.
+
+ If the new `value` has been set, the configuration file will be
+ immediately updated.
+
+ Throws a ``ConfigError`` if `value` couldn't be converted to
+ ``LazyConfigOption.cls`` or ``LazyConfigOption.validate`` fails."""
+ section, option_ = self._get_section_option(option)
+ try:
+ val = self._cfg[section][option_].cls(value)
+ if self._cfg[section][option_].validate:
+ val = self._cfg[section][option_].validate(val)
+ except (ValueError, ConfigValueError), err:
+ raise ConfigError(str(err), CONF_ERROR)
+ # Do not write default values also skip identical values
+ if not self._cfg[section][option_].default is None:
+ old_val = self.dget(option)
+ else:
+ old_val = self.pget(option)
+ if val == old_val:
+ return
+ if not RawConfigParser.has_section(self, section):
+ self.add_section(section)
+ RawConfigParser.set(self, section, option_, val)
+ self._save_changes()
+
+ def _save_changes(self):
+ """Writes changes to the configuration file."""
+ copy2(self._cfg_filename, self._cfg_filename + '.bak')
+ self._cfg_file = open(self._cfg_filename, 'w')
+ self.write(self._cfg_file)
+ self._cfg_file.close()
+
+del _