Update app/invite_logic.py

app/invite_logic.py

@@ -1,3 +1,4 @@
+# app/invite_logic.py
 import discord
 import os
 import asyncio
@@ -12,15 +13,17 @@ class DiscordInviteLogic:
         self._expiry_time = None
 
     async def start(self):
+        """Initialize the client with default intents."""
         intents = discord.Intents.default()
         self.client = discord.Client(intents=intents)
 
     async def stop(self):
-        if self.client:
+        """Clean up the client connection."""
+        if self.client and not self.client.is_closed():
             await self.client.close()
 
     async def get_invite(self):
-        Expires = 600
+        EXPIRY_DURATION = 600
         
         if self._cached_invite and self._expiry_time > datetime.now() + timedelta(seconds=30):
             remaining = self._expiry_time - datetime.now()
@@ -37,17 +40,21 @@ class DiscordInviteLogic:
             return {"error": "Missing Discord Config in Environment Secrets"}
 
         try:
+            if self.client is None or self.client.is_closed():
+                intents = discord.Intents.default()
+                self.client = discord.Client(intents=intents)
+
             await self.client.login(self.token)
             channel = await self.client.fetch_channel(int(self.channel_id))
             
             invite = await channel.create_invite(
-                max_age=Expires,
+                max_age=EXPIRY_DURATION,
                 max_uses=0, 
                 unique=True
             )
             
             self._cached_invite = invite.url
-            self._expiry_time = datetime.now() + timedelta(seconds=TEN_MINUTES_IN_SECONDS)
+            self._expiry_time = datetime.now() + timedelta(seconds=EXPIRY_DURATION)
             
             await self.client.close()
             
@@ -59,4 +66,4 @@ class DiscordInviteLogic:
         except Exception as e:
             if self.client and not self.client.is_closed():
                 await self.client.close()
-            return {"error": str(e)}
+            return {"error": f"Connection Error: {str(e)}"}