Source code for satcfdi.models.code

import logging

logger = logging.getLogger(__name__)


class Code:
    def __init__(self, code, description):
        self.code = code
        self.description = description

    def __str__(self):
        if self.description is None:
            return self.code
        if isinstance(self.description, list):
            return str(self.code) + " - " + "; ".join(self.description)
        return str(self.code) + " - " + str(self.description)

    def __repr__(self):
        # return '%s.%s(%s)' % (self.__class__.__module__,
        #                       self.__class__.__qualname__,
        #                       f'{repr(self.code)}, {repr(self.description)}')

        return '%s(%s)' % (
            self.__class__.__qualname__,
            f'{repr(self.code)}, {repr(self.description)}'
        )

    def __eq__(self, other):
        if isinstance(other, Code):
            return self.code == other.code
        return self.code == other

    def __ne__(self, other):
        if isinstance(other, Code):
            return self.code != other.code
        return self.code != other

    def __lt__(self, other):
        if isinstance(other, Code):
            return self.code < other.code
        return self.code < other

    def __le__(self, other):
        if isinstance(other, Code):
            return self.code <= other.code
        return self.code <= other

    def __gt__(self, other):
        if isinstance(other, Code):
            return self.code > other.code
        return self.code > other

    def __ge__(self, other):
        if isinstance(other, Code):
            return self.code >= other.code
        return self.code >= other

    def __hash__(self):
        return self.code.__hash__()