mirror of
https://github.com/ClovertaTheTrilobita/SanYeCao-Nonebot.git
synced 2026-04-01 22:04:51 +00:00
删除冗余文件
This commit is contained in:
parent
77de9f8362
commit
1d42541475
4 changed files with 0 additions and 145 deletions
5
bot.py
5
bot.py
|
|
@ -1,6 +1,5 @@
|
||||||
import nonebot
|
import nonebot
|
||||||
from nonebot.adapters.qq import Adapter as QQAdapter
|
from nonebot.adapters.qq import Adapter as QQAdapter
|
||||||
from src.my_sqlite.data_init import data_init
|
|
||||||
|
|
||||||
nonebot.init()
|
nonebot.init()
|
||||||
driver = nonebot.get_driver()
|
driver = nonebot.get_driver()
|
||||||
|
|
@ -11,10 +10,6 @@ from src.my_sqlite.data_init.db_connect import disconnect, init
|
||||||
driver.on_startup(init)
|
driver.on_startup(init)
|
||||||
driver.on_shutdown(disconnect)
|
driver.on_shutdown(disconnect)
|
||||||
|
|
||||||
def init_all():
|
|
||||||
# 初始化数据库
|
|
||||||
data_init.todo_init()
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
init_all()
|
|
||||||
nonebot.run()
|
nonebot.run()
|
||||||
|
|
|
||||||
|
|
@ -1,28 +0,0 @@
|
||||||
import sqlite3
|
|
||||||
|
|
||||||
# 连接到数据库,如果不存在则创建
|
|
||||||
conn = sqlite3.connect('chat_bot.db')
|
|
||||||
# 创建游标
|
|
||||||
c = conn.cursor()
|
|
||||||
#
|
|
||||||
# c.execute("""DROP TABLE IF EXISTS user_todo_list; """)
|
|
||||||
# c.execute("""DROP TABLE IF EXISTS user_list; """)
|
|
||||||
# 创建表
|
|
||||||
c.execute("""
|
|
||||||
|
|
||||||
CREATE TABLE IF NOT EXISTS user_list (
|
|
||||||
user_id VARCHAR(100) PRIMARY KEY
|
|
||||||
);
|
|
||||||
|
|
||||||
""")
|
|
||||||
|
|
||||||
c.execute("""
|
|
||||||
|
|
||||||
CREATE TABLE IF NOT EXISTS user_todo_list (
|
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
||||||
user_id VARCHAR(100),
|
|
||||||
content TEXT,
|
|
||||||
FOREIGN KEY (user_id) REFERENCES user_list(user_id)
|
|
||||||
);
|
|
||||||
""")
|
|
||||||
|
|
||||||
|
|
@ -1,110 +0,0 @@
|
||||||
from sqlalchemy import create_engine, text
|
|
||||||
from sqlalchemy.orm import sessionmaker
|
|
||||||
|
|
||||||
|
|
||||||
class SqliteSqlalchemy(object):
|
|
||||||
def __init__(self):
|
|
||||||
# 创建Sqlite连接引擎
|
|
||||||
engine = create_engine('sqlite:///./chat_bot.db', echo=True)
|
|
||||||
# 创建Sqlite的session连接对象
|
|
||||||
self.session = sessionmaker(bind=engine)()
|
|
||||||
|
|
||||||
|
|
||||||
def user_exists(member_openid):
|
|
||||||
"""
|
|
||||||
查询数据库中是否包括此用户。
|
|
||||||
|
|
||||||
:param member_openid:
|
|
||||||
:return: 若包含用户返回True,否则返回False
|
|
||||||
"""
|
|
||||||
session = SqliteSqlalchemy().session
|
|
||||||
result = session.execute(selectUserId,
|
|
||||||
{'member_openid': member_openid}).fetchone()
|
|
||||||
session.close()
|
|
||||||
if result is None:
|
|
||||||
return False
|
|
||||||
else:
|
|
||||||
return True
|
|
||||||
|
|
||||||
|
|
||||||
def insert_user_todo_list(member_openid, content):
|
|
||||||
"""
|
|
||||||
插入用户待办
|
|
||||||
|
|
||||||
:param member_openid:
|
|
||||||
:param content: 待办内容(str)
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
session = SqliteSqlalchemy().session
|
|
||||||
if not user_exists(member_openid):
|
|
||||||
session.execute(insertUserId, {'member_openid': member_openid})
|
|
||||||
|
|
||||||
session.execute(insertUserTodo, {'member_openid': member_openid, 'content': content})
|
|
||||||
session.commit()
|
|
||||||
session.close()
|
|
||||||
|
|
||||||
|
|
||||||
def get_user_todo_list(member_openid):
|
|
||||||
"""
|
|
||||||
展示用户所有待办
|
|
||||||
|
|
||||||
:param member_openid:
|
|
||||||
:return: 数据库查询结果(Sequence)
|
|
||||||
"""
|
|
||||||
if not user_exists(member_openid):
|
|
||||||
return False
|
|
||||||
session = SqliteSqlalchemy().session
|
|
||||||
result = session.execute(selectUserTodo,
|
|
||||||
{'member_openid': member_openid}).fetchall()
|
|
||||||
session.close()
|
|
||||||
if str(result) == "[]":
|
|
||||||
return False
|
|
||||||
|
|
||||||
return result
|
|
||||||
|
|
||||||
|
|
||||||
def delete_user_todo(member_openid, del_line_num):
|
|
||||||
"""
|
|
||||||
删除某用户指定行数的待办
|
|
||||||
|
|
||||||
:param member_openid:
|
|
||||||
:param del_line_num: 指定的行数(int)
|
|
||||||
:return:
|
|
||||||
"""
|
|
||||||
|
|
||||||
if not user_exists(member_openid):
|
|
||||||
return -1
|
|
||||||
|
|
||||||
session = SqliteSqlalchemy().session
|
|
||||||
todo_list_len = session.execute(checkUserTodoLength, {'member_openid': member_openid}).fetchone()
|
|
||||||
max_length = int(str(todo_list_len).lstrip("(").rstrip(",)"))
|
|
||||||
if del_line_num > max_length or del_line_num < 1:
|
|
||||||
return 1
|
|
||||||
|
|
||||||
session.execute(deleteUserTodo, {'member_openid': member_openid, 'del_line_num': del_line_num})
|
|
||||||
session.commit()
|
|
||||||
session.close()
|
|
||||||
return 0
|
|
||||||
|
|
||||||
# 查找用户
|
|
||||||
selectUserId = text(
|
|
||||||
"select * from user_list where user_id = :member_openid")
|
|
||||||
|
|
||||||
# 插入数据
|
|
||||||
insertUserTodo = text(
|
|
||||||
"INSERT INTO user_todo_list (user_id, content) VALUES (:member_openid, :content)")
|
|
||||||
|
|
||||||
# 插入用户id
|
|
||||||
insertUserId = text("INSERT INTO user_list (user_id) VALUES (:member_openid)")
|
|
||||||
|
|
||||||
# 删除特定行的待办
|
|
||||||
deleteUserTodo = text(
|
|
||||||
"DELETE FROM user_todo_list WHERE id=(SELECT id FROM (SELECT *, ROW_NUMBER() OVER (ORDER BY id) AS row_number FROM (SELECT * FROM user_todo_list WHERE user_id = :member_openid)) WHERE row_number = :del_line_num)")
|
|
||||||
|
|
||||||
# 查询用户待办数
|
|
||||||
checkUserTodoLength = text(
|
|
||||||
"SELECT COUNT(*) FROM user_todo_list WHERE user_id = :member_openid")
|
|
||||||
|
|
||||||
# 查找用户待办
|
|
||||||
selectUserTodo = text(
|
|
||||||
"select content from user_todo_list where user_id = :member_openid")
|
|
||||||
|
|
@ -1,8 +1,6 @@
|
||||||
from nonebot.adapters.qq import MessageEvent
|
from nonebot.adapters.qq import MessageEvent
|
||||||
from nonebot.plugin import on_command
|
from nonebot.plugin import on_command
|
||||||
from nonebot.rule import to_me
|
from nonebot.rule import to_me
|
||||||
|
|
||||||
from src.my_sqlite.todo_by_sqlite import *
|
|
||||||
from src.my_sqlite.models.to_do import ToDoList
|
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 = on_command("待办查询", rule=to_me(), priority=10, block=True, aliases={"代办", "daiban"})
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue