| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677 |
- from rlp.atomic import (
- Atomic,
- )
- from rlp.exceptions import (
- DeserializationError,
- SerializationError,
- )
- class Text:
- """
- A sedes object for encoded text data of certain length.
- :param min_length: the minimal length in encoded characters or `None` for no lower
- limit
- :param max_length: the maximal length in encoded characters or `None` for no upper
- limit
- :param allow_empty: if true, empty strings are considered valid even if
- a minimum length is required otherwise
- """
- def __init__(
- self, min_length=None, max_length=None, allow_empty=False, encoding="utf8"
- ):
- self.min_length = min_length or 0
- if max_length is None:
- self.max_length = float("inf")
- else:
- self.max_length = max_length
- self.allow_empty = allow_empty
- self.encoding = encoding
- @classmethod
- def fixed_length(cls, length, allow_empty=False):
- """Create a sedes for text data with exactly `length` encoded characters."""
- return cls(length, length, allow_empty=allow_empty)
- @classmethod
- def is_valid_type(cls, obj):
- return isinstance(obj, str)
- def is_valid_length(self, length):
- return any(
- (
- self.min_length <= length <= self.max_length,
- self.allow_empty and length == 0,
- )
- )
- def serialize(self, obj):
- if not self.is_valid_type(obj):
- raise SerializationError(f"Object is not a serializable ({type(obj)})", obj)
- if not self.is_valid_length(len(obj)):
- raise SerializationError("Object has invalid length", obj)
- return obj.encode(self.encoding)
- def deserialize(self, serial):
- if not isinstance(serial, Atomic):
- raise DeserializationError(
- f"Objects of type {type(serial).__name__} cannot be deserialized",
- serial,
- )
- try:
- text_value = serial.decode(self.encoding)
- except UnicodeDecodeError as err:
- raise DeserializationError(str(err), serial)
- if self.is_valid_length(len(text_value)):
- return text_value
- else:
- raise DeserializationError(f"{type(serial)} has invalid length", serial)
- text = Text()
|
