mirror of
https://github.com/openSUSE/osc.git
synced 2024-12-28 02:36:15 +01:00
e2abfe9e6a
This can be used to check if the option belongs to the section or is just "inherited" from the defaults.
361 lines
13 KiB
Python
361 lines
13 KiB
Python
# Copyright 2008,2009 Marcus Huewe <suse-tux@gmx.de>
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License version 2
|
|
# as published by the Free Software Foundation;
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
|
|
|
|
from __future__ import print_function
|
|
|
|
import sys
|
|
|
|
if sys.version_info >= ( 3, ):
|
|
import configparser
|
|
else:
|
|
#python 2.x
|
|
import ConfigParser as configparser
|
|
|
|
import re
|
|
|
|
# inspired from http://code.google.com/p/iniparse/ - although their implementation is
|
|
# quite different
|
|
|
|
class ConfigLineOrder:
|
|
"""
|
|
A ConfigLineOrder() instance task is to preserve the order of a config file.
|
|
It keeps track of all lines (including comments) in the _lines list. This list
|
|
either contains SectionLine() instances or CommentLine() instances.
|
|
"""
|
|
def __init__(self):
|
|
self._lines = []
|
|
|
|
def _append(self, line_obj):
|
|
self._lines.append(line_obj)
|
|
|
|
def _find_section(self, section):
|
|
for line in self._lines:
|
|
if line.type == 'section' and line.name == section:
|
|
return line
|
|
return None
|
|
|
|
def add_section(self, sectname):
|
|
self._append(SectionLine(sectname))
|
|
|
|
def get_section(self, sectname):
|
|
section = self._find_section(sectname)
|
|
if section:
|
|
return section
|
|
section = SectionLine(sectname)
|
|
self._append(section)
|
|
return section
|
|
|
|
def add_other(self, sectname, line):
|
|
if sectname:
|
|
self.get_section(sectname).add_other(line)
|
|
else:
|
|
self._append(CommentLine(line))
|
|
|
|
def keys(self):
|
|
return [ i.name for i in self._lines if i.type == 'section' ]
|
|
|
|
def __setitem__(self, key, value):
|
|
section = SectionLine(key)
|
|
self._append(section)
|
|
|
|
def __getitem__(self, key):
|
|
section = self._find_section(key)
|
|
if not section:
|
|
raise KeyError()
|
|
return section
|
|
|
|
def __delitem__(self, key):
|
|
line = self._find_section(key)
|
|
if not line:
|
|
raise KeyError(key)
|
|
self._lines.remove(line)
|
|
|
|
def __iter__(self):
|
|
#return self._lines.__iter__()
|
|
for line in self._lines:
|
|
if line.type == 'section':
|
|
yield line.name
|
|
raise StopIteration()
|
|
|
|
class Line:
|
|
"""Base class for all line objects"""
|
|
def __init__(self, name, type):
|
|
self.name = name
|
|
self.type = type
|
|
|
|
class SectionLine(Line):
|
|
"""
|
|
This class represents a [section]. It stores all lines which belongs to
|
|
this certain section in the _lines list. The _lines list either contains
|
|
CommentLine() or OptionLine() instances.
|
|
"""
|
|
def __init__(self, sectname, dict = {}):
|
|
Line.__init__(self, sectname, 'section')
|
|
self._lines = []
|
|
self._dict = dict
|
|
|
|
def _find(self, name):
|
|
for line in self._lines:
|
|
if line.name == name:
|
|
return line
|
|
return None
|
|
|
|
def _add_option(self, optname, value = None, line = None, sep = '='):
|
|
if value is None and line is None:
|
|
raise configparser.Error('Either value or line must be passed in')
|
|
elif value and line:
|
|
raise configparser.Error('value and line are mutually exclusive')
|
|
|
|
if value is not None:
|
|
line = '%s%s%s' % (optname, sep, value)
|
|
opt = self._find(optname)
|
|
if opt:
|
|
opt.format(line)
|
|
else:
|
|
self._lines.append(OptionLine(optname, line))
|
|
|
|
def add_other(self, line):
|
|
self._lines.append(CommentLine(line))
|
|
|
|
def copy(self):
|
|
return dict(self.items())
|
|
|
|
def items(self):
|
|
return [ (i.name, i.value) for i in self._lines if i.type == 'option' ]
|
|
|
|
def keys(self):
|
|
return [ i.name for i in self._lines ]
|
|
|
|
def __setitem__(self, key, val):
|
|
self._add_option(key, val)
|
|
|
|
def __getitem__(self, key):
|
|
line = self._find(key)
|
|
if not line:
|
|
raise KeyError(key)
|
|
return str(line)
|
|
|
|
def __delitem__(self, key):
|
|
line = self._find(key)
|
|
if not line:
|
|
raise KeyError(key)
|
|
self._lines.remove(line)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
# XXX: needed to support 'x' in cp._sections['sectname']
|
|
def __iter__(self):
|
|
for line in self._lines:
|
|
yield line.name
|
|
raise StopIteration()
|
|
|
|
|
|
class CommentLine(Line):
|
|
"""Store a commentline"""
|
|
def __init__(self, line):
|
|
Line.__init__(self, line.strip('\n'), 'comment')
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
class OptionLine(Line):
|
|
"""
|
|
This class represents an option. The class' "name" attribute is used
|
|
to store the option's name and the "value" attribute contains the option's
|
|
value. The "frmt" attribute preserves the format which was used in the configuration
|
|
file.
|
|
Example:
|
|
optionx:<SPACE><SPACE>value
|
|
=> self.frmt = '%s:<SPACE><SPACE>%s'
|
|
optiony<SPACE>=<SPACE>value<SPACE>;<SPACE>some_comment
|
|
=> self.frmt = '%s<SPACE>=<SPACE><SPACE>%s<SPACE>;<SPACE>some_comment
|
|
"""
|
|
|
|
def __init__(self, optname, line):
|
|
Line.__init__(self, optname, 'option')
|
|
self.name = optname
|
|
self.format(line)
|
|
|
|
def format(self, line):
|
|
mo = configparser.ConfigParser.OPTCRE.match(line.strip())
|
|
key, val = mo.group('option', 'value')
|
|
self.frmt = line.replace(key.strip(), '%s', 1)
|
|
pos = val.find(' ;')
|
|
if pos >= 0:
|
|
val = val[:pos]
|
|
self.value = val
|
|
self.frmt = self.frmt.replace(val.strip(), '%s', 1).rstrip('\n')
|
|
|
|
def __str__(self):
|
|
return self.value
|
|
|
|
|
|
class OscConfigParser(configparser.SafeConfigParser):
|
|
"""
|
|
OscConfigParser() behaves like a normal ConfigParser() object. The
|
|
only differences is that it preserves the order+format of configuration entries
|
|
and that it stores comments.
|
|
In order to keep the order and the format it makes use of the ConfigLineOrder()
|
|
class.
|
|
"""
|
|
def __init__(self, defaults={}):
|
|
configparser.SafeConfigParser.__init__(self, defaults)
|
|
self._sections = ConfigLineOrder()
|
|
|
|
# XXX: unfortunately we have to override the _read() method from the ConfigParser()
|
|
# class because a) we need to store comments b) the original version doesn't use
|
|
# the its set methods to add and set sections, options etc. instead they use a
|
|
# dictionary (this makes it hard for subclasses to use their own objects, IMHO
|
|
# a bug) and c) in case of an option we need the complete line to store the format.
|
|
# This all sounds complicated but it isn't - we only needed some slight changes
|
|
def _read(self, fp, fpname):
|
|
"""Parse a sectioned setup file.
|
|
|
|
The sections in setup file contains a title line at the top,
|
|
indicated by a name in square brackets (`[]'), plus key/value
|
|
options lines, indicated by `name: value' format lines.
|
|
Continuations are represented by an embedded newline then
|
|
leading whitespace. Blank lines, lines beginning with a '#',
|
|
and just about everything else are ignored.
|
|
"""
|
|
cursect = None # None, or a dictionary
|
|
optname = None
|
|
lineno = 0
|
|
e = None # None, or an exception
|
|
while True:
|
|
line = fp.readline()
|
|
if not line:
|
|
break
|
|
lineno = lineno + 1
|
|
# comment or blank line?
|
|
if line.strip() == '' or line[0] in '#;':
|
|
self._sections.add_other(cursect, line)
|
|
continue
|
|
if line.split(None, 1)[0].lower() == 'rem' and line[0] in "rR":
|
|
# no leading whitespace
|
|
continue
|
|
# continuation line?
|
|
if line[0].isspace() and cursect is not None and optname:
|
|
value = line.strip()
|
|
if value:
|
|
#cursect[optname] = "%s\n%s" % (cursect[optname], value)
|
|
#self.set(cursect, optname, "%s\n%s" % (self.get(cursect, optname), value))
|
|
if cursect == configparser.DEFAULTSECT:
|
|
self._defaults[optname] = "%s\n%s" % (self._defaults[optname], value)
|
|
else:
|
|
# use the raw value here (original version uses raw=False)
|
|
self._sections[cursect]._find(optname).value = '%s\n%s' % (self.get(cursect, optname, raw=True), value)
|
|
# a section header or option header?
|
|
else:
|
|
# is it a section header?
|
|
mo = self.SECTCRE.match(line)
|
|
if mo:
|
|
sectname = mo.group('header')
|
|
if sectname in self._sections:
|
|
cursect = self._sections[sectname]
|
|
elif sectname == configparser.DEFAULTSECT:
|
|
cursect = self._defaults
|
|
else:
|
|
#cursect = {'__name__': sectname}
|
|
#self._sections[sectname] = cursect
|
|
self.add_section(sectname)
|
|
self.set(sectname, '__name__', sectname)
|
|
# So sections can't start with a continuation line
|
|
cursect = sectname
|
|
optname = None
|
|
# no section header in the file?
|
|
elif cursect is None:
|
|
raise configparser.MissingSectionHeaderError(fpname, lineno, line)
|
|
# an option line?
|
|
else:
|
|
mo = self.OPTCRE.match(line)
|
|
if mo:
|
|
optname, vi, optval = mo.group('option', 'vi', 'value')
|
|
if vi in ('=', ':') and ';' in optval:
|
|
# ';' is a comment delimiter only if it follows
|
|
# a spacing character
|
|
pos = optval.find(';')
|
|
if pos != -1 and optval[pos-1].isspace():
|
|
optval = optval[:pos]
|
|
optval = optval.strip()
|
|
# allow empty values
|
|
if optval == '""':
|
|
optval = ''
|
|
optname = self.optionxform(optname.rstrip())
|
|
if cursect == configparser.DEFAULTSECT:
|
|
self._defaults[optname] = optval
|
|
else:
|
|
self._sections[cursect]._add_option(optname, line=line)
|
|
else:
|
|
# a non-fatal parsing error occurred. set up the
|
|
# exception but keep going. the exception will be
|
|
# raised at the end of the file and will contain a
|
|
# list of all bogus lines
|
|
if not e:
|
|
e = configparser.ParsingError(fpname)
|
|
e.append(lineno, repr(line))
|
|
# if any parsing errors occurred, raise an exception
|
|
if e:
|
|
raise e # pylint: disable-msg=E0702
|
|
|
|
def write(self, fp, comments = False):
|
|
"""
|
|
write the configuration file. If comments is True all comments etc.
|
|
will be written to fp otherwise the ConfigParsers' default write method
|
|
will be called.
|
|
"""
|
|
if comments:
|
|
fp.write(str(self))
|
|
fp.write('\n')
|
|
else:
|
|
configparser.SafeConfigParser.write(self, fp)
|
|
|
|
def has_option(self, section, option, proper=False, **kwargs):
|
|
"""
|
|
Returns True, if the passed section contains the specified option.
|
|
If proper is True, True is only returned if the option is owned by
|
|
this section and not "inherited" from the default.
|
|
"""
|
|
if proper:
|
|
return self.optionxform(option) in self._sections[section].keys()
|
|
return configparser.SafeConfigParser.has_option(self, section, option, **kwargs)
|
|
|
|
# XXX: simplify!
|
|
def __str__(self):
|
|
ret = []
|
|
first = True
|
|
for line in self._sections._lines:
|
|
if line.type == 'section':
|
|
if first:
|
|
first = False
|
|
else:
|
|
ret.append('')
|
|
ret.append('[%s]' % line.name)
|
|
for sline in line._lines:
|
|
if sline.name == '__name__':
|
|
continue
|
|
if sline.type == 'option':
|
|
# special handling for continuation lines
|
|
val = '\n '.join(sline.value.split('\n'))
|
|
ret.append(sline.frmt % (sline.name, val))
|
|
elif str(sline) != '':
|
|
ret.append(str(sline))
|
|
else:
|
|
ret.append(str(line))
|
|
return '\n'.join(ret)
|
|
|
|
# vim: sw=4 et
|