mirror of
https://github.com/ClovertaTheTrilobita/SanYeCao-Nonebot.git
synced 2026-04-01 22:04:51 +00:00
56 lines
2.2 KiB
Python
56 lines
2.2 KiB
Python
|
from nonebot.rule import to_me
|
||
|
|
from nonebot.plugin import on_command
|
||
|
|
from nonebot.adapters.qq import Message, MessageEvent, MessageSegment
|
||
|
|
from src.my_sqlite.todo_by_sqlite import *
|
||
|
|
|
||
|
|
|
||
|
|
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 = get_user_todo_list(member_openid)
|
||
|
|
if result is False:
|
||
|
|
await get_todo_list.finish("您还未创建任何待办\n尝试使用 /新建待办 创建一份吧")
|
||
|
|
|
||
|
|
todo_list = ""
|
||
|
|
count = 0
|
||
|
|
for each_content in result:
|
||
|
|
count += 1
|
||
|
|
todo_content = "\n" + str(count) + "." + str(each_content).lstrip("('").rstrip("',)") + "\n"
|
||
|
|
todo_list = "".join([todo_list, todo_content])
|
||
|
|
await get_todo_list.finish("您的待办有如下哦：⭐\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(" ")
|
||
|
|
insert_user_todo_list(member_openid, content)
|
||
|
|
await insert_todo.finish("成功添加待办，今后也要加油哦(ง •_•)ง")
|
||
|
|
|
||
|
|
|
||
|
|
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:
|
||
|
|
await delete_todo.finish("请检查您的输入是否正确。\n请输入 /删除待办+数字 哦。")
|
||
|
|
result = delete_user_todo(member_openid, int(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("没有找到这条待办哦，请检查您的输入是否正确.")
|