Update app/invite_logic.py

app/invite_logic.py

@@ -3,12 +3,14 @@ import httpx
 import os
 import dns.resolver
 from datetime import datetime, timedelta
+from zoneinfo import ZoneInfo
 
 
 class DiscordInviteLogic:
     def __init__(self):
         self.token = os.getenv("DISCORD_TOKEN")
         self.channel_id = os.getenv("CHANNEL_ID")
+        self.timezone = ZoneInfo(os.getenv("TIMEZONE", "Asia/Ho_Chi_Minh"))
         self.client = None
         self._cached_invite = None
         self._expiry_time = None
@@ -35,20 +37,22 @@ class DiscordInviteLogic:
 
     async def get_invite(self):
         EXPIRY_DURATION = 600
+        now = datetime.now(self.timezone)
 
         if (
             self._cached_invite
             and self._expiry_time
-            and self._expiry_time > datetime.utcnow() + timedelta(seconds=30)
+            and self._expiry_time > now + timedelta(seconds=30)
         ):
-            remaining = self._expiry_time - datetime.utcnow()
+            remaining = self._expiry_time - now
             mins, secs = divmod(int(remaining.total_seconds()), 60)
 
             return {
                 "status": "cached",
                 "invite": self._cached_invite,
                 "expires_in": f"{mins:02d}:{secs:02d}",
-                "expires_at": self._expiry_time.strftime("%Y-%m-%d %H:%M:%S"),
+                "expires_at": self._expiry_time.isoformat(),
+                "timezone": str(self.timezone),
             }
 
         if not self.token or not self.channel_id:
@@ -84,14 +88,13 @@ class DiscordInviteLogic:
                 invite_url = f"https://discord.gg/{data.get('code')}"
 
                 self._cached_invite = invite_url
-                self._expiry_time = datetime.utcnow() + timedelta(
-                    seconds=EXPIRY_DURATION
-                )
+                self._expiry_time = now + timedelta(seconds=EXPIRY_DURATION)
 
                 return {
                     "status": "generated",
                     "invite": invite_url,
-                    "expires_at": self._expiry_time.strftime("%Y-%m-%d %H:%M:%S"),
+                    "expires_at": self._expiry_time.isoformat(),
+                    "timezone": str(self.timezone),
                 }
 
             return {