nonebot-plugin-course-schedule 1.0.0__py3-none-any.whl

nonebot_plugin_course_schedule/__init__.py

@@ -0,0 +1,56 @@
+from nonebot.plugin import PluginMetadata
+from .config import Config, config
+
+__plugin_meta__ = PluginMetadata(
+    name="电子课程表",
+    description="绑定课表、查看课程、查看群友的课程，以及……上课排行",
+    usage="""▶ 课表帮助：打印本信息
+▶ 绑定课表：发送你的 .ics 文件或 WakeUp 分享口令来绑定课表
+  ▷ 可以重新绑定，可以通过旦夕导出
+▶ 解绑课表：删掉课表
+  ▷ 解绑会将你解绑所有群聊
+▶ 绑定群聊：让自己显示在本群的课表中
+  ▷ 绑定课表时会自动绑定群聊
+▶ 解绑群聊：让自己从本群的课表中消失
+▶ 查看课表 <offset|date>：显示你今天要上的课程
+▶ 群课表 <offset|date>：显示群友正在上的课和将要上的课
+▶ 上课排行：看看苦逼群友本周上了多少课
+""",
+    type="application",
+    homepage="https://github.com/GLDYM/nonebot-plugin-course-schedule",
+    config=Config,
+    supported_adapters={"~onebot.v11"},
+    extra={"author": "Polaris_Light", "version": "0.0.1", "priority": 5},
+)
+
+from nonebot import require
+
+require("nonebot_plugin_apscheduler")
+
+from typing import Union
+from nonebot import on_command
+from nonebot.adapters.onebot.v11 import (
+    GroupMessageEvent,
+    PrivateMessageEvent,
+)
+
+from .commands import (
+    bind_schedule,
+    bind_group,
+    show_today,
+    group_schedule,
+    weekly_ranking,
+)
+
+help_cmd = on_command(
+    "course_help",
+    aliases={"课表帮助", "课程帮助"},
+    force_whitespace=True,
+    priority=5,
+    block=True,
+)
+
+
+@help_cmd.handle()
+async def _(event: Union[GroupMessageEvent, PrivateMessageEvent]):
+    await help_cmd.finish(__plugin_meta__.usage)

nonebot_plugin_course_schedule/commands/bind_group.py

@@ -0,0 +1,35 @@
+import os
+
+from nonebot import on_command
+from nonebot.adapters.onebot.v11 import GroupMessageEvent, Message
+from ..utils.data_manager import data_manager
+
+
+bind_group = on_command(
+    "bind_group", aliases={"绑定群聊"}, force_whitespace=True, priority=5, block=True
+)
+unbind_group = on_command(
+    "unbind_group", aliases={"解绑群聊"}, force_whitespace=True, priority=5, block=True
+)
+
+
+@bind_group.handle()
+async def bind_group_handle(event: GroupMessageEvent):
+    group_id = event.group_id
+    user_id = event.user_id
+
+    ics_path = data_manager.get_ics_file_path(user_id)
+    if not os.path.exists(ics_path):
+        await bind_group.finish(f"请先绑定课表！")
+
+    data_manager.add_user_to_group(user_id, group_id)
+    await bind_group.finish(f"绑定群聊成功！")
+
+
+@unbind_group.handle()
+async def bind_group_handle(event: GroupMessageEvent):
+    group_id = event.group_id
+    user_id = event.user_id
+
+    data_manager.remove_user_from_group(user_id, group_id)
+    await bind_group.finish(f"解绑群聊成功！")

nonebot_plugin_course_schedule/commands/bind_schedule.py

