fix: 优化scheduler_generator日志&格式

src/plugins/schedule/schedule_generator.py

@@ -13,21 +13,21 @@ from ..models.utils_model import LLM_request
 driver = get_driver()
 config = driver.config
 
-
 Database.initialize(
-            host= config.MONGODB_HOST,
+    host=config.MONGODB_HOST,
-            port= int(config.MONGODB_PORT),
+    port=int(config.MONGODB_PORT),
-            db_name=  config.DATABASE_NAME,
+    db_name=config.DATABASE_NAME,
-            username= config.MONGODB_USERNAME,
+    username=config.MONGODB_USERNAME,
-            password= config.MONGODB_PASSWORD,
+    password=config.MONGODB_PASSWORD,
-            auth_source=config.MONGODB_AUTH_SOURCE
+    auth_source=config.MONGODB_AUTH_SOURCE
-        )
+)
+
 
 class ScheduleGenerator:
     def __init__(self):
-        #根据global_config.llm_normal这一字典配置指定模型
+        # 根据global_config.llm_normal这一字典配置指定模型
         # self.llm_scheduler = LLMModel(model = global_config.llm_normal,temperature=0.9)
-        self.llm_scheduler = LLM_request(model = global_config.llm_normal,temperature=0.9)
+        self.llm_scheduler = LLM_request(model=global_config.llm_normal, temperature=0.9)
         self.db = Database.get_instance()
         self.today_schedule_text = ""
         self.today_schedule = {}
@@ -42,27 +42,29 @@ class ScheduleGenerator:
         yesterday = datetime.datetime.now() - datetime.timedelta(days=1)
 
         self.today_schedule_text, self.today_schedule = await self.generate_daily_schedule(target_date=today)
-        self.tomorrow_schedule_text, self.tomorrow_schedule = await self.generate_daily_schedule(target_date=tomorrow,read_only=True)
+        self.tomorrow_schedule_text, self.tomorrow_schedule = await self.generate_daily_schedule(target_date=tomorrow,
-        self.yesterday_schedule_text, self.yesterday_schedule = await self.generate_daily_schedule(target_date=yesterday,read_only=True)
+                                                                                                 read_only=True)
+        self.yesterday_schedule_text, self.yesterday_schedule = await self.generate_daily_schedule(
+            target_date=yesterday, read_only=True)
 
-    async def generate_daily_schedule(self, target_date: datetime.datetime = None,read_only:bool = False) -> Dict[str, str]:
+    async def generate_daily_schedule(self, target_date: datetime.datetime = None, read_only: bool = False) -> Dict[
+        str, str]:
 
         date_str = target_date.strftime("%Y-%m-%d")
         weekday = target_date.strftime("%A")
 
-
         schedule_text = str
 
         existing_schedule = self.db.db.schedule.find_one({"date": date_str})
         if existing_schedule:
-            print(f"{date_str}的日程已存在:")
+            logger.info(f"{date_str}的日程已存在:")
             schedule_text = existing_schedule["schedule"]
             # print(self.schedule_text)
 
         elif read_only == False:
-            print(f"{date_str}的日程不存在，准备生成新的日程。")
+            logger.info(f"{date_str}的日程不存在，准备生成新的日程。")
-            prompt = f"""我是{global_config.BOT_NICKNAME}，{global_config.PROMPT_SCHEDULE_GEN}，请为我生成{date_str}（{weekday}）的日程安排，包括："""+\
+            prompt = f"""我是{global_config.BOT_NICKNAME}，{global_config.PROMPT_SCHEDULE_GEN}，请为我生成{date_str}（{weekday}）的日程安排，包括：""" + \
-            """
+                     """
             1. 早上的学习和工作安排
             2. 下午的活动和任务
             3. 晚上的计划和休息时间
@@ -76,13 +78,13 @@ class ScheduleGenerator:
                 schedule_text = "生成日程时出错了"
             # print(self.schedule_text)
         else:
-            print(f"{date_str}的日程不存在。")
+            logger.info(f"{date_str}的日程不存在。")
             schedule_text = "忘了"
 
-            return schedule_text,None
+            return schedule_text, None
 
         schedule_form = self._parse_schedule(schedule_text)
-        return schedule_text,schedule_form
+        return schedule_text, schedule_form
 
     def _parse_schedule(self, schedule_text: str) -> Union[bool, Dict[str, str]]:
         """解析日程文本，转换为时间和活动的字典"""
@@ -90,8 +92,7 @@ class ScheduleGenerator:
             schedule_dict = json.loads(schedule_text)
             return schedule_dict
         except json.JSONDecodeError as e:
-            print(schedule_text)
+            logger.exception("解析日程失败: {}".format(schedule_text))
-            print(f"解析日程失败: {str(e)}")
             return False
 
     def _parse_time(self, time_str: str) -> str:
@@ -132,10 +133,10 @@ class ScheduleGenerator:
 
     def _time_diff(self, time1: str, time2: str) -> int:
         """计算两个时间字符串之间的分钟差"""
-        if time1=="24:00":
+        if time1 == "24:00":
-            time1="23:59"
+            time1 = "23:59"
-        if time2=="24:00":
+        if time2 == "24:00":
-            time2="23:59"
+            time2 = "23:59"
         t1 = datetime.datetime.strptime(time1, "%H:%M")
         t2 = datetime.datetime.strptime(time2, "%H:%M")
         diff = int((t2 - t1).total_seconds() / 60)
@@ -150,13 +151,14 @@ class ScheduleGenerator:
     def print_schedule(self):
         """打印完整的日程安排"""
         if not self._parse_schedule(self.today_schedule_text):
-            print("今日日程有误，将在下次运行时重新生成")
+            logger.warning("今日日程有误，将在下次运行时重新生成")
             self.db.db.schedule.delete_one({"date": datetime.datetime.now().strftime("%Y-%m-%d")})
         else:
-            print("\n=== 今日日程安排 ===")
+            logger.info("\n=== 今日日程安排 ===")
             for time_str, activity in self.today_schedule.items():
-                print(f"时间[{time_str}]: 活动[{activity}]")
+                logger.info(f"时间[{time_str}]: 活动[{activity}]")
-            print("==================\n")
+            logger.info("==================\n")
+
 
 # def main():
 #     # 使用示例