SanYeCao-Nonebot/src/qq_plugins/to_do.py

from nonebot.adapters.qq import MessageEvent
from nonebot.plugin import on_command
from nonebot.rule import to_me

from src.my_sqlite.todo_by_sqlite import *
from src.my_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 = get_user_todo_list(member_openid)
    result = await ToDoList.get_todo_list(member_openid)
    if result is False:
        await get_todo_list.finish("\n您还未创建待办\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 ToDoList.insert_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))
    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("没有找到这条待办哦，请检查您的输入是否正确.")