@@ -0,0 +1,175 @@
+import time
+import os
+from datetime import datetime, timedelta
+from typing import Union
+
+import aiohttp
+
+from nonebot import on_command, logger
+from nonebot.matcher import Matcher
+from nonebot.params import Arg
+
+from nonebot.adapters.onebot.v11 import (
+    Bot,
+    GroupMessageEvent,
+    PrivateMessageEvent,
+    Message,
+    MessageSegment,
+)
+from nonebot_plugin_apscheduler import scheduler
+
+from ..utils.data_manager import data_manager
+from ..utils.ics_parser import ics_parser
+
+
+bind_schedule = on_command(
+    "bind_schedule",
+    aliases={"绑定课表", "绑定课程"},
+    force_whitespace=True,
+    priority=5,
+    block=True,
+)
+unbind_schedule = on_command(
+    "unbind_schedule",
+    aliases={"解绑课表", "解绑课程"},
+    force_whitespace=True,
+    priority=5,
+    block=True,
+)
+binding_requests = {}
+
+
+@bind_schedule.handle()
+async def handle_bind_entry(
+    matcher: Matcher, event: Union[GroupMessageEvent, PrivateMessageEvent]
+):
+    user_id = event.user_id
+
+    async def timeout():
+        await matcher.send(
+            MessageSegment.at(user_id)
+            + "绑定请求已过期，请重新发送 绑定课表 命令以绑定。"
+        )
+        return None
+
+    scheduler.add_job(
+        func=timeout,
+        trigger="date",
+        run_date=datetime.now() + timedelta(seconds=60),
+        id=f"expire_bind_request_{user_id}",
+        replace_existing=True,
+    )
+
+    await matcher.send("请在60秒内发送你的 .ics 文件或 WakeUp 分享口令。")
+
+
+@bind_schedule.got("schedule_input", prompt="等待你的课表文件或口令中…")
+async def handle_schedule_input(
+    bot: Bot,
+    matcher: Matcher,
+    event: Union[GroupMessageEvent, PrivateMessageEvent],
+    schedule_input: Message = Arg(),
+):
+    group_id = event.group_id if isinstance(event, GroupMessageEvent) else None
+    user_id = event.user_id
+    scheduler.remove_job(f"expire_bind_request_{user_id}")
+
+    # await matcher.send(f"Arg: {str(schedule_input)}")
+
+    # Wake Up Token
+    token = ics_parser.parse_wakeup_token(str(schedule_input))
+    if token:
+        try:
+            json_data = await ics_parser.fetch_wakeup_schedule(token)
+
+            if not json_data:
+                await matcher.send(
+                    "无法获取 WakeUp 课程表数据，请检查口令是否正确或已过期。"
+                )
+                return None
+            ics_content = ics_parser.convert_wakeup_to_ics(json_data)
+            if not ics_content:
+                await matcher.send("课程表数据解析失败，无法生成 ICS 文件。")
+                return None
+            ics_path = data_manager.get_ics_file_path(user_id)
+            with open(ics_path, "w", encoding="utf-8") as f:
+                f.write(ics_content)
+
+            if group_id:
+                data_manager.add_user_to_group(user_id, group_id)
+
+            ics_parser.clear_cache(ics_path)
+            await matcher.send("通过 WakeUp 口令绑定课表成功！")
+            return None
+        except Exception as e:
+            await matcher.finish(f"处理 WakeUp 口令失败: {e}")
+            logger.error(e)
+
+    # .ics 文件上传
+    for seg in schedule_input:
+        if seg.type == "file":
+            try:
+                file_id = seg.data.get("file_id")
+                file_url = await get_file_url(bot, event, file_id)
+
+                async with aiohttp.ClientSession() as session:
+                    async with session.get(file_url) as resp:
+                        ics_content = await resp.text()
+
+                ics_path = data_manager.get_ics_file_path(user_id)
+                with open(ics_path, "w", encoding="utf-8") as f:
+                    f.write(ics_content)
+
+                # 防止炒饭
+                ics_parser.clear_cache(ics_path)
+                parsed = ics_parser.parse_ics_file(ics_path)
+                if parsed == []:
+                    os.remove(ics_path)
+                    raise ValueError("Not Valid ICS File.")
+
+                if group_id:
+                    data_manager.add_user_to_group(user_id, group_id)
+
+                await matcher.send("课表文件绑定成功！")
+
+            except Exception as e:
+                raise e
+                # await matcher.finish(f"下载或保存课表文件失败：{e}")
+            return
+
+    await matcher.finish(
+        "未识别的口令或文件格式，请确认是否为 WakeUp 分享口令或 .ics 文件。"
+    )
+
+
+async def get_file_url(
+    bot: Bot, event: Union[GroupMessageEvent, PrivateMessageEvent], file_id: str
+) -> str:
+    if isinstance(event, GroupMessageEvent):
+        data = await bot.get_group_file_url(
+            group=event.group_id, group_id=event.group_id, file_id=file_id
+        )
+        return data["url"]
+    elif isinstance(event, PrivateMessageEvent):
+        data = await bot.get_private_file_url(user_id=bot.self_id, file_id=file_id)
+        return data["url"]
+
+
+@unbind_schedule.handle()
+async def handle_unbind_entry(event: Union[GroupMessageEvent, PrivateMessageEvent]):
+    user_id = event.user_id
+
+    ics_path = data_manager.get_ics_file_path(user_id)
+    if os.path.exists(ics_path):
+        os.remove(ics_path)
+    ics_parser.clear_cache(str(ics_path))
+
+    user_data = data_manager.load_user_data()
+
+    for group_id in list(user_data.keys()):
+        if user_id in user_data[group_id]:
+            user_data[group_id].remove(user_id)
+
+    data_manager.save_user_data(user_data)
+
+    await unbind_schedule.finish(f"解绑成功啦！")

