mirror of
https://github.com/ClovertaTheTrilobita/SanYeCao-Nonebot.git
synced 2026-04-01 22:04:51 +00:00
52 lines
2.2 KiB
Python
52 lines
2.2 KiB
Python
|
from nonebot.adapters.qq import MessageEvent
|
||
|
|
from nonebot.plugin import on_command
|
||
|
|
from nonebot.rule import to_me
|
||
|
|
from src.clover_sqlite.models.to_do import ToDoList
|
||
|
|
|
||
|
|
get_todo_list = on_command("待办查询", rule=to_me(), priority=10, block=True, aliases={"代办", "daiban"})
|
||
|
|
@get_todo_list.handle()
|
||
|
|
async def show_todo_list(message: MessageEvent):
|
||
|
|
"""
|
||
|
|
查询用户所有的待办
|
||
|
|
|
||
|
|
:param message:
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
member_openid = message.get_user_id()
|
||
|
|
result = await ToDoList.get_todo_list(member_openid)
|
||
|
|
if not result:
|
||
|
|
await get_todo_list.finish("\n您还未创建待办\n快使用 /新建待办 创建一份吧")
|
||
|
|
|
||
|
|
todo_list = "\n\n".join([f"{i + 1}. {content}" for i, content in enumerate(result)])
|
||
|
|
await get_todo_list.finish(f"您的待办有如下哦：⭐\n\n{todo_list}")
|
||
|
|
|
||
|
|
|
||
|
|
insert_todo = on_command("新建待办", rule=to_me(), priority=10, block=True)
|
||
|
|
@insert_todo.handle()
|
||
|
|
async def insert_todo_list(message: MessageEvent):
|
||
|
|
member_openid = message.get_user_id()
|
||
|
|
content = message.get_plaintext().replace("/新建待办", "").strip(" ")
|
||
|
|
success = await ToDoList.insert_todo_list(member_openid, content)
|
||
|
|
if success:
|
||
|
|
await insert_todo.finish("成功添加待办，今后也要加油哦(ง •_•)ง")
|
||
|
|
else:
|
||
|
|
await insert_todo.finish("\n请输入 /新建待办+待办内容 哦")
|
||
|
|
|
||
|
|
|
||
|
|
delete_todo = on_command("删除待办", rule=to_me(), priority=10, block=True)
|
||
|
|
@delete_todo.handle()
|
||
|
|
async def del_todo(message: MessageEvent):
|
||
|
|
member_openid = message.get_user_id()
|
||
|
|
del_line_str = message.get_plaintext().replace("/删除待办", "").strip(" ")
|
||
|
|
try:
|
||
|
|
del_line_num = int(del_line_str)
|
||
|
|
except BaseException:
|
||
|
|
await delete_todo.finish("请检查您的输入是否正确。\n请输入 /删除待办+数字 哦。")
|
||
|
|
result = await ToDoList.delete_user_todo(member_openid, del_line_num)
|
||
|
|
if result == -1:
|
||
|
|
await delete_todo.finish("您还未创建过待办哦。")
|
||
|
|
elif result == 0:
|
||
|
|
await delete_todo.finish("成功删除第" + str(del_line_num) + "条待办。")
|
||
|
|
elif result == 1:
|
||
|
|
await delete_todo.finish("没有找到这条待办哦，请检查您的输入是否正确.")
|