From 1be1a904f9c867acc19a6c727cc8684a43813825 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B9=94=E4=B8=8D=E6=80=9D?= <3504852205@qq.com>
Date: Sun, 24 Nov 2024 13:24:26 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=AE=9A=E6=97=B6=E4=BB=BB?=
 =?UTF-8?q?=E5=8A=A1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 lagrange/__init__.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/lagrange/__init__.py b/lagrange/__init__.py
index 47bf9da..fa1dfed 100644
--- a/lagrange/__init__.py
+++ b/lagrange/__init__.py
@@ -23,6 +23,7 @@ def __init__(
         device_info_path="./device.json",
         signinfo_path="./sig.bin",
     ):
+        self.scheduled_tasks = []  # 添加这行来存储计划任务
         self.im = InfoManager(uin, device_info_path, signinfo_path)
         self.uin = uin
         self.info = app_list[protocol]
@@ -40,6 +41,9 @@ async def login(self, client: Client):
             return True
         else:
             return await client.login()
+    def schedule_task(self, task):
+        """添加计划任务"""
+        self.scheduled_tasks.append(task)
 
     async def run(self):
         with self.im as im:
@@ -57,6 +61,9 @@ async def run(self):
         if not status:
             log.login.error("Login failed")
             return
+                    # 登录成功后执行计划任务
+        for task in self.scheduled_tasks:
+            asyncio.create_task(task(self.client))
         await self.client.wait_closed()
 
     def launch(self):