satcfdi.models.date_period

Complexity

Total Complexity

31

Size/Duplication

Total Lines	77
Duplicated Lines	0 %

Test Coverage

Coverage

92.45%

Importance

Changes

0

10 Methods

Rating	Name	Duplication	Size	Complexity
A	DatePeriod.__le__()	0	5	2
A	DatePeriod.__str__()	0	7	3
A	DatePeriod.__ge__()	0	5	2
A	DatePeriod.__lt__()	0	8	5
A	DatePeriod.__gt__()	0	8	5
A	DatePeriod.__init__()	0	8	5
A	DatePeriod.__eq__()	0	8	5
A	DatePeriod.__ne__()	0	5	2
A	DatePeriod.__hash__()	0	2	1
A	DatePeriod.strftime()	0	5	1

from datetime import date


class DatePeriod:
    # A class from defining a data range to then do date comparisons

    def __init__(self, year: int | None, month: int = None, day: int = None):
        self.year = year
        if year is None and month is not None:
            raise ValueError("Month can't have value if Year doesn't")
        self.month = month
        if month is None and day is not None:
            raise ValueError("Day can't have value if Month doesn't")
        self.day = day

    def __eq__(self, other):
        if isinstance(other, date | DatePeriod):
            for s, o in ((self.year, other.year), (self.month, other.month), (self.day, other.day)):
                if s is not None and s != o:
                    return False
            return True

        return NotImplemented

    def __ne__(self, other):
        if isinstance(other, date | DatePeriod):
            return not self.__eq__(other)

        return NotImplemented

    def __gt__(self, other):
        if isinstance(other, date | DatePeriod):
            for s, o in ((self.year, other.year), (self.month, other.month), (self.day, other.day)):
                if s is not None and s != o:
                    return s > o
            return False

        return NotImplemented

    def __ge__(self, other):
        if isinstance(other, date | DatePeriod):
            return self > other or self == other

        return NotImplemented

    def __lt__(self, other):
        if isinstance(other, date | DatePeriod):
            for s, o in ((self.year, other.year), (self.month, other.month), (self.day, other.day)):
                if s is not None and s != o:
                    return s < o
            return False

        return NotImplemented

    def __le__(self, other):
        if isinstance(other, date | DatePeriod):
            return self < other or self == other

        return NotImplemented

    def __str__(self):
        res = ""
        for n, f in ((self.year, "{:04d}"), (self.month, "-{:02d}"), (self.day, "-{:02d}")):
            if n:
                res += f.format(n)

        return res

    def __hash__(self):
        return (self.year or 0) * 1000 + (self.month or 0) * 100 + (self.day or 0)

    def strftime(self, fmt):
        return (fmt
                .replace("%Y", str(self.year))
                .replace("%m", str(self.month))
                .replace("%d", str(self.day))
                )