.venv/Lib/site-packages/rlp/sedes/text.py

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()