From 061f4460ddf93a4096666e57e41e347ada348177 Mon Sep 17 00:00:00 2001 From: SlyAimer <2289782085@qq.com> Date: Thu, 23 Jan 2025 12:10:06 +0800 Subject: [PATCH] =?UTF-8?q?rua=20=E5=A4=B4=E6=94=B9=E4=B8=BA=20tortoise?= =?UTF-8?q?=EF=BC=8C=E5=88=A0=E9=99=A4=E5=A4=9A=E4=BD=99=E7=9A=84=E5=88=9D?= =?UTF-8?q?=E5=A7=8B=E5=8C=96=E7=B1=BB?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- bot.py | 2 - src/my_sqlite/admin_manage_by_sqlite.py | 62 ----------- src/my_sqlite/data_init/admin_init.py | 23 ---- src/my_sqlite/data_init/data_init.py | 59 ----------- src/my_sqlite/data_init/touch_init_data.py | 79 -------------- src/my_sqlite/models/fortune.py | 3 + src/my_sqlite/models/touch.py | 116 +++++++++++++++++++++ src/my_sqlite/touch_by_sqlite.py | 83 --------------- src/qq_plugins/codeforces.py | 61 +++++------ src/qq_plugins/touch.py | 43 +++----- 10 files changed, 163 insertions(+), 368 deletions(-) delete mode 100644 src/my_sqlite/admin_manage_by_sqlite.py delete mode 100644 src/my_sqlite/data_init/admin_init.py delete mode 100644 src/my_sqlite/data_init/touch_init_data.py create mode 100644 src/my_sqlite/models/touch.py delete mode 100644 src/my_sqlite/touch_by_sqlite.py diff --git a/bot.py b/bot.py index 8cc61e5..1aacfc4 100644 --- a/bot.py +++ b/bot.py @@ -13,9 +13,7 @@ driver.on_shutdown(disconnect) def init_all(): # 初始化数据库 - data_init.touch_init() data_init.todo_init() - data_init.admin_init() if __name__ == "__main__": init_all() diff --git a/src/my_sqlite/admin_manage_by_sqlite.py b/src/my_sqlite/admin_manage_by_sqlite.py deleted file mode 100644 index 47b3796..0000000 --- a/src/my_sqlite/admin_manage_by_sqlite.py +++ /dev/null @@ -1,62 +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 insert_administrator(member_openid): - session = SqliteSqlalchemy().session - session.execute(insertAdminID, {'member_openid': member_openid}) - session.commit() - session.close() - -def check_admin_access(member_openid): - session = SqliteSqlalchemy().session - result = session.execute(selectAdminID, {'member_openid': member_openid}).fetchone() - session.close() - return result - - -def select_status(group_id): - session = SqliteSqlalchemy().session - result = session.execute(selectIsOn, {'group_id': group_id}).fetchone() - session.close() - return result - -def update_ai_availability(group_id, is_on): - session = SqliteSqlalchemy().session - result = session.execute(selectGroupID, {'group_id': group_id}).fetchone() - if result is None: - session.execute(insertGroupID, {'group_id': group_id, 'is_on': is_on}) - session.commit() - else: - session.execute(updateIsOn, {'group_id': group_id, 'is_on': is_on}) - session.commit() - session.close() - -# 插入数据 -insertAdminID = text("INSERT INTO admin_list (user_id) VALUES (:member_openid)") - -# 查找用户 -selectAdminID = text("SELECT * FROM admin_list WHERE user_id = :member_openid") - -# 更新数据 -selectIsOn = text("SELECT is_on FROM group_list WHERE group_id = :group_id") - -# 更新数据 -updateIsOn = text("UPDATE group_list SET is_on = :is_on WHERE group_id = :group_id") - -# 查询是否已存该群 -selectGroupID = text("SELECT * FROM group_list WHERE group_id = :group_id") - -# 存储群聊ID -insertGroupID = text("INSERT INTO group_list (group_id, is_on) VALUES (:group_id, :is_on)") - - -if __name__ == "__main__": - insert_administrator('1234') - print(check_admin_access('1234')) \ No newline at end of file diff --git a/src/my_sqlite/data_init/admin_init.py b/src/my_sqlite/data_init/admin_init.py deleted file mode 100644 index ed722c5..0000000 --- a/src/my_sqlite/data_init/admin_init.py +++ /dev/null @@ -1,23 +0,0 @@ -import sqlite3 - -# 连接到数据库，如果不存在则创建 -conn = sqlite3.connect('chat_bot.db') -# 创建游标 -c = conn.cursor() -# 创建表 -c.execute(""" - -CREATE TABLE IF NOT EXISTS admin_list ( - user_id VARCHAR(100) PRIMARY KEY -); - -""") - -c.execute(""" - -CREATE TABLE IF NOT EXISTS group_list ( - group_id VARCHAR(100) PRIMARY KEY, - is_on BOOLEAN DEFAULT false -); - -""") \ No newline at end of file diff --git a/src/my_sqlite/data_init/data_init.py b/src/my_sqlite/data_init/data_init.py index 57108f0..bcaae6a 100644 --- a/src/my_sqlite/data_init/data_init.py +++ b/src/my_sqlite/data_init/data_init.py @@ -12,31 +12,6 @@ class SqliteSqlalchemy(object): self.session = sessionmaker(bind=engine)() -def touch_init(): - session = SqliteSqlalchemy().session - # 检查某个表是否存在 - table_exists = session.execute(selectQrTouch).fetchone() - if table_exists: - return print("摸摸功能已初始化。") - else: - print("摸摸功能未初始化，开始执行初始化文件。") - execute_init_file2() - return "" - -""" -执行初始化文件touch_init_data.py -""" -def execute_init_file2(): - # 拼接文件的完整路径 - file_path = os.getcwd() + "/src/my_sqlite/data_init/touch_init_data.py" - init_file_path = os.path.join(os.path.dirname(__file__), file_path) - try: - # 执行初始化文件 - subprocess.run(["python", init_file_path], check=True) - print("初始化文件已成功执行。") - except subprocess.CalledProcessError as e: - print(f"执行初始化文件时出错: {e}") - def todo_init(): session = SqliteSqlalchemy().session @@ -66,36 +41,6 @@ def execute_init_file3(): print(f"执行初始化文件时出错: {e}") -def admin_init(): - session = SqliteSqlalchemy().session - # 检查某个表是否存在 - table_exists = session.execute(selectAdminList).fetchone() - table_exists2 = session.execute(selectGroupList).fetchone() - if table_exists and table_exists2: - return print("管理员表和群聊表状态正常。") - else: - print("管理员表或群聊表不存在，开始执行初始化文件。") - execute_init_file4() - return "" - - -def execute_init_file4(): - # 拼接文件的完整路径 - file_path = os.getcwd() + "/src/my_sqlite/data_init/admin_init.py" - init_file_path = os.path.join(os.path.dirname(__file__), file_path) - try: - # 执行初始化文件 - subprocess.run(["python", init_file_path], check=True) - print("初始化文件已成功执行。") - except subprocess.CalledProcessError as e: - print(f"执行初始化文件时出错: {e}") - - - - -# 查询初始化表是否存在 -selectQrTouch = text( - "SELECT name FROM sqlite_master WHERE type='table' AND name='qr_touch';") # 查询待办表是否存在 selectTodoTable = text( @@ -103,7 +48,3 @@ selectTodoTable = text( selectUserList = text( "SELECT name FROM sqlite_master WHERE type='table' AND name='user_list';") -# 查询管理员表是否存在 -selectAdminList = text("SELECT name FROM sqlite_master WHERE type='table' AND name='admin_list';") -# 查询群聊表是否存在 -selectGroupList = text("SELECT name FROM sqlite_master WHERE type='table' AND name='group_list';") \ No newline at end of file diff --git a/src/my_sqlite/data_init/touch_init_data.py b/src/my_sqlite/data_init/touch_init_data.py deleted file mode 100644 index e65383e..0000000 --- a/src/my_sqlite/data_init/touch_init_data.py +++ /dev/null @@ -1,79 +0,0 @@ -import sqlite3 - -# 连接到数据库，如果不存在则创建 -conn = sqlite3.connect('chat_bot.db') -# 创建游标 -c = conn.cursor() -# -c.execute("""DROP TABLE IF EXISTS qr_touch; """) -c.execute("""DROP TABLE IF EXISTS qu_touch_log; """) -# 创建表 -c.execute(""" - -CREATE TABLE qr_touch ( - id INTEGER PRIMARY KEY AUTOINCREMENT, - touch_status INTEGER NOT NULL, - reply_touch_content TEXT NOT NULL -); - """) - -c.execute(""" -CREATE TABLE qr_touch_log ( - id INTEGER PRIMARY KEY AUTOINCREMENT, - touch_status INTEGER NOT NULL, - reply_touch_content TEXT NOT NULL, - user_id TEXT NOT NULL, - extract_time DATE -); - """) - - -# 插入数据 -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'哼！你怎么突然摸我的头，把我吓了一跳！我刚刚整理好的毛发都被你弄乱了啦，你得负责帮我重新梳理好哦，喵！');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'哼，真是的，又来摸本喵的头。虽然…… 虽然也不是很讨厌，但你可别得寸进尺哦，不然我就用爪子抓你啦！喵～');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'喵～你在摸我的头呢，好舒服呀！感觉就像被柔软的云朵轻轻拂过一样呢 再用点力也没关系哦，我可是很享受的呢。');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'嘿，人类，你以为摸我的头就能讨好我啦？不过看在你手法还不错的份上，这次就允许你啦，喵～但要记得给我准备小鱼干哦。');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'喵呜～别停呀，你的摸头动作真是太棒啦，让我想起了妈妈舔我的感觉呢，我都快融化啦～');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'嘻嘻，你在给我挠痒痒呢，真是太有趣啦！我要在你身边转圈圈啦，多摸摸我的头，我会更加爱你的哦，喵～');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'哼，虽然我是一只高傲的猫，但被你摸摸头的感觉还不赖哦 不过，你可得小心点，要是把我惹毛了，我可是会亮出爪子的哦，喵！');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'喵～你的手好温暖呀，每次被摸摸头，我就觉得自己是世界上最幸福的猫咪啦，希望你能一直这样摸下去呢。');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'喵！你在摸我的头，这是在向我表示友好吗？那我就勉为其难地接受啦，要是能再给我一些猫薄荷就更好啦～');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'呜呜～被摸摸头的感觉好温馨哦，我都想在你的腿上睡一觉啦，记得要一直摸着我，不然我会生气的哦，喵～');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'嘿，你的摸头让我感觉自己像个小宝贝呢，我会用最可爱的样子来报答你哦，多摸摸我的头，让我更加可爱吧，喵～');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'喵～我喜欢这种被抚摸的感觉呢，就像沉浸在温暖的阳光中，你可以多摸摸我的头，这样我会更有活力哦，说不定会帮你抓到更多老鼠呢。');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'喵呜～好啦好啦，我已经被你摸得服服帖帖啦，感觉自己的毛都变得更加顺滑啦，继续保持哦，我会更听你的话呢。');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'你在摸我的头，是不是觉得我很可爱呀 我可以让你多摸一会儿，不过作为交换，你要陪我玩捉迷藏哦，喵～');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'喵～谢谢你的摸头啦，你真是个好人呢，我会给你带来好运的哦，就像招财猫一样，只要你继续给我摸摸头，喵～');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'嘿，你这个铲屎官，摸头的动作还挺熟练嘛，我允许你再摸一会儿啦，不过可别想随便摸我的尾巴哦，喵！');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'喵～你的摸头就像一场温柔的冒险，让我沉浸其中呢，希望你能多摸摸我，我会带你去探索很多有趣的地方哦。');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'喵呜～被摸摸头真是太美妙啦，感觉自己的九条命都变得更加精彩啦，你是我最喜欢的人类啦，继续给我摸摸头吧。');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'哼，虽然我很傲娇，但我不得不承认，被你摸摸头的感觉真的很棒哦，不过，你可别太得意啦，喵～');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'喵～我感觉自己像一只被宠爱的小猫咪，你是我最信任的人啦，多摸摸我的头，我会在你身边撒娇哦。');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'呜呜～被摸摸头的时候，我觉得自己是最幸福的啦，希望你能一直守护我，就像我守护自己的小鱼干一样，喵～');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'嘿，你在摸我的头呢，是不是在给我施展魔法呀 我已经被你的魔法控制啦，要一直摸下去哦，喵～');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'喵～每次你摸我的头，我就会变得更加慵懒哦，我会躺在你的怀里，享受这种惬意的时光，希望你能多陪陪我呢。');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'喵呜～你的摸头让我变得更加放松啦，我会像猫精灵一样为你带来快乐，只要你不停止给我摸头哦，喵～');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'嘿，别以为摸我的头是一件简单的事情哦，这可是需要技巧的呢，不过你现在做得还不错啦，继续加油哦，喵～');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'喵～我要把我的头靠在你手上啦，这样你就可以更方便地摸我啦，我会发出咕噜咕噜的声音来表示我的快乐哦。');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (0,'呜呜～你是在给我按摩吗？我太喜欢啦，我会用最可爱的动作来感谢你，多摸摸我的头，我会更有精神哦，喵～');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (1,'喵！你这家伙到底有完没完？！再敢碰我的头一下，我就把这房间里的东西全部抓烂，让你见识见识我的厉害！');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (1,'嘶嘶 —— 离我远点！你这不知死活的家伙，摸我的头？你是想尝尝我的爪子有多锋利吗？我可不会手下留情！');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (1,'喵呜！！我警告你，立刻停止你的动作！你要是再敢侵犯我的领地（头部），我就趁你睡觉的时候在你脸上留下几道漂亮的爪痕，让你知道我的愤怒！');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (1,'哼！你以为你是谁？竟敢摸本大爷 / 大小姐的头！我现在就去把你的鞋子咬得稀巴烂，看你还敢不敢乱动！');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (1,'喵！！！我已经忍无可忍了！你这无礼的行为简直不可饶恕，我要发动我的猫之诅咒，让你接下来的日子诸事不顺，除非你马上把手拿开！');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (1,'别逼我使出绝招！你再摸一下试试，我就把你心爱的那些小物件一个个从桌子上推下去摔碎，让你为你的鲁莽付出代价！');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (1,'喵！你是不是瞎啊？没看到我浑身的毛都因为生气竖起来了吗？还不把手缩回去，我马上就扑上去咬断你的手指！');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (1,'嘶 —— 我可不是好惹的！你这愚蠢的人类，摸我的头就像在挑战我的威严，我会让你知道什么叫恐惧，等着瞧吧！');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (1,'喵呜！！你触碰到了我的底线，我现在就去把你藏在角落里的零食找出来吃掉，一颗都不给你留，这就是你摸我头的下场！');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (1,'哼，摸什么摸呀，烦死了！你就不能让我清净会儿吗？别老想着动手动脚的，真当我好脾气呢！');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (1,'啧，又来摸我的头，你是有多无聊啊！我可没那闲工夫陪你玩这幼稚的把戏，离我远点啦！');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (1,'够了够了！别碰我头了行不行，每次都这样，我都快被你摸得不耐烦了，再这样我可真要发火了啊！');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (1,'我警告你最后一次，别碰我的头！你这没完没了的样子真的太讨厌了，小心我对你不客气！');") -c.execute("insert into qr_touch (touch_status,reply_touch_content) values (1,'啊！你怎么还摸啊，我都想抓狂了，你是听不懂人话吗？再摸我就把你手给剁了，烦死了！');") - - -# 提交事务 -conn.commit() -# 关闭游标和连接 -c.close() -conn.close() diff --git a/src/my_sqlite/models/fortune.py b/src/my_sqlite/models/fortune.py index 44b6504..65d4cef 100644 --- a/src/my_sqlite/models/fortune.py +++ b/src/my_sqlite/models/fortune.py @@ -34,6 +34,9 @@ class QrFortune(Model): class QrFortuneLog(Model): + """ + 获取运势日志表 + """ id = fields.IntField(primary_key=True, generated=True, auto_increment=True) fortune_summary = fields.CharField(max_length=64, description="运势总结", null=True) lucky_star = fields.CharField(max_length=64, description="幸运星", null=True) diff --git a/src/my_sqlite/models/touch.py b/src/my_sqlite/models/touch.py new file mode 100644 index 0000000..edd2cac --- /dev/null +++ b/src/my_sqlite/models/touch.py @@ -0,0 +1,116 @@ +from datetime import datetime +import random +from tortoise import fields +from typing_extensions import Self +from src.my_sqlite.data_init.db_connect import Model + + +class QrTouch(Model): + """ + 摸头回复内容 + """ + id = fields.IntField(primary_key=True, generated=True, auto_increment=True) + touch_status = fields.IntField(default=0,description="状态", null=True) + reply_touch_content = fields.CharField(max_length=255, description="响应内容", null=True) + class Meta: + # 指定表名 + table = "qr_touch" + table_description = "rua回复内容" + + @classmethod + async def touch(cls, status: int | None)-> Self | None: + """ + 获取摸头内容 + :param status: + :return: + """ + existing_record = await QrTouch.filter(touch_status=status).all() + if not existing_record: + # 执行初始化 + await cls.bulk_create(touch_initial_data) + existing_record = await QrTouch.filter(touch_status=status).all() + return random.choice(existing_record) + + +class QrTouchLog(Model): + """ + 摸头记录 + """ + id = fields.IntField(primary_key=True, generated=True, auto_increment=True) + touch_status = fields.IntField(default=0,description="状态", null=True) + reply_touch_content = fields.CharField(max_length=255, description="响应内容", null=True) + user_id = fields.CharField(max_length=64, description="用户id", null=True) + extract_time = fields.DateField(description="日期", null=True) + class Meta: + # 指定表名 + table = "qr_touch_log" + table_description = "rua回复内容" + + @classmethod + async def insert_touch_log(cls, model:QrTouch, member_openid): + """ + 插入摸头记录 + :param model: + :param member_openid: + :return: + """ + data = { + "touch_status": model.touch_status, + "reply_touch_content": model.reply_touch_content, + "user_id": member_openid, + "extract_time": datetime.now().date() + } + await cls.create(**data) + + @classmethod + async def touch_count(cls, member_openid): + """ + 获取摸头次数 + :param member_openid: + :return: + """ + result = await QrTouchLog.filter(user_id=member_openid, extract_time=datetime.now().date()).count() + return result + +touch_initial_data = [QrTouch(touch_status=0,reply_touch_content="哼！你怎么突然摸我的头，把我吓了一跳！我刚刚整理好的毛发都被你弄乱了啦，你得负责帮我重新梳理好哦，喵！"), + QrTouch(touch_status=0,reply_touch_content="哼，真是的，又来摸本喵的头。虽然……虽然也不是很讨厌，但你可别得寸进尺哦，不然我就用爪子抓你啦！喵～"), + QrTouch(touch_status=0,reply_touch_content="喵～你在摸我的头呢，好舒服呀！感觉就像被柔软的云朵轻轻拂过一样呢再用点力也没关系哦，我可是很享受的呢。"), + QrTouch(touch_status=0,reply_touch_content="嘿，人类，你以为摸我的头就能讨好我啦？不过看在你手法还不错的份上，这次就允许你啦，喵～但要记得给我准备小鱼干哦。"), + QrTouch(touch_status=0,reply_touch_content="喵呜～别停呀，你的摸头动作真是太棒啦，让我想起了妈妈舔我的感觉呢，我都快融化啦～"), + QrTouch(touch_status=0,reply_touch_content="嘻嘻，你在给我挠痒痒呢，真是太有趣啦！我要在你身边转圈圈啦，多摸摸我的头，我会更加爱你的哦，喵～"), + QrTouch(touch_status=0,reply_touch_content="哼，虽然我是一只高傲的猫，但被你摸摸头的感觉还不赖哦不过，你可得小心点，要是把我惹毛了，我可是会亮出爪子的哦，喵！"), + QrTouch(touch_status=0,reply_touch_content="喵～你的手好温暖呀，每次被摸摸头，我就觉得自己是世界上最幸福的猫咪啦，希望你能一直这样摸下去呢。"), + QrTouch(touch_status=0,reply_touch_content="喵！你在摸我的头，这是在向我表示友好吗？那我就勉为其难地接受啦，要是能再给我一些猫薄荷就更好啦～"), + QrTouch(touch_status=0,reply_touch_content="呜呜～被摸摸头的感觉好温馨哦，我都想在你的腿上睡一觉啦，记得要一直摸着我，不然我会生气的哦，喵～"), + QrTouch(touch_status=0,reply_touch_content="嘿，你的摸头让我感觉自己像个小宝贝呢，我会用最可爱的样子来报答你哦，多摸摸我的头，让我更加可爱吧，喵～"), + QrTouch(touch_status=0,reply_touch_content="喵～我喜欢这种被抚摸的感觉呢，就像沉浸在温暖的阳光中，你可以多摸摸我的头，这样我会更有活力哦，说不定会帮你抓到更多老鼠呢。"), + QrTouch(touch_status=0,reply_touch_content="喵呜～好啦好啦，我已经被你摸得服服帖帖啦，感觉自己的毛都变得更加顺滑啦，继续保持哦，我会更听你的话呢。"), + QrTouch(touch_status=0,reply_touch_content="你在摸我的头，是不是觉得我很可爱呀我可以让你多摸一会儿，不过作为交换，你要陪我玩捉迷藏哦，喵～"), + QrTouch(touch_status=0,reply_touch_content="喵～谢谢你的摸头啦，你真是个好人呢，我会给你带来好运的哦，就像招财猫一样，只要你继续给我摸摸头，喵～"), + QrTouch(touch_status=0,reply_touch_content="嘿，你这个铲屎官，摸头的动作还挺熟练嘛，我允许你再摸一会儿啦，不过可别想随便摸我的尾巴哦，喵！"), + QrTouch(touch_status=0,reply_touch_content="喵～你的摸头就像一场温柔的冒险，让我沉浸其中呢，希望你能多摸摸我，我会带你去探索很多有趣的地方哦。"), + QrTouch(touch_status=0,reply_touch_content="喵呜～被摸摸头真是太美妙啦，感觉自己的九条命都变得更加精彩啦，你是我最喜欢的人类啦，继续给我摸摸头吧。"), + QrTouch(touch_status=0,reply_touch_content="哼，虽然我很傲娇，但我不得不承认，被你摸摸头的感觉真的很棒哦，不过，你可别太得意啦，喵～"), + QrTouch(touch_status=0,reply_touch_content="喵～我感觉自己像一只被宠爱的小猫咪，你是我最信任的人啦，多摸摸我的头，我会在你身边撒娇哦。"), + QrTouch(touch_status=0,reply_touch_content="呜呜～被摸摸头的时候，我觉得自己是最幸福的啦，希望你能一直守护我，就像我守护自己的小鱼干一样，喵～"), + QrTouch(touch_status=0,reply_touch_content="嘿，你在摸我的头呢，是不是在给我施展魔法呀我已经被你的魔法控制啦，要一直摸下去哦，喵～"), + QrTouch(touch_status=0,reply_touch_content="喵～每次你摸我的头，我就会变得更加慵懒哦，我会躺在你的怀里，享受这种惬意的时光，希望你能多陪陪我呢。"), + QrTouch(touch_status=0,reply_touch_content="喵呜～你的摸头让我变得更加放松啦，我会像猫精灵一样为你带来快乐，只要你不停止给我摸头哦，喵～"), + QrTouch(touch_status=0,reply_touch_content="嘿，别以为摸我的头是一件简单的事情哦，这可是需要技巧的呢，不过你现在做得还不错啦，继续加油哦，喵～"), + QrTouch(touch_status=0,reply_touch_content="喵～我要把我的头靠在你手上啦，这样你就可以更方便地摸我啦，我会发出咕噜咕噜的声音来表示我的快乐哦。"), + QrTouch(touch_status=0,reply_touch_content="呜呜～你是在给我按摩吗？我太喜欢啦，我会用最可爱的动作来感谢你，多摸摸我的头，我会更有精神哦，喵～"), + QrTouch(touch_status=1,reply_touch_content="喵！你这家伙到底有完没完？！再敢碰我的头一下，我就把这房间里的东西全部抓烂，让你见识见识我的厉害！"), + QrTouch(touch_status=1,reply_touch_content="嘶嘶——离我远点！你这不知死活的家伙，摸我的头？你是想尝尝我的爪子有多锋利吗？我可不会手下留情！"), + QrTouch(touch_status=1,reply_touch_content="喵呜！！我警告你，立刻停止你的动作！你要是再敢侵犯我的领地（头部），我就趁你睡觉的时候在你脸上留下几道漂亮的爪痕，让你知道我的愤怒！"), + QrTouch(touch_status=1,reply_touch_content="哼！你以为你是谁？竟敢摸本大爷/大小姐的头！我现在就去把你的鞋子咬得稀巴烂，看你还敢不敢乱动！"), + QrTouch(touch_status=1,reply_touch_content="喵！！！我已经忍无可忍了！你这无礼的行为简直不可饶恕，我要发动我的猫之诅咒，让你接下来的日子诸事不顺，除非你马上把手拿开！"), + QrTouch(touch_status=1,reply_touch_content="别逼我使出绝招！你再摸一下试试，我就把你心爱的那些小物件一个个从桌子上推下去摔碎，让你为你的鲁莽付出代价！"), + QrTouch(touch_status=1,reply_touch_content="喵！你是不是瞎啊？没看到我浑身的毛都因为生气竖起来了吗？还不把手缩回去，我马上就扑上去咬断你的手指！"), + QrTouch(touch_status=1,reply_touch_content="嘶——我可不是好惹的！你这愚蠢的人类，摸我的头就像在挑战我的威严，我会让你知道什么叫恐惧，等着瞧吧！"), + QrTouch(touch_status=1,reply_touch_content="喵呜！！你触碰到了我的底线，我现在就去把你藏在角落里的零食找出来吃掉，一颗都不给你留，这就是你摸我头的下场！"), + QrTouch(touch_status=1,reply_touch_content="哼，摸什么摸呀，烦死了！你就不能让我清净会儿吗？别老想着动手动脚的，真当我好脾气呢！"), + QrTouch(touch_status=1,reply_touch_content="啧，又来摸我的头，你是有多无聊啊！我可没那闲工夫陪你玩这幼稚的把戏，离我远点啦！"), + QrTouch(touch_status=1,reply_touch_content="够了够了！别碰我头了行不行，每次都这样，我都快被你摸得不耐烦了，再这样我可真要发火了啊！"), + QrTouch(touch_status=1,reply_touch_content="我警告你最后一次，别碰我的头！你这没完没了的样子真的太讨厌了，小心我对你不客气！"), + QrTouch(touch_status=1,reply_touch_content="啊！你怎么还摸啊，我都想抓狂了，你是听不懂人话吗？再摸我就把你手给剁了，烦死了！"), + ] \ No newline at end of file diff --git a/src/my_sqlite/touch_by_sqlite.py b/src/my_sqlite/touch_by_sqlite.py deleted file mode 100644 index 090fded..0000000 --- a/src/my_sqlite/touch_by_sqlite.py +++ /dev/null @@ -1,83 +0,0 @@ -from sqlalchemy import Column, Integer, String, Date, create_engine, text -from sqlalchemy.orm import declarative_base, sessionmaker - - -def touch_count(member_openid): - session = SqliteSqlalchemy().session - # - result = session.execute(selectTouchCount, - {'member_openid': member_openid}).scalar() - session.close() - return result - - -def touch(status): - session = SqliteSqlalchemy().session - result = session.execute(selectTouchContent,{'status': status}).fetchone() - session.close() - return result - - -def insert_touch_log(QrTouchLog): - session = SqliteSqlalchemy().session - session.execute(insertTouchLog, - {'touch_status': QrTouchLog.touch_status, - 'reply_touch_content': QrTouchLog.reply_touch_content, - 'member_openid': QrTouchLog.user_id}) - session.commit() - session.close() - return "" - -# 申明基类对象 -Base = declarative_base() - -class QrTouch: - __tablename__ = 'qr_touch' - id = Column(Integer, primary_key=True) - touch_status = Column(Integer) - reply_touch_content = Column(String(255)) - - def __repr__(self): - return ( - "QrTouch(id:{},touch_status:{},reply_touch_content:{})" .format( - self.id, - self.touch_status, - self.reply_touch_content - )) - -class QrTouchLog: - __tablename__ = 'qr_touch_log' - id = Column(Integer, primary_key=True) - touch_status = Column(Integer) - reply_touch_content = Column(String(255)) - user_id = Column(String(255)) - extract_time = Column(Date) - - def __repr__(self): - return ( - "QrFortune(id:{},touch_status:{},reply_touch_content:{},user_id:{})" .format( - self.id, - self.touch_status, - self.reply_touch_content, - self.user_id)) - - -class SqliteSqlalchemy(object): - def __init__(self): - # 创建Sqlite连接引擎 - engine = create_engine('sqlite:///./chat_bot.db', echo=True) - # 创建Sqlite的session连接对象 - self.session = sessionmaker(bind=engine)() - - -# 查询初始化表是否存在 -selectInit = text( - "SELECT name FROM sqlite_master WHERE type='table' AND name='qr_touch';") -# 查询回复内容 -selectTouchContent = text("select * from qr_touch where touch_status = :status order by random() limit 1") -# 查询触摸次数 -selectTouchCount = text( - "select count(*) from qr_touch_log where user_id = :member_openid and extract_time = date('now')") -# 插入日志表 -insertTouchLog = text( - "insert into qr_touch_log (touch_status, reply_touch_content,user_id,extract_time) values (:touch_status, :reply_touch_content, :member_openid,date('now'))") diff --git a/src/qq_plugins/codeforces.py b/src/qq_plugins/codeforces.py index 1514b0a..db25802 100644 --- a/src/qq_plugins/codeforces.py +++ b/src/qq_plugins/codeforces.py @@ -8,46 +8,39 @@ async def get_cf_rounds(): await cf_query.send("正在为您整理近期比赛信息哦~\n请稍等💭💡🎈") try: result = requests.get('https://codeforces.com/api/contest.list?gym=false').json() - except: + except BaseException: await cf_query.finish("API请求失败，这绝对不是咱的错，绝对不是！") - - print("正在请求codeforces比赛API") - i = False all_matches = "" for matches in result['result']: - if i: - break - if matches["phase"] == "BEFORE": - phase = "未开始" - elif matches["phase"] == "FINISHED": - phase = "已结束" - elif matches["phase"] == "CODING": - phase = "进行中" - elif matches["phase"] == "PENDING_SYSTEM_TEST": - phase = "等待判题" - elif matches["phase"] == "SYSTEM_TEST": - phase = "判题中" - elif matches["phase"] == "FINISHED": - phase = "已结束" - else: - phase = "未知" - one_match = ("\n比赛：" + str(matches["name"]) + - "\n状态：" + phase + - "\n时长：" + str(int(matches["durationSeconds"]) / 3600) + "h\n") + phase = get_match_phase(matches["phase"]) + one_match = ("\n比赛：" + str(matches["name"]) + "\n状态：" + phase + "\n时长：" + str(int(matches["durationSeconds"]) / 3600) +"h\n") all_matches = "".join([all_matches, one_match]) - if phase == "未开始": until_start_time_min = 0 - int(matches["relativeTimeSeconds"]) / 60 - if until_start_time_min <= 180: - until_start = "距开始：" + str(int(until_start_time_min)) + "min\n" - elif 180 < until_start_time_min <= 1440: - until_start = "距开始：" + str(int(until_start_time_min / 60)) + "h\n" - elif until_start_time_min > 1440: - until_start = "距开始：" + str(int(until_start_time_min / 60 / 24)) + "days\n" - else: - until_start = "距开始：未知\n" + until_start = get_until_start_time(until_start_time_min) all_matches = "".join([all_matches, until_start]) if matches["phase"] == "FINISHED": - i = True - # print(all_matches) + break await cf_query.finish(all_matches) + + +def get_match_phase(phase): + phase_map = { + "BEFORE": "未开始", + "FINISHED": "已结束", + "CODING": "进行中", + "PENDING_SYSTEM_TEST": "等待判题", + "SYSTEM_TEST": "判题中", + } + return phase_map.get(phase, "未知") + + +def get_until_start_time(until_start_time_min): + if until_start_time_min <= 180: + return "距开始：" + str(int(until_start_time_min)) + "min\n" + elif 180 < until_start_time_min <= 1440: + return "距开始：" + str(int(until_start_time_min / 60)) + "h\n" + elif until_start_time_min > 1440: + return "距开始：" + str(int(until_start_time_min / 60 / 24)) + "days\n" + else: + return "距开始：未知\n" diff --git a/src/qq_plugins/touch.py b/src/qq_plugins/touch.py index 698f8fd..211943b 100644 --- a/src/qq_plugins/touch.py +++ b/src/qq_plugins/touch.py @@ -1,37 +1,28 @@ from pathlib import Path from nonebot.rule import to_me from nonebot.plugin import on_command -from nonebot.adapters.qq import Message,MessageEvent,MessageSegment -from src.my_sqlite.touch_by_sqlite import touch_count, QrTouchLog, insert_touch_log, touch -from src.image.get_image import download_qq_image_by_account,qq_image_delete,rua +from nonebot.adapters.qq import Message, MessageEvent, MessageSegment +from src.my_sqlite.models.touch import QrTouch, QrTouchLog +from src.image.get_image import download_qq_image_by_account, qq_image_delete, rua -to = on_command("摸摸头",rule=to_me(),priority=10,block=True) +to = on_command("摸摸头", rule=to_me(), priority=10, block=True) @to.handle() -async def handle_touch(event: MessageEvent): - - member_openid = event.get_user_id() - num = touch_count(member_openid) - q = QrTouchLog() - q.user_id = member_openid - if num > 10 : +async def handle_touch(message: MessageEvent): + member_openid = message.get_user_id() + num = await QrTouchLog.touch_count(member_openid) + if num > 10: await to.finish("你今天已经摸了太多次了，请明天再吧！") - elif num > 5: - result = touch(1) - q.touch_status = 1 - q.reply_touch_content = result.reply_touch_content + q = QrTouch() + q.touch_status = 1 + if num > 5: + result = await QrTouch.touch(1) else: - result = touch(0) - q.touch_status = 1 - q.reply_touch_content = result.reply_touch_content - - insert_touch_log(q) + result = await QrTouch.touch(0) + q.reply_touch_content = result.reply_touch_content + await QrTouchLog.insert_touch_log(q, member_openid) local_gif = rua(download_qq_image_by_account(None)).add_gif() - msg = Message([ - MessageSegment.file_image(Path(local_gif)), - MessageSegment.text(result.reply_touch_content), - ]) + msg = Message([MessageSegment.file_image(Path(local_gif)), + MessageSegment.text(result.reply_touch_content),]) qq_image_delete() await to.finish(msg) - -