fix: add hacky workaround for small timestamps on windows (#1626)

AstreaTSS · web-flow · commit af9866f54030 · 2024-03-30T04:11:16.000-04:00
diff --git a/interactions/models/discord/timestamp.py b/interactions/models/discord/timestamp.py
@@ -1,5 +1,6 @@
 import time
-from datetime import datetime, timezone
+import sys
+from datetime import tzinfo, datetime, timezone
 from enum import Enum
 from typing import TYPE_CHECKING, Optional, Union
 
@@ -56,6 +57,9 @@ def fromisocalendar(cls, year: int, week: int, day: int) -> "Timestamp":
 
     @classmethod
     def fromtimestamp(cls, t: float, tz=None) -> "Timestamp":
+        if sys.platform == "win32" and t < 0:
+            raise ValueError("Negative timestamps are not supported on Windows.")
+
         try:
             timestamp = super().fromtimestamp(t, tz=tz)
         except Exception:
@@ -85,6 +89,28 @@ def utcnow(cls) -> "Timestamp":
         t = time.time()
         return cls.utcfromtimestamp(t)
 
+    def astimezone(self, tz: tzinfo | None = None) -> "Timestamp":
+        # workaround of https://github.com/python/cpython/issues/107078
+
+        if sys.platform != "win32":
+            return super().astimezone(tz)
+
+        # this bound is loose, but it's good enough for our purposes
+        if self.year > 1970 or (self.year == 1970 and (self.month > 1 or self.day > 1)):
+            return super().astimezone(tz)
+
+        if self.year < 1969 or self.month < 12 or self.day < 31:
+            # windows kind of breaks down for dates before unix time
+            # technically this is solvable, but it's not worth the effort
+            # also, again, this is a loose bound, but it's good enough for our purposes
+            raise ValueError("astimezone with no arguments is not supported for dates before Unix Time on Windows.")
+
+        # to work around the issue to some extent, we'll use a timestamp with a date
+        # that doesn't trigger the bug, and use the timezone from it to modify this
+        # timestamp
+        sample_datetime = Timestamp(1970, 1, 5).astimezone()
+        return self.replace(tzinfo=sample_datetime.tzinfo)
+
     def to_snowflake(self, high: bool = False) -> Union[str, "Snowflake"]:
         """
         Returns a numeric snowflake pretending to be created at the given date.