nonebot_plugin_course_schedule/commands/group_schedule.py

@@ -0,0 +1,125 @@
+import os
+import shlex
+from datetime import datetime, time, timezone, timedelta
+from dateutil import parser
+
+from nonebot import on_command, logger
+from nonebot.adapters import Message
+from nonebot.params import CommandArg
+from nonebot.adapters.onebot.v11 import Bot, GroupMessageEvent, MessageSegment
+
+from ..utils.data_manager import data_manager
+from ..utils.ics_parser import ics_parser
+from ..utils.image_generator import image_generator
+
+
+group_schedule = on_command(
+    "group_schedule",
+    aliases={"群课表", "群友上什么", "群友在上什么", "群友在上什么课"},
+    priority=5,
+    force_whitespace=True,
+    block=True,
+)
+
+
+@group_schedule.handle()
+async def _(bot: Bot, event: GroupMessageEvent, arg: Message = CommandArg()):
+    group_id = event.group_id
+    user_data = data_manager.load_user_data()
+    if str(group_id) not in user_data:
+        await group_schedule.send("本群还没有人绑定课表哦~")
+        return None
+
+    user_ids = user_data[str(group_id)]
+
+    args = shlex.split(arg.extract_plain_text())
+    logger.info(f"{group_id} 查询群课表: {args}")
+    day = args[0].replace(".", "-") if args and args != [] else ""
+
+    shanghai_tz = timezone(timedelta(hours=8))
+    now = datetime.now(shanghai_tz)
+
+    try:
+        if day == "":
+            target_time = now
+            target_date = now.date()
+        elif day.isdigit():
+            offset_days = int(day)
+            target_date = now.date() + timedelta(days=offset_days)
+            target_time = datetime.combine(now.date(), time.min).astimezone(
+                shanghai_tz
+            ) + timedelta(seconds=1)
+        else:
+            target_time = parser.parse(day)
+            target_date = target_time.date()
+            target_time = datetime.combine(now.date(), time.min).astimezone(
+                shanghai_tz
+            ) + timedelta(seconds=1)
+    except Exception:
+        await group_schedule.finish(
+            "时间格式错误，请输入数字或日期，例如：3 或 2025-11-01"
+        )
+
+    next_courses = []
+
+    for user_id in user_ids:
+        ics_path = data_manager.get_ics_file_path(user_id)
+        if not os.path.exists(ics_path):
+            continue
+
+        try:
+            courses = ics_parser.parse_ics_file(ics_path)
+        except Exception:
+            continue
+
+        today_courses = [c for c in courses if c["start_time"].date() == target_date]
+        current = next_ = None
+
+        for course in today_courses:
+            if course["start_time"] <= target_time < course["end_time"]:
+                current = course
+                break
+            elif course["start_time"] > target_time:
+                if not next_ or course["start_time"] < next_["start_time"]:
+                    next_ = course
+
+        display = current or next_
+        user_info = await bot.get_group_member_info(group_id=group_id, user_id=user_id)
+        nickname = (
+            user_info["card"]
+            if user_info["card"] is not None and user_info["card"] != ""
+            else user_info["nickname"]
+        )
+
+        if display:
+            next_courses.append(
+                {
+                    "summary": display["summary"],
+                    "description": display["description"],
+                    "location": display["location"],
+                    "start_time": display["start_time"],
+                    "end_time": display["end_time"],
+                    "user_id": user_id,
+                    "nickname": nickname,
+                }
+            )
+        else:
+            next_courses.append(
+                {
+                    "summary": "今日无课",
+                    "description": "",
+                    "location": "",
+                    "start_time": None,
+                    "end_time": None,
+                    "user_id": user_id,
+                    "nickname": nickname,
+                }
+            )
+
+    if not next_courses:
+        await group_schedule.send("群友们接下来都没有课啦！")
+        return None
+
+    next_courses.sort(key=lambda x: (x["start_time"] is None, x["start_time"]))
+    image_path = await image_generator.generate_schedule_image(next_courses)
+    await group_schedule.send(MessageSegment.image(image_path))

nonebot_plugin_course_schedule/commands/show_today.py

@@ -0,0 +1,117 @@
+import os
+import shlex
+from datetime import datetime, timezone, timedelta
+from dateutil import parser
+from typing import Union
+
+from nonebot import on_command, logger
+from nonebot.adapters import Message
+from nonebot.params import CommandArg
+from nonebot.adapters.onebot.v11 import (
+    Bot,
+    GroupMessageEvent,
+    PrivateMessageEvent,
+    MessageSegment,
+)
+from ..utils.data_manager import data_manager
+from ..utils.ics_parser import ics_parser
+from ..utils.image_generator import image_generator
+
+show_today = on_command(
+    "show_today",
+    aliases={"课表", "查看课表", "查看今日课表", "查看我的课表"},
+    force_whitespace=True,
+    priority=5,
+    block=True,
+)
+
+
+@show_today.handle()
+async def _(
+    bot: Bot,
+    event: Union[GroupMessageEvent, PrivateMessageEvent],
+    arg: Message = CommandArg(),
+):
+    group_id = event.group_id if isinstance(event, GroupMessageEvent) else None
+    user_id = event.user_id
+
+    args = shlex.split(arg.extract_plain_text())
+    logger.info(f"{user_id} 查询课表: {args}")
+    day = args[0].replace(".", "-") if args and args != [] else ""
+
+    shanghai_tz = timezone(timedelta(hours=8))
+    now = datetime.now(shanghai_tz)
+
+    try:
+        if day == "":
+            target_date = now.date()
+            mode = "today"
+        elif day.isdigit():
+            offset_days = int(day)
+            target_date = now.date() + timedelta(days=offset_days)
+            mode = "offset"
+        else:
+            target_time = parser.parse(day)
+            target_date = target_time.date()
+            mode = "specific"
+    except Exception:
+        await show_today.finish("时间格式错误，请输入数字或日期，例如：3 或 2025-11-01")
+
+    ics_path = data_manager.get_ics_file_path(user_id)
+    if not os.path.exists(ics_path):
+        await show_today.finish(
+            "你还没有绑定课表哦~请在群内发送 绑定课表 指令，然后发送 .ics 文件或 WakeUp 口令来绑定。"
+        )
+
+    courses = ics_parser.parse_ics_file(ics_path)
+    if mode == "today":
+        # 不带参数，只显示剩下的课程
+        filtered_courses = [
+            c
+            for c in courses
+            if c["start_time"].date() == target_date and c["end_time"] > now
+        ]
+    else:
+        # 指定天，查询当天全部课程（包括 0）
+        filtered_courses = [c for c in courses if c["start_time"].date() == target_date]
+
+    if not filtered_courses:
+        await show_today.finish("当日没有课啦！")
+
+    filtered_courses.sort(key=lambda x: x["start_time"])
+
+    # 那是谁？是谁？是谁？ 那是复旦，复旦教务，复旦教务~
+    merged_courses = []
+    seen = {}
+    for course in filtered_courses:
+        key = (course["summary"], course["start_time"], course["end_time"])
+        if key in seen:
+            seen[key]["location"] += f", {course['location']}"
+        else:
+            seen[key] = course
+            merged_courses.append(course)
+    filtered_courses = merged_courses
+
+    if group_id:
+        user_info = await bot.get_group_member_info(group_id=group_id, user_id=user_id)
+        nickname = (
+            user_info["card"]
+            if user_info["card"] is not None and user_info["card"] != ""
+            else user_info["nickname"]
+        )
+    else:
+        user_info = await bot.get_stranger_info(user_id=user_id)
+        nickname = user_info["nickname"]
+
+    for course in filtered_courses:
+        course["nickname"] = nickname
+
+    image_path = (
+        await image_generator.generate_user_schedule_image(filtered_courses, nickname)
+        if mode == "today"
+        else await image_generator.generate_user_schedule_image(
+            filtered_courses, nickname, target_date
+        )
+    )
+
+    await show_today.finish(MessageSegment.image(image_path))

nonebot_plugin_course_schedule/commands/weekly_ranking.py

