Limnoria/src/registry.py

860 lines
30 KiB
Python
Raw Normal View History

2005-01-19 14:14:38 +01:00
###
2005-01-19 14:33:05 +01:00
# Copyright (c) 2004-2005, Jeremiah Fincher
# Copyright (c) 2009-2010, James McCoy
2005-01-19 14:14:38 +01:00
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice,
# this list of conditions, and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright notice,
# this list of conditions, and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
# * Neither the name of the author of this software nor the name of
# contributors to this software may be used to endorse or promote products
# derived from this software without specific prior written consent.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
###
import re
import os
import time
2012-12-08 20:59:36 +01:00
import json
import codecs
2005-01-19 14:14:38 +01:00
import string
import textwrap
2015-08-11 16:50:23 +02:00
from . import utils, i18n
from .utils import minisix
_ = i18n.PluginInternationalization()
2005-01-19 14:14:38 +01:00
def error(s):
"""Replace me with something better from another module!"""
print('***', s)
2005-01-19 14:14:38 +01:00
def exception(s):
"""Ditto!"""
print('***', s, 'A bad exception.')
2005-01-19 14:14:38 +01:00
class RegistryException(Exception):
pass
class InvalidRegistryFile(RegistryException):
pass
class InvalidRegistryName(RegistryException):
pass
class InvalidRegistryValue(RegistryException):
pass
class NonExistentRegistryEntry(RegistryException, AttributeError):
# If we use hasattr() on a configuration group/value, Python 3 calls
# __getattr__ and looks for an AttributeError, so __getattr__ has to
# raise an AttributeError if a registry entry does not exist.
2005-01-19 14:14:38 +01:00
pass
ENCODING = 'string_escape' if minisix.PY2 else 'unicode_escape'
decoder = codecs.getdecoder(ENCODING)
encoder = codecs.getencoder(ENCODING)
if hasattr(time, 'monotonic'):
monotonic_time = time.monotonic
else:
# fallback for python < 3.3
monotonic_time = time.time
2005-01-19 14:14:38 +01:00
_cache = utils.InsensitivePreservingDict()
_lastModified = 0
2012-08-04 13:13:16 +02:00
def open_registry(filename, clear=False):
2005-01-19 14:14:38 +01:00
"""Initializes the module by loading the registry file into memory."""
global _lastModified
if clear:
_cache.clear()
2012-08-04 13:13:16 +02:00
_fd = open(filename)
fd = utils.file.nonCommentNonEmptyLines(_fd)
2005-01-19 14:14:38 +01:00
acc = ''
slashEnd = re.compile(r'\\*$')
2005-01-19 14:14:38 +01:00
for line in fd:
line = line.rstrip('\r\n')
# XXX There should be some way to determine whether or not we're
# starting a new variable or not. As it is, if there's a backslash
# at the end of every line in a variable, it won't be read, and
# worse, the error will pass silently.
#
# If the line ends in an odd number of backslashes, then there is a
# line-continutation.
m = slashEnd.search(line)
if m and len(m.group(0)) % 2:
2005-01-19 14:14:38 +01:00
acc += line[:-1]
continue
else:
acc += line
try:
(key, value) = re.split(r'(?<!\\):', acc, 1)
key = key.strip()
value = value.strip()
value = decoder(value)[0]
2005-01-19 14:14:38 +01:00
acc = ''
except ValueError:
raise InvalidRegistryFile('Error unpacking line %r' % acc)
2005-01-19 14:14:38 +01:00
_cache[key] = value
_lastModified = monotonic_time()
2005-01-19 14:14:38 +01:00
_fd.close()
CONF_FILE_HEADER = """
######
2017-08-01 07:50:01 +02:00
# Although it is technically possible to do so, we do not recommend that
# you edit this file with a text editor.
# Whenever possible, do it on IRC using the Config plugin, which
# checks values you set are valid before writing them to the
# configuration.
# Moreover, if you edit this file while the bot is running, your
# changes may be lost.
######
"""
2005-01-19 14:14:38 +01:00
def close(registry, filename, private=True):
first = True
fd = utils.file.AtomicFile(filename)
fd.write(CONF_FILE_HEADER)
2005-01-19 14:14:38 +01:00
for (name, value) in registry.getValues(getChildren=True):
help = value.help()
if help:
lines = textwrap.wrap(value._help)
2005-01-19 14:14:38 +01:00
for (i, line) in enumerate(lines):
lines[i] = '# %s\n' % line
lines.insert(0, '###\n')
if first:
first = False
else:
lines.insert(0, '\n')
if hasattr(value, 'value'):
if value._showDefault:
lines.append('#\n')
try:
x = value.__class__(value._default, value._help)
except Exception as e:
2012-07-30 11:18:02 +02:00
exception('Exception instantiating default for %s:' %
2005-01-19 14:14:38 +01:00
value._name)
try:
lines.append('# Default value: %s\n' % x)
except Exception:
2012-07-30 11:18:02 +02:00
exception('Exception printing default value of %s:' %
2005-01-19 14:14:38 +01:00
value._name)
lines.append('###\n')
fd.writelines(lines)
if hasattr(value, 'value'): # This lets us print help for non-values.
2005-01-19 14:14:38 +01:00
try:
if private or not value._private:
s = value.serialize()
else:
s = 'CENSORED'
fd.write('%s: %s\n' % (name, s))
2015-03-03 09:02:29 +01:00
except Exception:
2005-01-19 14:14:38 +01:00
exception('Exception printing value:')
fd.close()
def isValidRegistryName(name):
# Now we can have . and : in names. I'm still gonna call shenanigans on
# anyone who tries to have spaces (though technically I can't see any
# reason why it wouldn't work). We also reserve all names starting with
# underscores for internal use.
return len(name.split()) == 1 and not name.startswith('_')
def escape(name):
name = encoder(name)[0].decode()
2005-01-19 14:14:38 +01:00
name = name.replace(':', '\\:')
name = name.replace('.', '\\.')
return name
def unescape(name):
name = name.replace('\\.', '.')
name = name.replace('\\:', ':')
name = decoder(name.encode())[0]
2005-01-19 14:14:38 +01:00
return name
_splitRe = re.compile(r'(?<!\\)\.')
def split(name):
return list(map(unescape, _splitRe.split(name)))
2005-01-19 14:14:38 +01:00
def join(names):
return '.'.join(map(escape, names))
class Group(object):
"""A group; it doesn't hold a value unless handled by a subclass."""
2018-09-25 18:59:46 +02:00
__slots__ = ('_help', '_name', '_added', '_children', '_lastModified',
'_private', '_supplyDefault', '_orderAlphabetically', '_wasSet')
2005-01-19 14:14:38 +01:00
def __init__(self, help='', supplyDefault=False,
2016-07-23 15:36:00 +02:00
orderAlphabetically=True, private=False):
self._help = utils.str.normalizeWhitespace(help)
2005-01-19 14:14:38 +01:00
self._name = 'unset'
self._added = []
self._children = utils.InsensitivePreservingDict()
self._lastModified = 0
self._private = private
self._supplyDefault = supplyDefault
self._orderAlphabetically = orderAlphabetically
self._wasSet = True
2005-01-19 14:14:38 +01:00
def __call__(self):
raise ValueError('Groups have no value.')
2005-01-19 14:14:38 +01:00
def __nonExistentEntry(self, attr):
2019-11-24 12:47:03 +01:00
s = _('%r is not a valid entry in %r') % (attr, self._name)
raise NonExistentRegistryEntry(s)
2005-01-19 14:14:38 +01:00
2019-08-15 12:22:43 +02:00
def _makeChild(self, attr, s):
2005-01-19 14:14:38 +01:00
v = self.__class__(self._default, self._help)
v.set(s)
v._wasSet = False
2005-01-19 14:14:38 +01:00
v._supplyDefault = False
v._help = '' # Clear this so it doesn't print a bazillion times.
self.register(attr, v)
return v
def __hasattr__(self, attr):
return attr in self._children
2005-01-19 14:14:38 +01:00
def __getattr__(self, attr):
2019-08-15 12:22:43 +02:00
if attr.startswith('_'):
raise AttributeError('%s has no attribute %s' %
(self.__class__.__name__, attr))
2019-08-15 12:22:43 +02:00
elif attr in self._children:
2005-01-19 14:14:38 +01:00
return self._children[attr]
elif self._supplyDefault:
2019-08-15 12:22:43 +02:00
return self._makeChild(attr, str(self))
2005-01-19 14:14:38 +01:00
else:
self.__nonExistentEntry(attr)
def help(self):
return i18n.PluginInternationalization().__call__(self._help)
2005-01-19 14:14:38 +01:00
def get(self, attr):
# Not getattr(self, attr) because some nodes might have groups that
# are named the same as their methods.
return self.__getattr__(attr)
def setName(self, name):
#print '***', name
self._name = name
if name in _cache and self._lastModified < _lastModified:
#print '***>', _cache[name]
self.set(_cache[name])
if self._supplyDefault:
2015-08-08 22:20:14 +02:00
for (k, v) in _cache.items():
2005-01-19 14:14:38 +01:00
if k.startswith(self._name):
rest = k[len(self._name)+1:] # +1 is for .
parts = split(rest)
if len(parts) == 1 and parts[0] == name:
try:
2019-08-15 12:22:43 +02:00
self._makeChild(name, v)
2005-01-19 14:14:38 +01:00
except InvalidRegistryValue:
# It's probably supposed to be registered later.
pass
def register(self, name, node=None):
if not isValidRegistryName(name):
raise InvalidRegistryName(name)
2005-01-19 14:14:38 +01:00
if node is None:
node = Group(private=self._private)
else:
node._private = node._private or self._private
2005-01-19 14:14:38 +01:00
# We tried in any number of horrible ways to make it so that
# re-registering something would work. It doesn't, plain and simple.
# For the longest time, we had an "Is this right?" comment here, but
# from experience, we now know that it most definitely *is* right.
if name not in self._children:
self._children[name] = node
self._added.append(name)
names = split(self._name)
names.append(name)
fullname = join(names)
node.setName(fullname)
else:
# We do this in order to reload the help, if it changed.
if node._help != '' and node._help != self._children[name]._help:
self._children[name]._help = node._help
2005-01-19 14:14:38 +01:00
# We do this so the return value from here is at least useful;
# otherwise, we're just returning a useless, unattached node
# that's simply a waste of space.
node = self._children[name]
return node
def unregister(self, name):
try:
node = self._children[name]
del self._children[name]
# We do this because we need to remove case-insensitively.
name = name.lower()
for elt in reversed(self._added):
if elt.lower() == name:
self._added.remove(elt)
if node._name in _cache:
del _cache[node._name]
return node
except KeyError:
self.__nonExistentEntry(name)
def rename(self, old, new):
node = self.unregister(old)
self.register(new, node)
def getValues(self, getChildren=False, fullNames=True):
L = []
if self._orderAlphabetically:
self._added.sort()
for name in self._added:
node = self._children[name]
if hasattr(node, 'value') or hasattr(node, 'help'):
if node._wasSet:
2005-01-19 14:14:38 +01:00
L.append((node._name, node))
if getChildren:
L.extend(node.getValues(getChildren, fullNames))
if not fullNames:
L = [(split(s)[-1], node) for (s, node) in L]
return L
class _NoValueGiven:
# Special value for Value.error()
pass
2005-01-19 14:14:38 +01:00
class Value(Group):
"""Invalid registry value. If you're getting this message, report it,
because we forgot to put a proper help string here."""
2018-09-25 18:59:46 +02:00
__slots__ = ('__parent', '_default', '_showDefault', '_help', '_callbacks',
2019-08-15 12:22:43 +02:00
'value', '_networkValue', '_channelValue', '_opSettable')
2005-01-19 14:14:38 +01:00
def __init__(self, default, help, setDefault=True,
showDefault=True, **kwargs):
self.__parent = super(Value, self)
self.__parent.__init__(help, **kwargs)
self._default = default
self._showDefault = showDefault
self._help = utils.str.normalizeWhitespace(help.strip())
self._callbacks = []
self._networkValue = False
self._channelValue = False
2005-01-19 14:14:38 +01:00
if setDefault:
self.setValue(default)
2019-08-15 12:22:43 +02:00
def _makeChild(self, attr, s):
v = self.__class__(self._default, self._help)
v.set(s)
v._wasSet = False
if self._networkValue and self._channelValue:
# If this is both a network-specific and channel-specific value,
# then the child is (only) channel-specific.
v._networkValue = False
v._channelValue = True
v._supplyDefault = True
else:
# Otherwise, the child is neither network-specific or
# channel-specific.
v._supplyDefault = False
v._help = '' # Clear this so it doesn't print a bazillion times.
self.register(attr, v)
return v
def error(self, value=_NoValueGiven):
if hasattr(self, 'errormsg') and value is not _NoValueGiven:
try:
s = self.errormsg % value
except TypeError:
s = self.errormsg
elif self.__doc__:
2005-01-19 14:14:38 +01:00
s = self.__doc__
else:
s = """%s has no docstring. If you're getting this message,
2005-02-01 12:57:26 +01:00
report it, because we forgot to put a proper help string here."""%\
self._name
e = InvalidRegistryValue(utils.str.normalizeWhitespace(s))
2005-01-19 14:14:38 +01:00
e.value = self
raise e
2019-08-15 12:22:43 +02:00
def getSpecific(self, network=None, channel=None, check=True):
"""Gets the network-specific and/or channel-specific value of this
Value.
If `check=True` (the default), this will raise an error if `network`
(resp. `channel`) is provided but this Value is not network-specific
(resp. channel-specific). If `check=False`, then `network` and/or
`channel` may be silently ignored.
"""
if network and not self._networkValue:
if check:
raise NonExistentRegistryEntry('%s is not network-specific' %
self._name)
else:
network = None
if channel and not self._channelValue:
if check:
raise NonExistentRegistryEntry('%s is not channel-specific' %
self._name)
else:
channel = None
if network and channel:
# The complicated case. We want a net+chan specific value,
# which may come in three different ways:
#
# 1. it was set explicitely net+chan
# 2. it's inherited from a net specific value (which may itself be
# inherited from the base value)
# 3. it's inherited from the chan specific value (which is not a
# actually a parent in the registry tree, but we need this to
# load configuration from old bots).
#
# The choice between 2 and 3 is done by checking which of the
# net-specific and chan-specific values was set explicitely by
# a user/admin. In case both were, the net-specific value is used
# (there is no particular reason for this, I just think it makes
# more sense).
network_value = self.get(':' + network)
network_channel_value = network_value.get(channel)
channel_value = self.get(channel)
if network_value._wasSet or network_channel_value._wasSet:
# cases 1 and 2
return network_channel_value
else:
# case 3
return channel_value
elif network:
return self.get(':' + network)
elif channel:
return self.get(channel)
else:
return self
2005-01-19 14:14:38 +01:00
def setName(self, *args):
if self._name == 'unset':
self._lastModified = 0
self.__parent.setName(*args)
self._lastModified = monotonic_time()
2005-01-19 14:14:38 +01:00
def set(self, s):
"""Override this with a function to convert a string to whatever type
you want, and call self.setValue to set the value."""
self.setValue(s)
self._wasSet = True
2005-01-19 14:14:38 +01:00
def setValue(self, v):
"""Check conditions on the actual value type here. I.e., if you're a
IntegerLessThanOneHundred (all your values must be integers less than
100) convert to an integer in set() and check that the integer is less
than 100 in this method. You *must* call this parent method in your
own setValue."""
self._setValue(v, inherited=False)
def _setValue(self, v, inherited):
"""Like setValue, but accepted an extra 'inherited' argument.
inherited=True means the value is inherited from the parent, so if
the parent gets a new value, this group will get the new value as
well."""
self._lastModified = monotonic_time()
2005-01-19 14:14:38 +01:00
self.value = v
if self._supplyDefault:
for (name, child) in list(self._children.items()):
if not child._wasSet:
child._setValue(v, inherited=True)
# We call the callback once everything is clean
for callback, args, kwargs in self._callbacks:
callback(*args, **kwargs)
self._wasSet = not inherited
def context(self, value):
"""Return a context manager object, which sets this variable to a
temporary value, and set the previous value back when exiting the
context."""
class Context:
def __enter__(self2):
self2._old_value = self.value
self.setValue(value)
def __exit__(self2, exc_type, exc_value, traceback):
self.setValue(self2._old_value)
return Context()
def addCallback(self, callback, *args, **kwargs):
"""Add a callback to the list. A callback is a function that will be
called when the value is changed. You can give this function as many
extra arguments as you wish, they will be passed to the callback."""
self._callbacks.append((callback, args, kwargs))
def removeCallback(self, callback):
"""Remove all occurences of this callbacks from the callback list."""
self._callbacks = [x for x in self._callbacks if x[0] is not callback]
2005-01-19 14:14:38 +01:00
def __str__(self):
return repr(self())
def serialize(self):
return encoder(str(self))[0].decode()
2005-01-19 14:14:38 +01:00
# We tried many, *many* different syntactic methods here, and this one was
# simply the best -- not very intrusive, easily overridden by subclasses,
# etc.
def __call__(self):
if _lastModified > self._lastModified:
if self._name in _cache:
self.set(_cache[self._name])
return self.value
class Boolean(Value):
"""Value must be either True or False (or On or Off)."""
2018-09-25 18:59:46 +02:00
__slots__ = ()
errormsg = _('Value must be either True or False (or On or Off), not %r.')
2005-01-19 14:14:38 +01:00
def set(self, s):
try:
v = utils.str.toBool(s)
2005-01-19 14:14:38 +01:00
except ValueError:
if s.strip().lower() == 'toggle':
v = not self.value
else:
self.error(s)
2005-01-19 14:14:38 +01:00
self.setValue(v)
def setValue(self, v):
super(Boolean, self).setValue(bool(v))
class Integer(Value):
"""Value must be an integer."""
2018-09-25 18:59:46 +02:00
__slots__ = ()
errormsg = _('Value must be an integer, not %r.')
2005-01-19 14:14:38 +01:00
def set(self, s):
try:
self.setValue(int(s))
except ValueError:
self.error(s)
2005-01-19 14:14:38 +01:00
class NonNegativeInteger(Integer):
"""Value must be a non-negative integer."""
2018-09-25 18:59:46 +02:00
__slots__ = ()
errormsg = _('Value must be a non-negative integer, not %r.')
2005-01-19 14:14:38 +01:00
def setValue(self, v):
if v < 0:
self.error(v)
2005-01-19 14:14:38 +01:00
super(NonNegativeInteger, self).setValue(v)
class PositiveInteger(NonNegativeInteger):
"""Value must be positive (non-zero) integer."""
2018-09-25 18:59:46 +02:00
__slots__ = ()
errormsg = _('Value must be positive (non-zero) integer, not %r.')
2005-01-19 14:14:38 +01:00
def setValue(self, v):
if not v:
self.error(v)
2005-01-19 14:14:38 +01:00
super(PositiveInteger, self).setValue(v)
class Float(Value):
"""Value must be a floating-point number."""
2018-09-25 18:59:46 +02:00
__slots__ = ()
errormsg = _('Value must be a floating-point number, not %r.')
2005-01-19 14:14:38 +01:00
def set(self, s):
try:
self.setValue(float(s))
except ValueError:
self.error(s)
2005-01-19 14:14:38 +01:00
def setValue(self, v):
try:
super(Float, self).setValue(float(v))
except ValueError:
self.error(v)
2005-01-19 14:14:38 +01:00
class PositiveFloat(Float):
"""Value must be a floating-point number greater than zero."""
2018-09-25 18:59:46 +02:00
__slots__ = ()
errormsg = _('Value must be a floating-point number greater than zero, '
'not %r.')
2005-01-19 14:14:38 +01:00
def setValue(self, v):
if v <= 0:
self.error(v)
2005-01-19 14:14:38 +01:00
else:
super(PositiveFloat, self).setValue(v)
class Probability(Float):
"""Value must be a floating point number in the range [0, 1]."""
2018-09-25 18:59:46 +02:00
__slots__ = ('__parent',)
errormsg = _('Value must be a floating point number in the range [0, 1], '
'not %r.')
2005-01-19 14:14:38 +01:00
def __init__(self, *args, **kwargs):
self.__parent = super(Probability, self)
self.__parent.__init__(*args, **kwargs)
def setValue(self, v):
if 0 <= v <= 1:
self.__parent.setValue(v)
else:
self.error(v)
2005-01-19 14:14:38 +01:00
class String(Value):
"""Value is not a valid Python string."""
2018-09-25 18:59:46 +02:00
__slots__ = ()
2017-01-11 00:10:46 +01:00
errormsg = _('Value should be a valid Python string, not %r.')
2005-01-19 14:14:38 +01:00
def set(self, s):
v = s
if not v:
v = '""'
elif v[0] != v[-1] or v[0] not in '\'"':
v = repr(v)
2005-01-19 14:14:38 +01:00
try:
v = utils.safeEval(v)
2015-08-10 20:24:11 +02:00
if not isinstance(v, minisix.string_types):
2005-01-19 14:14:38 +01:00
raise ValueError
self.setValue(v)
except ValueError: # This catches utils.safeEval(s) errors too.
self.error(s)
2005-01-19 14:14:38 +01:00
_printable = string.printable[:-4]
def _needsQuoting(self, s):
2012-08-05 09:54:34 +02:00
return any([x not in self._printable for x in s]) and s.strip() != s
2005-01-19 14:14:38 +01:00
def __str__(self):
s = self.value
if self._needsQuoting(s):
s = repr(s)
return s
class OnlySomeStrings(String):
2018-09-25 18:59:46 +02:00
__slots__ = ('__parent', '__dict__') # unfortunately, __dict__ is needed
# to set __doc__.
2005-01-19 14:14:38 +01:00
validStrings = ()
def __init__(self, *args, **kwargs):
assert self.validStrings, 'There must be some valid strings. ' \
'This is a bug.'
self.__parent = super(OnlySomeStrings, self)
self.__parent.__init__(*args, **kwargs)
self.__doc__ = format(_('Valid values include %L.'),
list(map(repr, self.validStrings)))
self.errormsg = format(_('Valid values include %L, not %%r.'),
list(map(repr, self.validStrings)))
2005-01-19 14:14:38 +01:00
def help(self):
strings = [s for s in self.validStrings if s]
2005-01-31 15:52:27 +01:00
return format('%s Valid strings: %L.', self._help, strings)
2005-01-19 14:14:38 +01:00
def normalize(self, s):
lowered = s.lower()
L = list(map(str.lower, self.validStrings))
try:
i = L.index(lowered)
except ValueError:
return s # This is handled in setValue.
return self.validStrings[i]
def setValue(self, s):
v = self.normalize(s)
2005-01-19 14:14:38 +01:00
if s in self.validStrings:
self.__parent.setValue(v)
2005-01-19 14:14:38 +01:00
else:
self.error(v)
2005-01-19 14:14:38 +01:00
class NormalizedString(String):
2018-09-25 18:59:46 +02:00
__slots__ = ('__parent')
2005-01-19 14:14:38 +01:00
def __init__(self, default, *args, **kwargs):
default = self.normalize(default)
self.__parent = super(NormalizedString, self)
self.__parent.__init__(default, *args, **kwargs)
self._showDefault = False
def normalize(self, s):
return utils.str.normalizeWhitespace(s.strip())
2005-01-19 14:14:38 +01:00
def set(self, s):
s = self.normalize(s)
self.__parent.set(s)
def setValue(self, s):
s = self.normalize(s)
self.__parent.setValue(s)
def serialize(self):
s = self.__parent.serialize()
2005-01-19 14:14:38 +01:00
prefixLen = len(self._name) + 2
lines = textwrap.wrap(s, width=76-prefixLen)
last = len(lines)-1
for (i, line) in enumerate(lines):
if i != 0:
line = ' '*prefixLen + line
if i != last:
line += '\\'
lines[i] = line
ret = os.linesep.join(lines)
return ret
class StringSurroundedBySpaces(String):
2018-09-25 18:59:46 +02:00
__slots__ = ()
2005-01-19 14:14:38 +01:00
def setValue(self, v):
if v and v.lstrip() == v:
2005-01-19 14:14:38 +01:00
v= ' ' + v
if v.rstrip() == v:
v += ' '
super(StringSurroundedBySpaces, self).setValue(v)
class StringWithSpaceOnRight(String):
2018-09-25 18:59:46 +02:00
__slots__ = ()
2005-01-19 14:14:38 +01:00
def setValue(self, v):
if v and v.rstrip() == v:
2005-01-19 14:14:38 +01:00
v += ' '
super(StringWithSpaceOnRight, self).setValue(v)
class Regexp(Value):
"""Value must be a valid regular expression."""
2018-09-25 18:59:46 +02:00
__slots__ = ('sr', 'value', '__parent')
errormsg = _('Value must be a valid regular expression, not %r.')
2005-01-19 14:14:38 +01:00
def __init__(self, *args, **kwargs):
kwargs['setDefault'] = False
self.sr = ''
self.value = None
self.__parent = super(Regexp, self)
self.__parent.__init__(*args, **kwargs)
def error(self, e):
s = 'Value must be a regexp of the form m/.../ or /.../. %s' % e
e = InvalidRegistryValue(s)
e.value = self
raise e
2005-01-19 14:14:38 +01:00
def set(self, s):
try:
if s:
self.setValue(utils.str.perlReToPythonRe(s), sr=s)
2005-01-19 14:14:38 +01:00
else:
self.setValue(None)
except ValueError as e:
2005-01-19 14:14:38 +01:00
self.error(e)
def setValue(self, v, sr=None):
if v is None:
self.sr = ''
self.__parent.setValue(None)
2005-01-19 14:14:38 +01:00
elif sr is not None:
self.sr = sr
self.__parent.setValue(v)
2005-01-19 14:14:38 +01:00
else:
raise InvalidRegistryValue('Can\'t setValue a regexp, there would be an inconsistency '\
'between the regexp and the recorded string value.')
2005-01-19 14:14:38 +01:00
def __str__(self):
self() # Gotta update if we've been reloaded.
return self.sr
class SeparatedListOf(Value):
2018-09-25 18:59:46 +02:00
__slots__ = ()
2005-01-19 14:14:38 +01:00
List = list
Value = Value
sorted = False
def splitter(self, s):
"""Override this with a function that takes a string and returns a list
of strings."""
raise NotImplementedError
def joiner(self, L):
"""Override this to join the internal list for output."""
raise NotImplementedError
def set(self, s):
L = self.splitter(s)
for (i, s) in enumerate(L):
v = self.Value(s, '')
L[i] = v()
self.setValue(L)
def setValue(self, v):
super(SeparatedListOf, self).setValue(self.List(v))
def __str__(self):
values = self()
if self.sorted:
values = sorted(values)
if values:
return self.joiner(values)
else:
# We must return *something* here, otherwise down along the road we
# can run into issues showing users the value if they've disabled
# nick prefixes in any of the numerous ways possible. Since the
# config parser doesn't care about this space, we'll use it :)
return ' '
class SpaceSeparatedListOf(SeparatedListOf):
2018-09-25 18:59:46 +02:00
__slots__ = ()
2005-01-19 14:14:38 +01:00
def splitter(self, s):
return s.split()
joiner = ' '.join
class SpaceSeparatedListOfStrings(SpaceSeparatedListOf):
2018-09-25 18:59:46 +02:00
__slots__ = ()
2005-01-19 14:14:38 +01:00
Value = String
class SpaceSeparatedSetOfStrings(SpaceSeparatedListOfStrings):
2018-09-25 18:59:46 +02:00
__slots__ = ()
2005-02-04 04:11:02 +01:00
List = set
2005-01-19 14:14:38 +01:00
class CommaSeparatedListOfStrings(SeparatedListOf):
2018-09-25 18:59:46 +02:00
__slots__ = ()
2005-01-19 14:14:38 +01:00
Value = String
def splitter(self, s):
return re.split(r'\s*,\s*', s)
joiner = ', '.join
class CommaSeparatedSetOfStrings(SeparatedListOf):
2018-09-25 18:59:46 +02:00
__slots__ = ()
List = set
Value = String
def splitter(self, s):
return re.split(r'\s*,\s*', s)
joiner = ', '.join
class TemplatedString(String):
2018-09-25 18:59:46 +02:00
__slots__ = ()
requiredTemplates = []
def __init__(self, *args, **kwargs):
assert self.requiredTemplates, \
'There must be some templates. This is a bug.'
self.__parent = super(String, self)
self.__parent.__init__(*args, **kwargs)
2005-01-19 14:14:38 +01:00
def setValue(self, v):
def hasTemplate(s):
return re.search(r'\$%s\b|\${%s}' % (s, s), v) is not None
if utils.iter.all(hasTemplate, self.requiredTemplates):
self.__parent.setValue(v)
else:
self.error(v)
2005-01-19 14:14:38 +01:00
2012-12-08 20:59:36 +01:00
class Json(String):
2018-09-25 18:59:46 +02:00
__slots__ = ()
2012-12-08 20:59:36 +01:00
# Json-serializable data
2012-12-15 22:00:23 +01:00
def set(self, v):
self.setValue(json.loads(v))
2012-12-08 20:59:36 +01:00
def setValue(self, v):
super(Json, self).setValue(json.dumps(v))
def __call__(self):
return json.loads(super(Json, self).__call__())
class _Context:
def __init__(self, var):
self._var = var
def __enter__(self):
self._dict = self._var()
return self._dict
def __exit__(self, *args):
self._var.setValue(self._dict)
def editable(self):
"""Return an editable dict usable within a 'with' statement and
committed to the configuration variable at the end."""
return self._Context(self)
# vim:set shiftwidth=4 softtabstop=4 expandtab textwidth=79: