From d4061922ee510bec5f3d9073cf38344cc55062c3 Mon Sep 17 00:00:00 2001
From: Ching L <loooching@gmail.com>
Date: Fri, 7 Mar 2025 11:45:09 +0800
Subject: [PATCH] feat: Implement delayed start for daily rank changes check

---
 discord_bot.py | 27 ++++++++++++++-------------
 1 file changed, 14 insertions(+), 13 deletions(-)

diff --git a/discord_bot.py b/discord_bot.py
index e7dd262..edbd616 100644
--- a/discord_bot.py
+++ b/discord_bot.py
@@ -8,6 +8,7 @@ import utils
 import sentry_sdk
 
 import datetime
+import asyncio
 
 sentry_sdk.init(
     dsn="https://272f1e4ecb3847ac8d24be796515e558@o4506942768021504.ingest.us.sentry.io/4506986058743808",
@@ -133,7 +134,7 @@ async def heartbeat():
 async def check_rank_changes(channel):
   logger.info("Checking for rank changes")
   try:
-    data = dota.check_rank_changes_for_discord()
+    data = await dota.check_rank_changes_for_discord()
     if data:
       logger.info(f"Sending rank changes: {data}")
       await channel.send(content=data['content'], embeds=[discord.Embed.from_dict(embed) for embed in data['embeds']])
@@ -141,24 +142,24 @@ async def check_rank_changes(channel):
     logger.error(f"Error checking rank changes: {e}")
     sentry_sdk.capture_exception(e)
 
+@check_rank_changes.before_loop
+async def before_check_rank_changes():
+  # 等待到晚上9点再开始第一次运行
+  now = datetime.datetime.now()
+  target_time = now.replace(hour=21, minute=0, second=0, microsecond=0)
+  if now >= target_time:
+    target_time = target_time + datetime.timedelta(days=1)
+  
+  seconds_until_target = (target_time - now).total_seconds()
+  await asyncio.sleep(seconds_until_target)
+
 @bot.event
 async def on_ready():
   channel = bot.get_channel(channel_id)
   send_message.start(channel)
   heartbeat.start()
   
-  # Start the rank check task and make it run at a specific time (e.g., noon)
+  # 启动天梯检查任务
   check_rank_changes.start(channel)
-  
-  # Calculate time until next noon
-  now = datetime.datetime.now()
-  target_time = now.replace(hour=21, minute=0, second=0, microsecond=0)
-  if now >= target_time:
-    target_time = target_time + datetime.timedelta(days=1)
-  
-  # Adjust first run to happen at noon
-  seconds_until_target = (target_time - now).total_seconds()
-  check_rank_changes.change_interval(hours=24)
-  check_rank_changes.next_iteration = discord.utils.utcnow() + datetime.timedelta(seconds=seconds_until_target)
 
 bot.run('MTE1MjE2NTc3NDMwNDIyMzI2Mg.GEi-17.VvuIkRy_cFD9XF6wtTagY95LKEbTxKaxy-FxGw') # 这里替换成你自己的 token