@@ -0,0 +1,84 @@
+from nonebot import on_command
+from nonebot.adapters.onebot.v11 import Bot, GroupMessageEvent, MessageSegment
+from datetime import datetime, timedelta, timezone
+from ..utils.data_manager import data_manager
+from ..utils.ics_parser import ics_parser
+from ..utils.image_generator import image_generator
+import os
+
+weekly_ranking = on_command(
+    "weekly_ranking", aliases={"上课排行", "本周上课排行"}, priority=5, block=True
+)
+
+
+@weekly_ranking.handle()
+async def _(bot: Bot, event: GroupMessageEvent):
+    group_id = event.group_id
+    user_data = data_manager.load_user_data()
+    if str(group_id) not in user_data:
+        await weekly_ranking.send("本群还没有人绑定课表哦~")
+        return None
+
+    user_ids = user_data[str(group_id)]
+
+    now = datetime.now(timezone(timedelta(hours=8)))
+    today = now.date()
+    start_of_week = today - timedelta(days=today.weekday())
+    end_of_week = start_of_week + timedelta(days=6)
+
+    ranking_data = []
+
+    for user_id in user_ids:
+        ics_path = data_manager.get_ics_file_path(user_id)
+        if not os.path.exists(ics_path):
+            continue
+
+        try:
+            courses = ics_parser.parse_ics_file(ics_path)
+        except Exception:
+            continue
+
+        total_duration = timedelta()
+        course_count = 0
+        seen = {}
+
+        for course in courses:
+            # 那是谁？是谁？是谁？ 那是复旦，复旦教务，复旦教务~
+            key = (course["summary"], course["start_time"], course["end_time"])
+            if key in seen:
+                continue
+            else:
+                seen[key] = course
+
+            course_date = course["start_time"].date()
+            if start_of_week <= course_date <= end_of_week:
+                total_duration += course["end_time"] - course["start_time"]
+                course_count += 1
+
+        if course_count > 0:
+            user_info = await bot.get_group_member_info(
+                group_id=group_id, user_id=user_id
+            )
+            nickname = (
+                user_info["card"]
+                if user_info["card"] is not None and user_info["card"] != ""
+                else user_info["nickname"]
+            )
+            ranking_data.append(
+                {
+                    "user_id": user_id,
+                    "nickname": nickname,
+                    "total_duration": total_duration,
+                    "course_count": course_count,
+                }
+            )
+
+    if not ranking_data:
+        await weekly_ranking.send("本周大家都没有课呢！")
+        return
+
+    ranking_data.sort(key=lambda x: x["total_duration"], reverse=True)
+    image_path = await image_generator.generate_ranking_image(
+        ranking_data, start_of_week, end_of_week
+    )
+    await weekly_ranking.send(MessageSegment.image(image_path))

nonebot_plugin_course_schedule/config.py

@@ -0,0 +1,15 @@
+from pydantic import BaseModel, Field
+from pathlib import Path
+
+from nonebot import get_plugin_config
+
+
+# fmt:off
+class Config(BaseModel):
+    data_path: str = "data/course_schedule"
+    fort_path: str = Field(default_factory=lambda: str(
+        Path(__file__).parent / "resources" / "MapleMono-NF-CN-Medium.ttf"
+    ))
+# fmt:on
+
+config = get_plugin_config(Config)

nonebot_plugin_course_schedule/utils/constants.py

@@ -0,0 +1,48 @@
+# -*- coding: utf-8 -*-
+"""
+本模块用于存放课程表插件的常量
+"""
+
+# --- Group Schedule Image Styles ---
+GS_BG_COLOR = "#FFFFFF"
+GS_FONT_COLOR = "#333333"
+GS_TITLE_COLOR = "#000000"
+GS_SUBTITLE_COLOR = "#888888"
+GS_STATUS_COLORS = {
+    "进行中": ("#D32F2F", "#FFFFFF"),
+    "下一节": ("#1976D2", "#FFFFFF"),
+    "已结束": ("#388E3C", "#FFFFFF"),
+    "无课程": ("#757575", "#FFFFFF"),
+}
+GS_AVATAR_SIZE = 80
+GS_ROW_HEIGHT = 120
+GS_PADDING = 40
+GS_WIDTH = 1200
+
+# --- User Schedule Image Styles ---
+US_BG_COLOR = "#FFFFFF"
+US_FONT_COLOR = "#333333"
+US_TITLE_COLOR = "#000000"
+US_SUBTITLE_COLOR = "#888888"
+US_COURSE_BG_COLOR = "#E3F2FD"
+US_ROW_HEIGHT = 100
+US_ROW_PADDING = 15
+US_ROW_SPACING = 5
+US_ROW_MAX_UNIT = 66
+US_PADDING = 40
+US_WIDTH = 1000
+US_SPACING = 5
+US_MAX_UNIT = 38
+
+# --- Ranking Image Styles ---
+RANKING_BG_COLOR = "#FFFFFF"
+RANKING_TITLE_COLOR = "#1F2937"
+RANKING_SUBTITLE_COLOR = "#6B7280"
+RANKING_FONT_COLOR = "#374151"
+RANKING_ROW_BG_COLOR = "#F9FAFB"
+RANKING_COLORS = {1: "#FBBF24", 2: "#9CA3AF", 3: "#F59E0B"}
+RANKING_WIDTH = 1200
+RANKING_PADDING = 60
+RANKING_HEADER_HEIGHT = 160
+RANKING_ROW_HEIGHT = 120
+RANKING_AVATAR_SIZE = 80