# -*- 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 # # ***************************************************************************** """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, isUTF8=False: StringType(minchars=minchars, maxchars=maxchars, isUTF8=isUTF8), 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 .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)