mirror of
https://github.com/ClovertaTheTrilobita/SanYeCao-Nonebot.git
synced 2026-04-01 22:04:51 +00:00
61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
|
from tortoise import fields
|
||
|
|
from typing_extensions import Self
|
||
|
|
from src.my_sqlite.data_init.db_connect import Model
|
||
|
|
|
||
|
|
|
||
|
|
class ToDoList(Model):
|
||
|
|
id = fields.IntField(primary_key=True, generated=True, auto_increment=True)
|
||
|
|
user_id = fields.CharField(max_length=100,description="用户member_openid")
|
||
|
|
content = fields.TextField()
|
||
|
|
|
||
|
|
class Meta:
|
||
|
|
table = "user_todo_list"
|
||
|
|
table_description = "用户待办表"
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
async def _get_data(cls, user_id: str | None) -> list | None:
|
||
|
|
"""
|
||
|
|
获取对应用户的待办
|
||
|
|
"""
|
||
|
|
if not user_id:
|
||
|
|
print("用户id为空")
|
||
|
|
else:
|
||
|
|
return (
|
||
|
|
await cls.filter(user_id=user_id).order_by("id").values_list("id","user_id","content")
|
||
|
|
)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
async def get_todo_list(cls, user_id: str | None) -> Self | None:
|
||
|
|
todo_table = await cls._get_data(user_id)
|
||
|
|
todo_list = [row[2] for row in todo_table]
|
||
|
|
# print(todo_list)
|
||
|
|
if todo_list:
|
||
|
|
return todo_list
|
||
|
|
else:
|
||
|
|
return False
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
async def insert_todo_list(cls, user_id: str | None, content: str | None):
|
||
|
|
data = {
|
||
|
|
"user_id": user_id,
|
||
|
|
"content": content
|
||
|
|
}
|
||
|
|
await cls.create(**data)
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
async def delete_user_todo(cls, user_id: str | None, del_line_num: int | None) -> int:
|
||
|
|
todo_table = await cls._get_data(user_id)
|
||
|
|
|
||
|
|
if todo_table is None:
|
||
|
|
return -1
|
||
|
|
|
||
|
|
max_length = len(todo_table)
|
||
|
|
if del_line_num > max_length or del_line_num < 1:
|
||
|
|
return 1
|
||
|
|
|
||
|
|
del_id = todo_table[del_line_num - 1][0]
|
||
|
|
await cls.filter(id=del_id).delete()
|
||
|
|
return 0
|
||
|
|
|
||
|
|
|