pocketpy/python/datetime.py

from time import localtime
import operator

class timedelta:
    def __init__(self, days=0, seconds=0):
        self.days = days
        self.seconds = seconds

    def __repr__(self):
        return f"datetime.timedelta(days={self.days}, seconds={self.seconds})"

    def __eq__(self, other: 'timedelta') -> bool:
        if type(other) is not timedelta:
            return NotImplemented
        return (self.days, self.seconds) == (other.days, other.seconds)

    def __ne__(self, other: 'timedelta') -> bool:
        if type(other) is not timedelta:
            return NotImplemented
        return (self.days, self.seconds) != (other.days, other.seconds)


class date:
    def __init__(self, year: int, month: int, day: int):
        self.year = year
        self.month = month
        self.day = day

    @staticmethod
    def today():
        t = localtime()
        return date(t.tm_year, t.tm_mon, t.tm_mday)
    
    def __cmp(self, other, op):
        if not isinstance(other, date):
            return NotImplemented
        if self.year != other.year:
            return op(self.year, other.year)
        if self.month != other.month:
            return op(self.month, other.month)
        return op(self.day, other.day)

    def __eq__(self, other: 'date') -> bool:
        return self.__cmp(other, operator.eq)
    
    def __ne__(self, other: 'date') -> bool:
        return self.__cmp(other, operator.ne)

    def __lt__(self, other: 'date') -> bool:
        return self.__cmp(other, operator.lt)

    def __le__(self, other: 'date') -> bool:
        return self.__cmp(other, operator.le)

    def __gt__(self, other: 'date') -> bool:
        return self.__cmp(other, operator.gt)

    def __ge__(self, other: 'date') -> bool:
        return self.__cmp(other, operator.ge)

    def __str__(self):
        return f"{self.year}-{self.month:02}-{self.day:02}"

    def __repr__(self):
        return f"datetime.date({self.year}, {self.month}, {self.day})"


class datetime(date):
    def __init__(self, year: int, month: int, day: int, hour: int, minute: int, second: int):
        super().__init__(year, month, day)
        # Validate and set hour, minute, and second
        if not 0 <= hour <= 23:
            raise ValueError("Hour must be between 0 and 23")
        self.hour = hour
        if not 0 <= minute <= 59:
            raise ValueError("Minute must be between 0 and 59")
        self.minute = minute
        if not 0 <= second <= 59:
            raise ValueError("Second must be between 0 and 59")
        self.second = second

    def date(self) -> date:
        return date(self.year, self.month, self.day)

    @staticmethod
    def now():
        t = localtime()
        tm_sec = t.tm_sec
        if tm_sec == 60:
            tm_sec = 59
        return datetime(t.tm_year, t.tm_mon, t.tm_mday, t.tm_hour, t.tm_min, tm_sec)

    def __str__(self):
        return f"{self.year}-{self.month:02}-{self.day:02} {self.hour:02}:{self.minute:02}:{self.second:02}"

    def __repr__(self):
        return f"datetime.datetime({self.year}, {self.month}, {self.day}, {self.hour}, {self.minute}, {self.second})"

    def __cmp(self, other, op):
        if not isinstance(other, datetime):
            return NotImplemented
        if self.year != other.year:
            return op(self.year, other.year)
        if self.month != other.month:
            return op(self.month, other.month)
        if self.day != other.day:
            return op(self.day, other.day)
        if self.hour != other.hour:
            return op(self.hour, other.hour)
        if self.minute != other.minute:
            return op(self.minute, other.minute)
        return op(self.second, other.second)

    def __eq__(self, other) -> bool:
        return self.__cmp(other, operator.eq)
    
    def __ne__(self, other) -> bool:
        return self.__cmp(other, operator.ne)
    
    def __lt__(self, other) -> bool:
        return self.__cmp(other, operator.lt)
    
    def __le__(self, other) -> bool:
        return self.__cmp(other, operator.le)
    
    def __gt__(self, other) -> bool:
        return self.__cmp(other, operator.gt)
    
    def __ge__(self, other) -> bool:
        return self.__cmp(other, operator.ge)