| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667 |
- from rlp.atomic import (
- Atomic,
- )
- from rlp.exceptions import (
- DeserializationError,
- SerializationError,
- )
- class Binary:
- """
- A sedes object for binary data of certain length.
- :param min_length: the minimal length in bytes or `None` for no lower limit
- :param max_length: the maximal length in bytes 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):
- 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
- @classmethod
- def fixed_length(cls, length, allow_empty=False):
- """Create a sedes for binary data with exactly `length` bytes."""
- return cls(length, length, allow_empty=allow_empty)
- @classmethod
- def is_valid_type(cls, obj):
- return isinstance(obj, (bytes, bytearray))
- 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 Binary.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
- def deserialize(self, serial):
- if not isinstance(serial, Atomic):
- raise DeserializationError(
- f"Objects of type {type(serial).__name__} cannot be deserialized",
- serial,
- )
- if self.is_valid_length(len(serial)):
- return serial
- else:
- raise DeserializationError(f"{type(serial)} has invalid length", serial)
- binary = Binary()
|
