feat: Implement DMChannel model and enhance DM handling

Author: seriaati

hoyo_buddy/bot/bot.py

@@ -288,9 +288,13 @@ async def dm_user(
         logger.debug(f"DMing user {user_id}")
 
         try:
-            user = self.get_user(user_id) or await self.fetch_user(user_id)
-            message = await user.send(content, **kwargs)
-        except (discord.Forbidden) as e:
+            channel = await models.DMChannel.get_or_none(user_id=user_id)
+            if channel is None:
+                user = self.get_user(user_id) or await self.fetch_user(user_id)
+                dm_channel = user.dm_channel or await user.create_dm()
+                channel = await models.DMChannel.create(user_id=user_id, id=dm_channel.id)
+            message = await self.get_partial_messageable(channel.id).send(content, **kwargs)
+        except discord.HTTPException as e:
             logger.debug(f"Failed to DM user {user_id}: {e}")
         except Exception as e:
             self.capture_exception(e)

hoyo_buddy/db/models.py

@@ -610,3 +610,8 @@ async def create(
             task_type=task_type,
             type="default" if isinstance(embed, DefaultEmbed) else "error",
         )
+
+
+class DMChannel(BaseModel):
+    id = fields.BigIntField(pk=True, generated=False)
+    user_id = fields.BigIntField()

migrations/models/32_20250314161940_update.py

@@ -0,0 +1,14 @@
+from tortoise import BaseDBAsyncClient
+
+
+async def upgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        CREATE TABLE IF NOT EXISTS "dmchannel" (
+    "id" BIGINT NOT NULL PRIMARY KEY,
+    "user_id" BIGINT NOT NULL
+);"""
+
+
+async def downgrade(db: BaseDBAsyncClient) -> str:
+    return """
+        DROP TABLE IF EXISTS "dmchannel";"""