2f1a9fa966
清理了整个代码库中所有无用的模块导入、重复定义以及冗余变量引用，具体包括： - bot.py 去掉了 random、typing 的未使用 import - antipromptinjector 模块统一移除未引用的 DetectionResult、Dict、List 等 - chat_loop 中删除了未调用的 Timer、mai_thinking_manager、events_manager 等引用 - qzone_service 删除多余 f-string 大括号，避免日志警告格式问题 - 其他模块同步剔除各自范围内的冗余 import（asyncio、datetime 等共 20+ 处） 保持功能不变，仅作代码整洁度优化，无破坏性变更。(并添加了一个现在暂时还没加进去的必应搜索源文件)
67 lines
2.2 KiB
Python
67 lines
2.2 KiB
Python
# mmc/src/common/database/monthly_plan_db.py
|
|
|
|
from typing import List
|
|
from src.common.database.sqlalchemy_models import MonthlyPlan, get_db_session
|
|
from src.common.logger import get_logger
|
|
|
|
logger = get_logger("monthly_plan_db")
|
|
|
|
def add_new_plans(plans: List[str], month: str):
|
|
"""
|
|
批量添加新生成的月度计划到数据库。
|
|
|
|
:param plans: 计划内容列表。
|
|
:param month: 目标月份，格式为 "YYYY-MM"。
|
|
"""
|
|
with get_db_session() as session:
|
|
try:
|
|
new_plan_objects = [
|
|
MonthlyPlan(plan_text=plan, target_month=month)
|
|
for plan in plans
|
|
]
|
|
session.add_all(new_plan_objects)
|
|
session.commit()
|
|
logger.info(f"成功向数据库添加了 {len(new_plan_objects)} 条 {month} 的月度计划。")
|
|
except Exception as e:
|
|
logger.error(f"添加月度计划时发生错误: {e}")
|
|
session.rollback()
|
|
raise
|
|
|
|
def get_active_plans_for_month(month: str) -> List[MonthlyPlan]:
|
|
"""
|
|
获取指定月份所有未被软删除的计划。
|
|
|
|
:param month: 目标月份，格式为 "YYYY-MM"。
|
|
:return: MonthlyPlan 对象列表。
|
|
"""
|
|
with get_db_session() as session:
|
|
try:
|
|
plans = session.query(MonthlyPlan).filter(
|
|
MonthlyPlan.target_month == month,
|
|
MonthlyPlan.is_deleted == False
|
|
).all()
|
|
return plans
|
|
except Exception as e:
|
|
logger.error(f"查询 {month} 的有效月度计划时发生错误: {e}")
|
|
return []
|
|
|
|
def soft_delete_plans(plan_ids: List[int]):
|
|
"""
|
|
将指定ID的计划标记为软删除。
|
|
|
|
:param plan_ids: 需要软删除的计划ID列表。
|
|
"""
|
|
if not plan_ids:
|
|
return
|
|
|
|
with get_db_session() as session:
|
|
try:
|
|
session.query(MonthlyPlan).filter(
|
|
MonthlyPlan.id.in_(plan_ids)
|
|
).update({"is_deleted": True}, synchronize_session=False)
|
|
session.commit()
|
|
logger.info(f"成功软删除了 {len(plan_ids)} 条月度计划。")
|
|
except Exception as e:
|
|
logger.error(f"软删除月度计划时发生错误: {e}")
|
|
session.rollback()
|
|
raise |