config.py
159 lines
| 5.8 KiB
| text/x-python
|
PythonLexer
/ mercurial / config.py
Martin Geisler
|
r8229 | # config.py - configuration parsing for Mercurial | ||
# | ||||
# Copyright 2009 Matt Mackall <mpm@selenic.com> and others | ||||
# | ||||
# This software may be used and distributed according to the terms of the | ||||
Matt Mackall
|
r10263 | # GNU General Public License version 2 or any later version. | ||
Martin Geisler
|
r8229 | |||
Matt Mackall
|
r8144 | from i18n import _ | ||
Chad Dombrova
|
r11224 | import error, util | ||
Bryan O'Sullivan
|
r16944 | import os, errno | ||
Matt Mackall
|
r8144 | |||
Matt Mackall
|
r8186 | class config(object): | ||
Matt Mackall
|
r8144 | def __init__(self, data=None): | ||
self._data = {} | ||||
Matt Mackall
|
r8185 | self._source = {} | ||
FUJIWARA Katsunori
|
r19087 | self._unset = [] | ||
Matt Mackall
|
r8144 | if data: | ||
for k in data._data: | ||||
self._data[k] = data[k].copy() | ||||
Matt Mackall
|
r8185 | self._source = data._source.copy() | ||
Matt Mackall
|
r8144 | def copy(self): | ||
return config(self) | ||||
def __contains__(self, section): | ||||
return section in self._data | ||||
Matt Mackall
|
r8186 | def __getitem__(self, section): | ||
return self._data.get(section, {}) | ||||
def __iter__(self): | ||||
for d in self.sections(): | ||||
yield d | ||||
Matt Mackall
|
r8193 | def update(self, src): | ||
FUJIWARA Katsunori
|
r19087 | for s, n in src._unset: | ||
if s in self and n in self._data[s]: | ||||
del self._data[s][n] | ||||
del self._source[(s, n)] | ||||
Matt Mackall
|
r8193 | for s in src: | ||
Matt Mackall
|
r8144 | if s not in self: | ||
Angel Ezquerra
|
r21813 | self._data[s] = util.sortdict() | ||
Matt Mackall
|
r8193 | self._data[s].update(src._data[s]) | ||
self._source.update(src._source) | ||||
Matt Mackall
|
r8144 | def get(self, section, item, default=None): | ||
Matt Mackall
|
r8185 | return self._data.get(section, {}).get(item, default) | ||
Pierre-Yves David
|
r15919 | |||
def backup(self, section, item): | ||||
timeless@mozdev.org
|
r17527 | """return a tuple allowing restore to reinstall a previous value | ||
Pierre-Yves David
|
r15919 | |||
timeless@mozdev.org
|
r17530 | The main reason we need it is because it handles the "no data" case. | ||
Pierre-Yves David
|
r15919 | """ | ||
try: | ||||
value = self._data[section][item] | ||||
source = self.source(section, item) | ||||
return (section, item, value, source) | ||||
except KeyError: | ||||
return (section, item) | ||||
Matt Mackall
|
r8198 | def source(self, section, item): | ||
Matt Mackall
|
r8185 | return self._source.get((section, item), "") | ||
Matt Mackall
|
r8144 | def sections(self): | ||
return sorted(self._data.keys()) | ||||
def items(self, section): | ||||
Matt Mackall
|
r8185 | return self._data.get(section, {}).items() | ||
Matt Mackall
|
r8144 | def set(self, section, item, value, source=""): | ||
if section not in self: | ||||
Angel Ezquerra
|
r21813 | self._data[section] = util.sortdict() | ||
Matt Mackall
|
r8185 | self._data[section][item] = value | ||
Mads Kiilerich
|
r20789 | if source: | ||
self._source[(section, item)] = source | ||||
Matt Mackall
|
r8144 | |||
Pierre-Yves David
|
r15919 | def restore(self, data): | ||
"""restore data returned by self.backup""" | ||||
if len(data) == 4: | ||||
# restore old data | ||||
section, item, value, source = data | ||||
self._data[section][item] = value | ||||
self._source[(section, item)] = source | ||||
else: | ||||
# no data before, remove everything | ||||
section, item = data | ||||
if section in self._data: | ||||
del self._data[section][item] | ||||
self._source.pop((section, item), None) | ||||
Matt Mackall
|
r8265 | def parse(self, src, data, sections=None, remap=None, include=None): | ||
Bryan O'Sullivan
|
r16944 | sectionre = util.compilere(r'\[([^\[]+)\]') | ||
itemre = util.compilere(r'([^=\s][^=]*?)\s*=\s*(.*\S|)') | ||||
contre = util.compilere(r'\s+(\S|\S.*\S)\s*$') | ||||
emptyre = util.compilere(r'(;|#|\s*$)') | ||||
commentre = util.compilere(r'(;|#)') | ||||
unsetre = util.compilere(r'%unset\s+(\S+)') | ||||
includere = util.compilere(r'%include\s+(\S|\S.*\S)\s*$') | ||||
Matt Mackall
|
r8144 | section = "" | ||
item = None | ||||
line = 0 | ||||
r9339 | cont = False | |||
Matt Mackall
|
r8180 | |||
Nicolas Dumazet
|
r9136 | for l in data.splitlines(True): | ||
Matt Mackall
|
r8144 | line += 1 | ||
Matt Mackall
|
r16348 | if line == 1 and l.startswith('\xef\xbb\xbf'): | ||
# Someone set us up the BOM | ||||
l = l[3:] | ||||
Matt Mackall
|
r8144 | if cont: | ||
Matt Mackall
|
r14642 | if commentre.match(l): | ||
continue | ||||
Matt Mackall
|
r8144 | m = contre.match(l) | ||
if m: | ||||
Matt Mackall
|
r8193 | if sections and section not in sections: | ||
continue | ||||
Matt Mackall
|
r8144 | v = self.get(section, item) + "\n" + m.group(1) | ||
Matt Mackall
|
r8265 | self.set(section, item, v, "%s:%d" % (src, line)) | ||
Matt Mackall
|
r8144 | continue | ||
item = None | ||||
Nicolas Dumazet
|
r9469 | cont = False | ||
Matt Mackall
|
r8183 | m = includere.match(l) | ||
if m: | ||||
Chad Dombrova
|
r11224 | inc = util.expandpath(m.group(1)) | ||
Matt Mackall
|
r8265 | base = os.path.dirname(src) | ||
Matt Mackall
|
r8183 | inc = os.path.normpath(os.path.join(base, inc)) | ||
Matt Mackall
|
r8265 | if include: | ||
Martin Geisler
|
r10042 | try: | ||
include(inc, remap=remap, sections=sections) | ||||
except IOError, inst: | ||||
Matt Mackall
|
r14486 | if inst.errno != errno.ENOENT: | ||
raise error.ParseError(_("cannot include %s (%s)") | ||||
% (inc, inst.strerror), | ||||
"%s:%s" % (src, line)) | ||||
Matt Mackall
|
r8183 | continue | ||
Matt Mackall
|
r8144 | if emptyre.match(l): | ||
continue | ||||
m = sectionre.match(l) | ||||
if m: | ||||
section = m.group(1) | ||||
Matt Mackall
|
r8298 | if remap: | ||
section = remap.get(section, section) | ||||
Matt Mackall
|
r8144 | if section not in self: | ||
Angel Ezquerra
|
r21813 | self._data[section] = util.sortdict() | ||
Matt Mackall
|
r8144 | continue | ||
m = itemre.match(l) | ||||
if m: | ||||
item = m.group(1) | ||||
r9339 | cont = True | |||
Matt Mackall
|
r8193 | if sections and section not in sections: | ||
continue | ||||
Matt Mackall
|
r8265 | self.set(section, item, m.group(2), "%s:%d" % (src, line)) | ||
Matt Mackall
|
r8144 | continue | ||
Matt Mackall
|
r8184 | m = unsetre.match(l) | ||
if m: | ||||
name = m.group(1) | ||||
Matt Mackall
|
r8193 | if sections and section not in sections: | ||
continue | ||||
Martin Geisler
|
r13031 | if self.get(section, name) is not None: | ||
Matt Mackall
|
r8184 | del self._data[section][name] | ||
FUJIWARA Katsunori
|
r19087 | self._unset.append((section, name)) | ||
Matt Mackall
|
r8184 | continue | ||
Matt Mackall
|
r11288 | raise error.ParseError(l.rstrip(), ("%s:%s" % (src, line))) | ||
Matt Mackall
|
r8265 | |||
def read(self, path, fp=None, sections=None, remap=None): | ||||
if not fp: | ||||
Adrian Buehlmann
|
r13664 | fp = util.posixfile(path) | ||
Matt Mackall
|
r8265 | self.parse(path, fp.read(), sections, remap, self.read) | ||