Umgang mit "Jahren und Monaten" in Python

Ich schrieb eine Klasse für "Jahre und Monate" in Python, erkannte aber später, dass dies mit dateutil möglich war, sodass es sich um einen Gedenkgottesdienst handelte. Ich habe es mit Schwung geschrieben und es gibt keinen Test, also benutze es auf eigenes Risiko.

Implementierung

import re
from datetime import date


class ym:
    """Datentyp, der das Jahr und den Monat behandelt"""

    def __init__(self, y: int, m: int = None):
        if type(y) is date:
            y, m = y.year, y.month
        if not (0 <= y <= 10000 and 1 <= m <= 12):
            raise ValueError()
        self.x = y * 12 + m - 1

    @classmethod
    def current(cls):
        return cls(date.today())

    @classmethod
    def parse(cls, s: str):
        matched = re.match(r"^(\d+)-(\d+)$", s)
        if not matched:
            raise ValueError()
        return cls(int(matched[1]), int(matched[2]))

    @property
    def y(self):
        return self.x // 12

    @property
    def m(self):
        return self.x % 12 + 1

    def to(self, y, m=None) -> list:
        target = y if type(y) is self.__class__ else self.__class__(y, m)
        return [self + i for i in range(target - self)] if self < target else []

    def __repr__(self):
        cls = self.__class__
        return f"{cls.__module__}.{cls.__name__}({self.y}, {self.m})"

    def __str__(self):
        return f"{self.y:04d}-{self.m:02d}"

    def __add__(self, n: int):
        if type(n) is int:
            return self.__class__((self.x + n) // 12, (self.x + n) % 12 + 1)
        raise TypeError()

    def __sub__(self, n):
        if type(n) is int:
            return self + (-1 * n)
        if type(n) is self.__class__:
            return self.x - n.x
        raise TypeError()

    def __eq__(self, other):
        return self.x == other.x

    def __ne__(self, other):
        return self.x != other.x

    def __lt__(self, other):
        return self.x < other.x

    def __gt__(self, other):
        return self.x > other.x

Beispiel

Erstellen Sie eine Instanz, indem Sie Jahr und Monat angeben

ym(2020, 4)  #=> ym(2020, 4)

Instanz aus aktuellem Monat erstellen

ym.current()  #=> ym(2020, 7)

Erstellen Sie eine Instanz aus einer Zeichenfolge

ym.parse("2020-04")  #=> ym(2020, 4)

Erstellen Sie eine Instanz aus einem Datumsobjekt

from datetime import date

ym.parse(date.today())  #=> ym(2020, 7)

from datetime import datetime

dt = dateime.now()
ym.parse(dt.date())  #=> ym(2020, 7)

Holen Sie sich Jahr, Monat

current = ym.current()  #=> ym(2020, 7)
current.y  #=> 2020
current.m  #=> 7

Addition und Subtraktion

ym(2020, 04) + 10  #=> ym(2021, 2)
ym(2020, 04) - 10  #=> ym(2019, 6)

Jahres- / Monatsunterschied

ym(2020, 7) - ym(2020, 4)  #=> 3

Vergleichsberechnung

ym(2020, 7) < ym(2020, 4)  #=> False
ym(2020, 7) > ym(2020, 4)  #=> True
ym(2020, 7) == ym(2020, 4)  #=> False

Reichweitenerfassung

ym(2020, 4).to(2020, 7)  #=> [ym(2020, 4), ym(2020, 5), ym(2020, 6)]

current = ym.current()  #=> ym(2020, 7)
current.to(current + 6)  #=> [ym(2020, 7), ym(2020, 8), ym(2020, 9), ym(2020, 10), ym(2020, 11), ym(2020, 12)]