complaints by pylint are mainly related to - remove object from base list in class definitions - unnecessary else/elif after return/raise Change-Id: I13d15449149cc8bba0562338d0c9c42e97163bdf Reviewed-on: https://forge.frm2.tum.de/review/c/sine2020/secop/playground/+/21325 Tested-by: JenkinsCodeReview <bjoern_pedersen@frm2.tum.de> Reviewed-by: Markus Zolliker <markus.zolliker@psi.ch>
921 lines
33 KiB
Python
921 lines
33 KiB
Python
# -*- coding: utf-8 -*-
|
|
# *****************************************************************************
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify it under
|
|
# the terms of the GNU General Public License as published by the Free Software
|
|
# Foundation; either version 2 of the License, or (at your option) any later
|
|
# version.
|
|
#
|
|
# 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.,
|
|
# 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
#
|
|
# Module authors:
|
|
# Enrico Faulhaber <enrico.faulhaber@frm2.tum.de>
|
|
#
|
|
# *****************************************************************************
|
|
"""Define validated data types."""
|
|
|
|
# pylint: disable=abstract-method
|
|
|
|
|
|
from base64 import b64decode, b64encode
|
|
|
|
from secop.errors import ProgrammingError, ProtocolError, BadValueError
|
|
from secop.lib.enum import Enum
|
|
from secop.parse import Parser
|
|
|
|
|
|
# Only export these classes for 'from secop.datatypes import *'
|
|
__all__ = [
|
|
'DataType',
|
|
'FloatRange', 'IntRange',
|
|
'BoolType', 'EnumType',
|
|
'BLOBType', 'StringType',
|
|
'TupleOf', 'ArrayOf', 'StructOf',
|
|
'CommandType',
|
|
]
|
|
|
|
# *DEFAULT* limits for IntRange/ScaledIntegers transport serialisation
|
|
DEFAULT_MIN_INT = -16777216
|
|
DEFAULT_MAX_INT = 16777216
|
|
|
|
Parser = Parser()
|
|
|
|
# base class for all DataTypes
|
|
class DataType:
|
|
IS_COMMAND = False
|
|
unit = ''
|
|
fmtstr = '%r'
|
|
default = None
|
|
|
|
def __call__(self, value):
|
|
"""check if given value (a python obj) is valid for this datatype
|
|
|
|
returns the value or raises an appropriate exception"""
|
|
raise NotImplementedError
|
|
|
|
def from_string(self, text):
|
|
"""interprets a given string and returns a validated (internal) value"""
|
|
# to evaluate values from configfiles, ui, etc...
|
|
raise NotImplementedError
|
|
|
|
def export_datatype(self):
|
|
"""return a python object which after jsonifying identifies this datatype"""
|
|
raise NotImplementedError
|
|
|
|
def export_value(self, value):
|
|
"""if needed, reformat value for transport"""
|
|
return value
|
|
|
|
def import_value(self, value):
|
|
"""opposite of export_value, reformat from transport to internal repr
|
|
|
|
note: for importing from gui/configfile/commandline use :meth:`from_string`
|
|
instead.
|
|
"""
|
|
return value
|
|
|
|
def format_value(self, value, unit=None):
|
|
"""format a value of this type into a str string
|
|
|
|
This is intended for 'nice' formatting for humans and is NOT
|
|
the opposite of :meth:`from_string`
|
|
if unit is given, use it, else use the unit of the datatype (if any)"""
|
|
raise NotImplementedError
|
|
|
|
def set_prop(self, key, value, default, func=lambda x:x):
|
|
"""set an optional datatype property and store the default"""
|
|
self._defaults[key] = default
|
|
if value is None:
|
|
value = default
|
|
setattr(self, key, func(value))
|
|
|
|
def get_info(self, **kwds):
|
|
"""prepare dict for export or repr
|
|
|
|
get a dict with all items different from default
|
|
plus mandatory keys from kwds"""
|
|
for k,v in self._defaults.items():
|
|
value = getattr(self, k)
|
|
if value != v:
|
|
kwds[k] = value
|
|
return kwds
|
|
|
|
def copy(self):
|
|
"""make a deep copy of the datatype"""
|
|
|
|
# looks like the simplest way to make a deep copy
|
|
return get_datatype(self.export_datatype())
|
|
|
|
|
|
class FloatRange(DataType):
|
|
"""Restricted float type"""
|
|
|
|
def __init__(self, minval=None, maxval=None, unit=None, fmtstr=None,
|
|
absolute_resolution=None, relative_resolution=None,):
|
|
self._defaults = {}
|
|
self.set_prop('min', minval, float('-inf'), float)
|
|
self.set_prop('max', maxval, float('+inf'), float)
|
|
self.set_prop('unit', unit, '', str)
|
|
self.set_prop('fmtstr', fmtstr, '%g', str)
|
|
self.set_prop('absolute_resolution', absolute_resolution, 0.0, float)
|
|
self.set_prop('relative_resolution', relative_resolution, 1.2e-7, float)
|
|
self.default = 0 if self.min <= 0 <= self.max else self.min
|
|
|
|
# check values
|
|
if self.min > self.max:
|
|
raise BadValueError('max must be larger then min!')
|
|
if '%' not in self.fmtstr:
|
|
raise BadValueError('Invalid fmtstr!')
|
|
if self.absolute_resolution < 0:
|
|
raise BadValueError('absolute_resolution MUST be >=0')
|
|
if self.relative_resolution < 0:
|
|
raise BadValueError('relative_resolution MUST be >=0')
|
|
|
|
def export_datatype(self):
|
|
return self.get_info(type='double')
|
|
|
|
def __call__(self, value):
|
|
try:
|
|
value = float(value)
|
|
except Exception:
|
|
raise BadValueError('Can not __call__ %r to float' % value)
|
|
prec = max(abs(value * self.relative_resolution), self.absolute_resolution)
|
|
if self.min - prec <= value <= self.max + prec:
|
|
return min(max(value, self.min), self.max)
|
|
raise BadValueError('%.14g should be a float between %.14g and %.14g' %
|
|
(value, self.min, self.max))
|
|
|
|
def __repr__(self):
|
|
hints = self.get_info()
|
|
if 'min' in hints:
|
|
hints['minval'] = hints.pop('min')
|
|
if 'max' in hints:
|
|
hints['maxval'] = hints.pop('max')
|
|
return 'FloatRange(%s)' % (', '.join('%s=%r' % (k,v) for k,v in hints.items()))
|
|
|
|
def export_value(self, value):
|
|
"""returns a python object fit for serialisation"""
|
|
return float(value)
|
|
|
|
def import_value(self, value):
|
|
"""returns a python object from serialisation"""
|
|
return float(value)
|
|
|
|
def from_string(self, text):
|
|
value = float(text)
|
|
return self(value)
|
|
|
|
def format_value(self, value, unit=None):
|
|
if unit is None:
|
|
unit = self.unit
|
|
if unit:
|
|
return ' '.join([self.fmtstr % value, unit])
|
|
return self.fmtstr % value
|
|
|
|
|
|
class IntRange(DataType):
|
|
"""Restricted int type"""
|
|
|
|
def __init__(self, minval=None, maxval=None):
|
|
self.min = DEFAULT_MIN_INT if minval is None else int(minval)
|
|
self.max = DEFAULT_MAX_INT if maxval is None else int(maxval)
|
|
self.default = 0 if self.min <= 0 <= self.max else self.min
|
|
# a unit on an int is now allowed in SECoP, but do we need them in Frappy?
|
|
# self.set_prop('unit', unit, '', str)
|
|
|
|
# check values
|
|
if self.min > self.max:
|
|
raise BadValueError('Max must be larger then min!')
|
|
|
|
def export_datatype(self):
|
|
return dict(type='int', min=self.min, max=self.max)
|
|
|
|
def __call__(self, value):
|
|
try:
|
|
value = int(value)
|
|
if value < self.min:
|
|
raise BadValueError('%r should be an int between %d and %d' %
|
|
(value, self.min, self.max or 0))
|
|
if value > self.max:
|
|
raise BadValueError('%r should be an int between %d and %d' %
|
|
(value, self.min or 0, self.max))
|
|
return value
|
|
except Exception:
|
|
raise BadValueError('Can not convert %r to int' % value)
|
|
|
|
def __repr__(self):
|
|
return 'IntRange(%d, %d)' % (self.min, self.max)
|
|
|
|
def export_value(self, value):
|
|
"""returns a python object fit for serialisation"""
|
|
return int(value)
|
|
|
|
def import_value(self, value):
|
|
"""returns a python object from serialisation"""
|
|
return int(value)
|
|
|
|
def from_string(self, text):
|
|
value = int(text)
|
|
return self(value)
|
|
|
|
def format_value(self, value, unit=None):
|
|
return '%d' % value
|
|
|
|
|
|
class ScaledInteger(DataType):
|
|
"""Scaled integer int type
|
|
|
|
note: limits are for the scaled value (i.e. the internal value)
|
|
the scale is only used for calculating to/from transport serialisation"""
|
|
|
|
def __init__(self, scale, minval=None, maxval=None, unit=None, fmtstr=None,
|
|
absolute_resolution=None, relative_resolution=None,):
|
|
self._defaults = {}
|
|
self.scale = float(scale)
|
|
if not self.scale > 0:
|
|
raise BadValueError('Scale MUST be positive!')
|
|
self.set_prop('unit', unit, '', str)
|
|
self.set_prop('fmtstr', fmtstr, '%g', str)
|
|
self.set_prop('absolute_resolution', absolute_resolution, self.scale, float)
|
|
self.set_prop('relative_resolution', relative_resolution, 1.2e-7, float)
|
|
|
|
self.min = DEFAULT_MIN_INT * self.scale if minval is None else float(minval)
|
|
self.max = DEFAULT_MAX_INT * self.scale if maxval is None else float(maxval)
|
|
self.default = 0 if self.min <= 0 <= self.max else self.min
|
|
|
|
# check values
|
|
if self.min > self.max:
|
|
raise BadValueError('Max must be larger then min!')
|
|
if '%' not in self.fmtstr:
|
|
raise BadValueError('Invalid fmtstr!')
|
|
if self.absolute_resolution < 0:
|
|
raise BadValueError('absolute_resolution MUST be >=0')
|
|
if self.relative_resolution < 0:
|
|
raise BadValueError('relative_resolution MUST be >=0')
|
|
# Remark: Datatype.copy() will round min, max to a multiple of self.scale
|
|
# this should be o.k.
|
|
|
|
def export_datatype(self):
|
|
return self.get_info(type='scaled', scale=self.scale,
|
|
min = int((self.min + self.scale * 0.5) // self.scale),
|
|
max = int((self.max + self.scale * 0.5) // self.scale))
|
|
|
|
def __call__(self, value):
|
|
try:
|
|
value = float(value)
|
|
except Exception:
|
|
raise BadValueError('Can not convert %r to float' % value)
|
|
prec = max(self.scale, abs(value * self.relative_resolution),
|
|
self.absolute_resolution)
|
|
if self.min - prec <= value <= self.max + prec:
|
|
value = min(max(value, self.min), self.max)
|
|
else:
|
|
raise BadValueError('%g should be a float between %g and %g' %
|
|
(value, self.min, self.max))
|
|
intval = int((value + self.scale * 0.5) // self.scale)
|
|
value = float(intval * self.scale)
|
|
return value # return 'actual' value (which is more discrete than a float)
|
|
|
|
def __repr__(self):
|
|
hints = self.get_info(scale='%g' % self.scale,
|
|
min = int((self.min + self.scale * 0.5) // self.scale),
|
|
max = int((self.max + self.scale * 0.5) // self.scale))
|
|
return 'ScaledInteger(%s)' % (', '.join('%s=%r' % kv for kv in hints.items()))
|
|
|
|
def export_value(self, value):
|
|
"""returns a python object fit for serialisation"""
|
|
# note: round behaves different in Py2 vs. Py3, so use floor division
|
|
return int((value + self.scale * 0.5) // self.scale)
|
|
|
|
def import_value(self, value):
|
|
"""returns a python object from serialisation"""
|
|
return self.scale * int(value)
|
|
|
|
def from_string(self, text):
|
|
value = float(text)
|
|
return self(value)
|
|
|
|
def format_value(self, value, unit=None):
|
|
if unit is None:
|
|
unit = self.unit
|
|
if unit:
|
|
return ' '.join([self.fmtstr % value, unit])
|
|
return self.fmtstr % value
|
|
|
|
|
|
class EnumType(DataType):
|
|
|
|
def __init__(self, enum_or_name='', **kwds):
|
|
if 'members' in kwds:
|
|
kwds = dict(kwds)
|
|
kwds.update(kwds['members'])
|
|
kwds.pop('members')
|
|
self._enum = Enum(enum_or_name, **kwds)
|
|
self.default = self._enum[self._enum.members[0]]
|
|
|
|
def copy(self):
|
|
# as the name is not exported, we have to implement copy ourselfs
|
|
return EnumType(self._enum)
|
|
|
|
def export_datatype(self):
|
|
return {'type': 'enum', 'members':dict((m.name, m.value) for m in self._enum.members)}
|
|
|
|
def __repr__(self):
|
|
return u"EnumType(%r, %s)" % (self._enum.name, ', '.join('%s=%d' %(m.name, m.value) for m in self._enum.members))
|
|
|
|
def export_value(self, value):
|
|
"""returns a python object fit for serialisation"""
|
|
return int(self(value))
|
|
|
|
def import_value(self, value):
|
|
"""returns a python object from serialisation"""
|
|
return self(value)
|
|
|
|
def __call__(self, value):
|
|
"""return the validated (internal) value or raise"""
|
|
try:
|
|
return self._enum[value]
|
|
except (KeyError, TypeError): # TypeError will be raised when value is not hashable
|
|
raise BadValueError('%r is not a member of enum %r' % (value, self._enum))
|
|
|
|
def from_string(self, text):
|
|
return self(text)
|
|
|
|
def format_value(self, value, unit=None):
|
|
return '%s<%s>' % (self._enum[value].name, self._enum[value].value)
|
|
|
|
|
|
class BLOBType(DataType):
|
|
minbytes = 0
|
|
maxbytes = None
|
|
|
|
def __init__(self, minbytes=0, maxbytes=None):
|
|
# if only one argument is given, use exactly that many bytes
|
|
# if nothing is given, default to 255
|
|
if maxbytes is None:
|
|
maxbytes = minbytes or 255
|
|
self._defaults = {}
|
|
self.set_prop('minbytes', minbytes, 0, int)
|
|
self.maxbytes = int(maxbytes)
|
|
if self.minbytes < 0:
|
|
raise BadValueError('sizes must be bigger than or equal to 0!')
|
|
if self.minbytes > self.maxbytes:
|
|
raise BadValueError('maxbytes must be bigger than or equal to minbytes!')
|
|
self.default = b'\0' * self.minbytes
|
|
|
|
def export_datatype(self):
|
|
return self.get_info(type='blob', maxbytes=self.maxbytes)
|
|
|
|
def __repr__(self):
|
|
return 'BLOBType(%d, %d)' % (self.minbytes, self.maxbytes)
|
|
|
|
def __call__(self, value):
|
|
"""return the validated (internal) value or raise"""
|
|
if not isinstance(value, bytes):
|
|
raise BadValueError('%r has the wrong type!' % value)
|
|
size = len(value)
|
|
if size < self.minbytes:
|
|
raise BadValueError(
|
|
'%r must be at least %d bytes long!' % (value, self.minbytes))
|
|
if size > self.maxbytes:
|
|
raise BadValueError(
|
|
'%r must be at most %d bytes long!' % (value, self.maxbytes))
|
|
return value
|
|
|
|
def export_value(self, value):
|
|
"""returns a python object fit for serialisation"""
|
|
return b64encode(value).decode('ascii')
|
|
|
|
def import_value(self, value):
|
|
"""returns a python object from serialisation"""
|
|
return b64decode(value)
|
|
|
|
def from_string(self, text):
|
|
value = text
|
|
# XXX: what should we do here?
|
|
return self(value)
|
|
|
|
def format_value(self, value, unit=None):
|
|
return repr(value)
|
|
|
|
|
|
class StringType(DataType):
|
|
MAXCHARS = 0xffffffff
|
|
|
|
def __init__(self, minchars=0, maxchars=None, isUTF8=False):
|
|
if maxchars is None:
|
|
maxchars = minchars or self.MAXCHARS
|
|
self._defaults = {}
|
|
self.set_prop('minchars', minchars, 0, int)
|
|
self.set_prop('maxchars', maxchars, self.MAXCHARS, int)
|
|
self.set_prop('isUTF8', isUTF8, False, bool)
|
|
if self.minchars < 0:
|
|
raise BadValueError('sizes must be bigger than or equal to 0!')
|
|
if self.minchars > self.maxchars:
|
|
raise BadValueError('maxchars must be bigger than or equal to minchars!')
|
|
self.default = ' ' * self.minchars
|
|
|
|
def export_datatype(self):
|
|
return self.get_info(type='string')
|
|
|
|
def __repr__(self):
|
|
return 'StringType(%s)' % (', '.join('%s=%r' % kv for kv in self.get_info().items()))
|
|
|
|
def __call__(self, value):
|
|
"""return the validated (internal) value or raise"""
|
|
if not isinstance(value, str):
|
|
raise BadValueError('%r has the wrong type!' % value)
|
|
if not self.isUTF8:
|
|
try:
|
|
value.encode('ascii')
|
|
except UnicodeEncodeError:
|
|
raise BadValueError('%r contains non-ascii character!' % value)
|
|
size = len(value)
|
|
if size < self.minchars:
|
|
raise BadValueError(
|
|
'%r must be at least %d bytes long!' % (value, self.minchars))
|
|
if size > self.maxchars:
|
|
raise BadValueError(
|
|
'%r must be at most %d bytes long!' % (value, self.maxchars))
|
|
if '\0' in value:
|
|
raise BadValueError(
|
|
'Strings are not allowed to embed a \\0! Use a Blob instead!')
|
|
return value
|
|
|
|
def export_value(self, value):
|
|
"""returns a python object fit for serialisation"""
|
|
return '%s' % value
|
|
|
|
def import_value(self, value):
|
|
"""returns a python object from serialisation"""
|
|
# XXX: do we keep it as str str, or convert it to something else? (UTF-8 maybe?)
|
|
return str(value)
|
|
|
|
def from_string(self, text):
|
|
value = str(text)
|
|
return self(value)
|
|
|
|
def format_value(self, value, unit=None):
|
|
return repr(value)
|
|
|
|
|
|
# TextType is a special StringType intended for longer texts (i.e. embedding \n),
|
|
# whereas StringType is supposed to not contain '\n'
|
|
# unfortunately, SECoP makes no distinction here....
|
|
# note: content is supposed to follow the format of a git commit message, i.e. a line of text, 2 '\n' + a longer explanation
|
|
class TextType(StringType):
|
|
def __init__(self, maxchars=None):
|
|
if maxchars is None:
|
|
maxchars = self.MAXCHARS
|
|
super(TextType, self).__init__(0, maxchars)
|
|
|
|
def __repr__(self):
|
|
return 'TextType(%d, %d)' % (self.minchars, self.maxchars)
|
|
|
|
def copy(self):
|
|
# DataType.copy will not work, because it is exported as 'string'
|
|
return TextType(self.maxchars)
|
|
|
|
|
|
# Bool is a special enum
|
|
class BoolType(DataType):
|
|
default = False
|
|
|
|
def export_datatype(self):
|
|
return {'type': 'bool'}
|
|
|
|
def __repr__(self):
|
|
return 'BoolType()'
|
|
|
|
def __call__(self, value):
|
|
"""return the validated (internal) value or raise"""
|
|
if value in [0, '0', 'False', 'false', 'no', 'off', False]:
|
|
return False
|
|
if value in [1, '1', 'True', 'true', 'yes', 'on', True]:
|
|
return True
|
|
raise BadValueError('%r is not a boolean value!' % value)
|
|
|
|
def export_value(self, value):
|
|
"""returns a python object fit for serialisation"""
|
|
return bool(self(value))
|
|
|
|
def import_value(self, value):
|
|
"""returns a python object from serialisation"""
|
|
return self(value)
|
|
|
|
def from_string(self, text):
|
|
value = text
|
|
return self(value)
|
|
|
|
|
|
def format_value(self, value, unit=None):
|
|
return repr(bool(value))
|
|
|
|
#
|
|
# nested types
|
|
#
|
|
|
|
|
|
class ArrayOf(DataType):
|
|
minlen = None
|
|
maxlen = None
|
|
members = None
|
|
|
|
def __init__(self, members, minlen=0, maxlen=None, unit=None):
|
|
if not isinstance(members, DataType):
|
|
raise BadValueError(
|
|
'ArrayOf only works with a DataType as first argument!')
|
|
# one argument -> exactly that size
|
|
# argument default to 100
|
|
if maxlen is None:
|
|
maxlen = minlen or 100
|
|
self.members = members
|
|
if unit:
|
|
self.members.unit = unit
|
|
|
|
self.minlen = int(minlen)
|
|
self.maxlen = int(maxlen)
|
|
if self.minlen < 0:
|
|
raise BadValueError('sizes must be > 0')
|
|
if self.maxlen < 1:
|
|
raise BadValueError('Maximum size must be >= 1!')
|
|
if self.minlen > self.maxlen:
|
|
raise BadValueError('maxlen must be bigger than or equal to minlen!')
|
|
self.default = [members.default] * self.minlen
|
|
|
|
def export_datatype(self):
|
|
return dict(type='array', minlen=self.minlen, maxlen=self.maxlen,
|
|
members=self.members.export_datatype())
|
|
|
|
def __repr__(self):
|
|
return 'ArrayOf(%s, %s, %s)' % (
|
|
repr(self.members), self.minlen, self.maxlen)
|
|
|
|
def __call__(self, value):
|
|
"""validate an external representation to an internal one"""
|
|
if isinstance(value, (tuple, list)):
|
|
# check number of elements
|
|
if self.minlen is not None and len(value) < self.minlen:
|
|
raise BadValueError(
|
|
'Array too small, needs at least %d elements!' %
|
|
self.minlen)
|
|
if self.maxlen is not None and len(value) > self.maxlen:
|
|
raise BadValueError(
|
|
'Array too big, holds at most %d elements!' % self.minlen)
|
|
# apply subtype valiation to all elements and return as list
|
|
return [self.members(elem) for elem in value]
|
|
raise BadValueError(
|
|
'Can not convert %s to ArrayOf DataType!' % repr(value))
|
|
|
|
def export_value(self, value):
|
|
"""returns a python object fit for serialisation"""
|
|
return [self.members.export_value(elem) for elem in value]
|
|
|
|
def import_value(self, value):
|
|
"""returns a python object from serialisation"""
|
|
return [self.members.import_value(elem) for elem in value]
|
|
|
|
def from_string(self, text):
|
|
value, rem = Parser.parse(text)
|
|
if rem:
|
|
raise ProtocolError('trailing garbage: %r' % rem)
|
|
return self(value)
|
|
|
|
def format_value(self, value, unit=None):
|
|
if unit is None:
|
|
unit = self.unit or self.members.unit
|
|
res = '[%s]' % (', '.join([self.members.format_value(elem, '') for elem in value]))
|
|
if unit:
|
|
return ' '.join([res, unit])
|
|
return res
|
|
|
|
|
|
class TupleOf(DataType):
|
|
|
|
def __init__(self, *members):
|
|
if not members:
|
|
raise BadValueError('Empty tuples are not allowed!')
|
|
for subtype in members:
|
|
if not isinstance(subtype, DataType):
|
|
raise BadValueError(
|
|
'TupleOf only works with DataType objs as arguments!')
|
|
self.members = members
|
|
self.default = tuple(el.default for el in members)
|
|
|
|
def export_datatype(self):
|
|
return dict(type='tuple', members=[subtype.export_datatype() for subtype in self.members])
|
|
|
|
def __repr__(self):
|
|
return 'TupleOf(%s)' % ', '.join([repr(st) for st in self.members])
|
|
|
|
def __call__(self, value):
|
|
"""return the validated value or raise"""
|
|
# keep the ordering!
|
|
try:
|
|
if len(value) != len(self.members):
|
|
raise BadValueError(
|
|
'Illegal number of Arguments! Need %d arguments.' %
|
|
(len(self.members)))
|
|
# validate elements and return as list
|
|
return [sub(elem)
|
|
for sub, elem in zip(self.members, value)]
|
|
except Exception as exc:
|
|
raise BadValueError('Can not validate:', str(exc))
|
|
|
|
def export_value(self, value):
|
|
"""returns a python object fit for serialisation"""
|
|
return [sub.export_value(elem) for sub, elem in zip(self.members, value)]
|
|
|
|
def import_value(self, value):
|
|
"""returns a python object from serialisation"""
|
|
return [sub.import_value(elem) for sub, elem in zip(self.members, value)]
|
|
|
|
def from_string(self, text):
|
|
value, rem = Parser.parse(text)
|
|
if rem:
|
|
raise ProtocolError('trailing garbage: %r' % rem)
|
|
return self(value)
|
|
|
|
def format_value(self, value, unit=None):
|
|
return '(%s)' % (', '.join([sub.format_value(elem)
|
|
for sub, elem in zip(self.members, value)]))
|
|
|
|
|
|
class StructOf(DataType):
|
|
|
|
def __init__(self, optional=None, **members):
|
|
self.members = members
|
|
if not members:
|
|
raise BadValueError('Empty structs are not allowed!')
|
|
self.optional = list(optional or [])
|
|
for name, subtype in list(members.items()):
|
|
if not isinstance(subtype, DataType):
|
|
raise ProgrammingError(
|
|
'StructOf only works with named DataType objs as keyworded arguments!')
|
|
if not isinstance(name, str):
|
|
raise ProgrammingError(
|
|
'StructOf only works with named DataType objs as keyworded arguments!')
|
|
for name in self.optional:
|
|
if name not in members:
|
|
raise ProgrammingError(
|
|
'Only members of StructOf may be declared as optional!')
|
|
self.default = dict((k,el.default) for k, el in members.items())
|
|
|
|
def export_datatype(self):
|
|
res = dict(type='struct', members=dict((n, s.export_datatype())
|
|
for n, s in list(self.members.items())))
|
|
if self.optional:
|
|
res['optional'] = self.optional
|
|
return res
|
|
|
|
def __repr__(self):
|
|
opt = self.optional if self.optional else ''
|
|
return 'StructOf(%s%s)' % (', '.join(
|
|
['%s=%s' % (n, repr(st)) for n, st in list(self.members.items())]), opt)
|
|
|
|
def __call__(self, value):
|
|
"""return the validated value or raise"""
|
|
try:
|
|
# XXX: handle optional elements !!!
|
|
if len(list(value.keys())) != len(list(self.members.keys())):
|
|
raise BadValueError(
|
|
'Illegal number of Arguments! Need %d arguments.' %
|
|
len(list(self.members.keys())))
|
|
# validate elements and return as dict
|
|
return dict((str(k), self.members[k](v))
|
|
for k, v in list(value.items()))
|
|
except Exception as exc:
|
|
raise BadValueError('Can not validate %s: %s' % (repr(value), str(exc)))
|
|
|
|
def export_value(self, value):
|
|
"""returns a python object fit for serialisation"""
|
|
if len(list(value.keys())) != len(list(self.members.keys())):
|
|
raise BadValueError(
|
|
'Illegal number of Arguments! Need %d arguments.' % len(
|
|
list(self.members.keys())))
|
|
return dict((str(k), self.members[k].export_value(v))
|
|
for k, v in list(value.items()))
|
|
|
|
def import_value(self, value):
|
|
"""returns a python object from serialisation"""
|
|
if len(list(value.keys())) != len(list(self.members.keys())):
|
|
raise BadValueError(
|
|
'Illegal number of Arguments! Need %d arguments.' % len(
|
|
list(self.members.keys())))
|
|
return dict((str(k), self.members[k].import_value(v))
|
|
for k, v in list(value.items()))
|
|
|
|
def from_string(self, text):
|
|
value, rem = Parser.parse(text)
|
|
if rem:
|
|
raise ProtocolError('trailing garbage: %r' % rem)
|
|
return self(dict(value))
|
|
|
|
def format_value(self, value, unit=None):
|
|
return '{%s}' % (', '.join(['%s=%s' % (k, self.members[k].format_value(v)) for k, v in sorted(value.items())]))
|
|
|
|
|
|
class CommandType(DataType):
|
|
IS_COMMAND = True
|
|
argument = None
|
|
result = None
|
|
|
|
def __init__(self, argument=None, result=None):
|
|
if argument is not None:
|
|
if not isinstance(argument, DataType):
|
|
raise BadValueError('CommandType: Argument type must be a DataType!')
|
|
if result is not None:
|
|
if not isinstance(result, DataType):
|
|
raise BadValueError('CommandType: Result type must be a DataType!')
|
|
self.argument = argument
|
|
self.result = result
|
|
|
|
def export_datatype(self):
|
|
a, r = self.argument, self.result
|
|
props = {'type': 'command'}
|
|
if a is not None:
|
|
props['argument'] = a.export_datatype()
|
|
if r is not None:
|
|
props['result'] = r.export_datatype()
|
|
return props
|
|
|
|
def __repr__(self):
|
|
argstr = repr(self.argument) if self.argument else ''
|
|
if self.result is None:
|
|
return 'CommandType(%s)' % argstr
|
|
return 'CommandType(%s)->%s' % (argstr, repr(self.result))
|
|
|
|
def __call__(self, value):
|
|
"""return the validated argument value or raise"""
|
|
return self.argument(value)
|
|
|
|
def export_value(self, value):
|
|
raise ProgrammingError('values of type command can not be transported!')
|
|
|
|
def import_value(self, value):
|
|
raise ProgrammingError('values of type command can not be transported!')
|
|
|
|
def from_string(self, text):
|
|
value, rem = Parser.parse(text)
|
|
if rem:
|
|
raise ProtocolError('trailing garbage: %r' % rem)
|
|
return self(value)
|
|
|
|
def format_value(self, value, unit=None):
|
|
# actually I have no idea what to do here!
|
|
raise NotImplementedError
|
|
|
|
|
|
# internally used datatypes (i.e. only for programming the SEC-node
|
|
class DataTypeType(DataType):
|
|
def __call__(self, value):
|
|
"""check if given value (a python obj) is a valid datatype
|
|
|
|
returns the value or raises an appropriate exception"""
|
|
if isinstance(value, DataType):
|
|
return value
|
|
raise ProgrammingError('%r should be a DataType!' % value)
|
|
|
|
def export_value(self, value):
|
|
"""if needed, reformat value for transport"""
|
|
return value.export_datatype()
|
|
|
|
def import_value(self, value):
|
|
"""opposite of export_value, reformat from transport to internal repr
|
|
|
|
note: for importing from gui/configfile/commandline use :meth:`from_string`
|
|
instead.
|
|
"""
|
|
raise NotImplementedError
|
|
|
|
|
|
class ValueType(DataType):
|
|
"""validates any python value"""
|
|
def __call__(self, value):
|
|
"""check if given value (a python obj) is valid for this datatype
|
|
|
|
returns the value or raises an appropriate exception"""
|
|
return value
|
|
|
|
def export_value(self, value):
|
|
"""if needed, reformat value for transport"""
|
|
return value
|
|
|
|
def import_value(self, value):
|
|
"""opposite of export_value, reformat from transport to internal repr
|
|
|
|
note: for importing from gui/configfile/commandline use :meth:`from_string`
|
|
instead.
|
|
"""
|
|
raise NotImplementedError
|
|
|
|
class NoneOr(DataType):
|
|
"""validates a None or smth. else"""
|
|
default = None
|
|
|
|
def __init__(self, other):
|
|
self.other = other
|
|
|
|
def __call__(self, value):
|
|
return None if value is None else self.other(value)
|
|
|
|
def export_value(self, value):
|
|
if value is None:
|
|
return None
|
|
return self.other.export_value(value)
|
|
|
|
|
|
class OrType(DataType):
|
|
def __init__(self, *types):
|
|
self.types = types
|
|
self.default = self.types[0].default
|
|
|
|
def __call__(self, value):
|
|
for t in self.types:
|
|
try:
|
|
return t(value)
|
|
except Exception:
|
|
pass
|
|
raise BadValueError(u"Invalid Value, must conform to one of %s" % (', '.join((str(t) for t in self.types))))
|
|
|
|
|
|
Int8 = IntRange(-(1 << 7), (1 << 7) - 1)
|
|
Int16 = IntRange(-(1 << 15), (1 << 15) - 1)
|
|
Int32 = IntRange(-(1 << 31), (1 << 31) - 1)
|
|
Int64 = IntRange(-(1 << 63), (1 << 63) - 1)
|
|
UInt8 = IntRange(0, (1 << 8) - 1)
|
|
UInt16 = IntRange(0, (1 << 16) - 1)
|
|
UInt32 = IntRange(0, (1 << 32) - 1)
|
|
UInt64 = IntRange(0, (1 << 64) - 1)
|
|
|
|
|
|
# Goodie: Convenience Datatypes for Programming
|
|
class LimitsType(StructOf):
|
|
def __init__(self, _min=None, _max=None):
|
|
StructOf.__init__(self, min=FloatRange(_min,_max), max=FloatRange(_min, _max))
|
|
|
|
def __call__(self, value):
|
|
limits = StructOf.__call__(self, value)
|
|
if limits['max'] < limits['min']:
|
|
raise BadValueError('Maximum Value %s must be greater than minimum value %s!' % (limits['max'], limits['min']))
|
|
return limits
|
|
|
|
|
|
class Status(TupleOf):
|
|
# shorten initialisation and allow acces to status enumMembers from status values
|
|
def __init__(self, enum):
|
|
TupleOf.__init__(self, EnumType(enum), StringType())
|
|
self.enum = enum
|
|
|
|
def __getattr__(self, key):
|
|
enum = TupleOf.__getattr__(self, 'enum')
|
|
if hasattr(enum, key):
|
|
return getattr(enum, key)
|
|
return TupleOf.__getattr__(self, key)
|
|
|
|
|
|
# argumentnames to lambda from spec!
|
|
DATATYPES = dict(
|
|
bool =BoolType,
|
|
int =lambda min, max, **kwds: IntRange(minval=min, maxval=max, **kwds),
|
|
scaled =lambda scale, min, max, **kwds: ScaledInteger(scale=scale, minval=min*scale, maxval=max*scale, **kwds),
|
|
double =lambda min=None, max=None, **kwds: FloatRange(minval=min, maxval=max, **kwds),
|
|
blob =lambda maxbytes, minbytes=0: BLOBType(minbytes=minbytes, maxbytes=maxbytes),
|
|
string =lambda minchars=0, maxchars=None: StringType(minchars=minchars, maxchars=maxchars),
|
|
array =lambda maxlen, members, minlen=0: ArrayOf(get_datatype(members), minlen=minlen, maxlen=maxlen),
|
|
tuple =lambda members: TupleOf(*tuple(map(get_datatype, members))),
|
|
enum =lambda members: EnumType('', members=members),
|
|
struct =lambda members, optional=None: StructOf(optional,
|
|
**dict((n, get_datatype(t)) for n, t in list(members.items()))),
|
|
command = lambda argument=None, result=None: CommandType(get_datatype(argument), get_datatype(result)),
|
|
)
|
|
|
|
|
|
# important for getting the right datatype from formerly jsonified descr.
|
|
def get_datatype(json):
|
|
"""returns a DataType object from description
|
|
|
|
inverse of <DataType>.export_datatype()
|
|
"""
|
|
if json is None:
|
|
return json
|
|
if isinstance(json, list) and len(json) == 2:
|
|
base, args = json # still allow old syntax
|
|
else:
|
|
try:
|
|
args = json.copy()
|
|
base = args.pop('type')
|
|
except (TypeError, KeyError, AttributeError):
|
|
raise BadValueError('a data descriptor must be a dict containing a "type" key, not %r' % json)
|
|
try:
|
|
return DATATYPES[base](**args)
|
|
except (TypeError, AttributeError, KeyError):
|
|
raise BadValueError('invalid data descriptor: %r' % json)
|