| Index: chainedconfigparser.py |
| =================================================================== |
| --- a/chainedconfigparser.py |
| +++ b/chainedconfigparser.py |
| @@ -4,7 +4,9 @@ |
| # License, v. 2.0. If a copy of the MPL was not distributed with this |
| # file, You can obtain one at http://mozilla.org/MPL/2.0/. |
| -import os, codecs, ConfigParser |
| +import os |
| +import re |
| +import ConfigParser |
| class Item(tuple): |
| def __new__(cls, name, value, source): |
| @@ -12,8 +14,8 @@ |
| result.source = source |
| return result |
| -class ChainedConfigParser: |
| - """ |
| +class ChainedConfigParser(ConfigParser.SafeConfigParser): |
| + ''' |
| This class provides essentially the same interfaces as SafeConfigParser but |
| allows chaining configuration files so that one config file provides the |
| default values for the other. To specify the config file to inherit from |
| @@ -33,82 +35,73 @@ |
| method is provided to get the path of the configuration file defining this |
| option (for relative paths). Items returned by the items() function also |
| have a source attribute serving the same purpose. |
|
Wladimir Palant
2015/06/23 09:43:36
Extend documentation to mention the new += and -=
Sebastian Noack
2015/06/24 08:51:09
Done.
|
| - """ |
| + ''' |
| - def __init__(self, path): |
| - self.chain = [] |
| - self.read_path(path) |
| + def __init__(self): |
| + ConfigParser.SafeConfigParser.__init__(self) |
| + self._origin = {} |
| - def read_path(self, path): |
| - if len(self.chain) >= 5: |
| - raise Exception('Too much inheritance in config files') |
| + def _get_parser_chain(self, file, filename): |
| + parsers = [] |
| - config = ConfigParser.SafeConfigParser() |
| - config.optionxform = str |
| - config.source_path = path |
| - handle = codecs.open(path, 'rb', encoding='utf-8') |
| - config.readfp(handle) |
| - handle.close() |
| - self.chain.append(config) |
| + parser = ConfigParser.SafeConfigParser() |
| + parser._read(file, filename) |
|
Wladimir Palant
2015/06/23 09:43:35
Please use public API here:
parser.readfp(file,
Sebastian Noack
2015/06/24 08:51:10
This code has been gone while addressing the comme
|
| - if config.has_section('default') and config.has_option('default', 'inherit'): |
| - parts = config.get('default', 'inherit').split('/') |
| - defaults_path = os.path.join(os.path.dirname(path), *parts) |
| - self.read_path(defaults_path) |
| + while True: |
| + parsers.insert(0, (parser, filename)) |
| - def defaults(self): |
| - result = {} |
| - for config in reverse(self.chain): |
| - for key, value in config.defaults().iteritems(): |
| - result[key] = value |
| - return result |
| + try: |
| + inherit = parser.get('default', 'inherit') |
| + except (ConfigParser.NoSectionError, ConfigParser.NoOptionError): |
| + return parsers |
| - def sections(self): |
| - result = set() |
| - for config in self.chain: |
| - for section in config.sections(): |
| - result.add(section) |
| - return list(result) |
| + filename = os.path.join(os.path.dirname(filename), *inherit.split('/')) |
| + parser = ConfigParser.SafeConfigParser() |
| + parser.read(filename) |
| - def has_section(self, section): |
| - for config in self.chain: |
| - if config.has_section(section): |
| - return True |
| - return False |
| + def _apply_diff(self, section, option, value): |
| + addition = option.endswith('+') |
| + removal = option.endswith('-') |
| - def options(self, section): |
| - result = set() |
| - for config in self.chain: |
| - if config.has_section(section): |
| - for option in config.options(section): |
| - result.add(option) |
| - return list(result) |
| + if addition or removal: |
| + option = option[:-1].rstrip() |
| + old_value = self.get(section, option) |
|
Wladimir Palant
2015/06/23 09:43:36
This will throw when trying to change an option th
Sebastian Noack
2015/06/24 08:51:09
Done.
|
| - def has_option(self, section, option): |
| - for config in self.chain: |
| - if config.has_section(section) and config.has_option(section, option): |
| - return True |
| - return False |
| + if addition: |
| + value = '%s %s' % (old_value, value) |
|
Wladimir Palant
2015/06/23 09:43:35
I don't think that we want duplicate values - we a
Sebastian Noack
2015/06/24 08:51:10
Done.
|
| + elif removal: |
|
Wladimir Palant
2015/06/23 09:43:35
Nit: no need for elif, can be simply else. But I d
Sebastian Noack
2015/06/24 08:51:09
Done.
|
| + value = re.sub(r'\b(?:%s)\b\s*' % '|'.join(map(re.escape, value.split())), '', old_value).rstrip() |
|
Wladimir Palant
2015/06/23 09:43:35
I don't think that regular expressions are the rig
Sebastian Noack
2015/06/24 08:51:09
Done.
|
| - def get(self, section, option): |
| - for config in self.chain: |
| - if config.has_section(section) and config.has_option(section, option): |
| - return config.get(section, option) |
| - raise ConfigParser.NoOptionError(option, section) |
| + return option, value |
| - def items(self, section): |
| - seen = set() |
| - result = [] |
| - for config in self.chain: |
| - if config.has_section(section): |
| - for name, value in config.items(section): |
| - if name not in seen: |
| - seen.add(name) |
| - result.append(Item(name, value, config.source_path)) |
| - return result |
| + def _read(self, file, filename): |
|
Wladimir Palant
2015/06/23 09:43:35
Is it really a good idea to override private metho
Sebastian Noack
2015/06/24 08:51:10
Overriding read() and readfp() would require quite
|
| + parsers = self._get_parser_chain(file, filename) |
| + |
| + for parser, filename in parsers: |
| + for section in parser.sections(): |
| + for option, value in parser.items(section): |
| + option, value = self._apply_diff(section, option, value) |
| + try: |
| + self.set(section, option, value) |
| + except ConfigParser.NoSectionError: |
| + try: |
| + self.add_section(section) |
| + except ValueError: |
| + # add_section() hardcodes 'default' and raises a ValueError if |
| + # you try to add a section called like that (case insensitive). |
| + # This bug has been fixed in Python 3. |
| + self._sections[section] = self._dict() |
|
Wladimir Palant
2015/06/23 09:43:36
I cannot say that I like this hack. How about simp
Sebastian Noack
2015/06/24 08:51:09
"default" != "DEFAULT". The latter is handled spec
Wladimir Palant
2015/06/25 14:11:40
I see. Still, should we access private variables i
Sebastian Noack
2015/06/25 16:06:28
The except block will only be reached in Python ve
Wladimir Palant
2015/06/25 16:12:28
That is: every Python version we support. And I'm
Sebastian Noack
2015/06/25 23:05:31
In case I didn't stress this enough, this is a bug
Wladimir Palant
2015/06/26 13:17:21
It doesn't matter what you call it - it's document
Sebastian Noack
2015/06/26 13:37:53
So your concerns here are only about using interna
|
| + self.set(section, option, value) |
| + self._origin[(section, option)] = filename |
|
Wladimir Palant
2015/06/23 09:43:35
Ok, we have a problem right here... Consider the f
Sebastian Noack
2015/06/24 08:51:09
I first thought that we can simply rely on the sou
|
| + |
| + def items(self, section, *args, **kwargs): |
| + items = [] |
| + for option, value in ConfigParser.SafeConfigParser.items(self, section, *args, **kwargs): |
| + items.append(Item(option, value, self._origin[(section, option)])) |
| + return items |
| def option_source(self, section, option): |
| - for config in self.chain: |
| - if config.has_section(section) and config.has_option(section, option): |
| - return config.source_path |
| - raise ConfigParser.NoOptionError(option, section) |
| + try: |
| + return self._origin[(section, option)] |
| + except KeyError: |
| + raise ConfigParser.NoOptionError(option, section) |