Merge branch 'master' into master

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "assets/lexicon/THUOCL"]
+	path = assets/lexicon/THUOCL
+	url = https://github.com/thunlp/THUOCL.git

konabot/plugins/idiomgame/__init__.py

@@ -0,0 +1,169 @@
+import base64
+import secrets
+import json
+from typing import Literal
+
+from nonebot import on_message
+from nonebot.adapters import Event as BaseEvent
+from nonebot.adapters.console.event import MessageEvent as ConsoleMessageEvent
+from nonebot.adapters.discord.event import MessageEvent as DiscordMessageEvent
+from nonebot_plugin_alconna import (Alconna, Args, Field, Subcommand,
+                                    UniMessage, UniMsg, on_alconna)
+
+from konabot.common.path import ASSETS_PATH
+
+ALL_WORDS = [] # 所有四字词语
+ALL_IDIOMS = [] # 所有成语
+IDIOM_FIRST_CHAR = {} # 成语首字字典
+
+INITED = False
+
+def init_lexicon():
+    global ALL_WORDS, ALL_IDIOMS, IDIOM_FIRST_CHAR
+    # 成语大表
+    with open(ASSETS_PATH / "lexicon" / "idiom.json", "r", encoding="utf-8") as f:
+        ALL_IDIOMS_INFOS = json.load(f)
+
+    # 词语大表
+    with open(ASSETS_PATH / "lexicon" / "ci.json", "r", encoding="utf-8") as f:
+        ALL_WORDS = json.load(f)
+
+    # 读取 THUOCL 成语库
+    with open(ASSETS_PATH / "lexicon" / "THUOCL" / "data" / "THUOCL_chengyu.txt", "r", encoding="utf-8") as f:
+        THUOCL_IDIOMS = [line.split(" ")[0].strip() for line in f]
+
+    # 读取 THUOCL 剩下的所有 txt 文件，只保留四字词
+    THUOCL_WORDS = []
+    import os
+    for filename in os.listdir(ASSETS_PATH / "lexicon" / "THUOCL" / "data"):
+        if filename.endswith(".txt") and filename != "THUOCL_chengyu.txt":
+            with open(ASSETS_PATH / "lexicon" / "THUOCL" / "data" / filename, "r", encoding="utf-8") as f:
+                for line in f:
+                    word = line.lstrip().split(" ")[0].strip()
+                    if len(word) == 4:
+                        THUOCL_WORDS.append(word)
+
+
+    # 只有成语的大表
+    ALL_IDIOMS = [idiom["word"] for idiom in ALL_IDIOMS_INFOS] + THUOCL_IDIOMS
+    ALL_IDIOMS = list(set(ALL_IDIOMS)) # 去重
+
+    # 其他四字词语表，仅表示可以有这个词
+    ALL_WORDS = [word for word in ALL_WORDS if len(word) == 4] + THUOCL_WORDS
+    ALL_WORDS = list(set(ALL_WORDS)) # 去重
+
+    # 根据成语大表，划分出成语首字字典
+    IDIOM_FIRST_CHAR = {}
+    for idiom in ALL_IDIOMS + ALL_WORDS:
+        if idiom[0] not in IDIOM_FIRST_CHAR:
+            IDIOM_FIRST_CHAR[idiom[0]] = []
+        IDIOM_FIRST_CHAR[idiom[0]].append(idiom)
+
+NOW_PLAYING = False
+
+SCORE_BOARD = {}
+
+LAST_CHAR = ""
+
+USER_NAME_CACHE = {} # 缓存用户名称，避免多次获取
+
+evt = on_alconna(Alconna(
+    "我要玩成语接龙"
+), use_cmd_start=True, use_cmd_sep=False, skip_for_unmatch=True)
+
+@evt.handle()
+async def _(event: BaseEvent):
+    global NOW_PLAYING, LAST_CHAR, INITED
+    if not INITED:
+        init_lexicon()
+        INITED = True
+    if NOW_PLAYING:
+        await evt.send(await UniMessage().text("当前已有成语接龙游戏在进行中，请稍后再试！").export())
+        return
+    NOW_PLAYING = True
+    await evt.send(await UniMessage().text("你小子，还真有意思！\n好，成语接龙游戏开始！我说一个成语，请大家接下去！").export())
+    # 选择一个随机成语
+    idiom = secrets.choice(ALL_IDIOMS)
+    LAST_CHAR = idiom[-1]
+    # 发布成语
+    await evt.send(await UniMessage().text(f"第一个成语：「{idiom}」，请接！").export())
+
+evt = on_alconna(Alconna(
+    "不玩了"
+), use_cmd_start=True, use_cmd_sep=False, skip_for_unmatch=True)
+
+@evt.handle()
+async def _(event: BaseEvent):
+    global NOW_PLAYING, SCORE_BOARD, LAST_CHAR
+    if NOW_PLAYING:
+        NOW_PLAYING = False
+        # 发送好吧狗图片
+        # 打开好吧狗本地文件
+        with open(ASSETS_PATH / "img" / "dog" / "haoba_dog.jpg", "rb") as f:
+            img_data = f.read()
+        await evt.send(await UniMessage().image(raw=img_data).export())
+        result_text = UniMessage().text("游戏结束！\n最终得分榜：\n")
+        if len(SCORE_BOARD) == 0:
+            result_text += "无人得分！"
+        else:
+            # 按分数排序，名字用 at 的方式
+            sorted_score = sorted(SCORE_BOARD.items(), key=lambda x: x[1]["score"], reverse=True)
+            for i, (user_id, info) in enumerate(sorted_score):
+                result_text += f"{i+1}. " + UniMessage().at(user_id) + f" - {info['score']} 分\n"
+        await evt.send(await result_text.export())
+        # 重置分数板
+        SCORE_BOARD = {}
+        LAST_CHAR = ""
+    else:
+        await evt.send(await UniMessage().text("当前没有成语接龙游戏在进行中！").export())
+
+# 跳过
+evt = on_alconna(Alconna(
+    "跳过成语"
+), use_cmd_start=True, use_cmd_sep=False, skip_for_unmatch=True)
+
+@evt.handle()
+async def _(event: BaseEvent):
+    global NOW_PLAYING, LAST_CHAR
+    if not NOW_PLAYING:
+        return
+    await evt.send(await UniMessage().text("你们太菜了！全部扣100分！").export())
+    for user_id in SCORE_BOARD:
+        SCORE_BOARD[user_id]["score"] -= 100
+    # 选择下一个成语
+    idiom = secrets.choice(ALL_IDIOMS)
+    LAST_CHAR = idiom[-1]
+    await evt.send(await UniMessage().text(f"重新开始，下一个成语是「{idiom}」").export())
+
+# 直接读取消息
+evt = on_message()
+
+@evt.handle()
+async def _(event: BaseEvent, msg: UniMsg):
+    global NOW_PLAYING, LAST_CHAR, SCORE_BOARD
+    if not NOW_PLAYING:
+        return
+    user_idiom = msg.extract_plain_text().strip()
+    if(user_idiom[0] != LAST_CHAR):
+        return
+    if(user_idiom not in ALL_IDIOMS and user_idiom not in ALL_WORDS):
+        await evt.send(await UniMessage().text("接不上！这个不一样！").export())
+        return
+    # 成功接上
+    if isinstance(event, DiscordMessageEvent):
+        user_id = str(event.author.id)
+        user_name = str(event.author.name)
+    else:
+        user_id = str(event.get_user_id())
+        user_name = str(event.get_user_id())
+
+    if user_id not in SCORE_BOARD:
+        SCORE_BOARD[user_id] = {
+            "name": user_name,
+            "score": 0
+        }
+    SCORE_BOARD[user_id]["score"] += 1
+    # at 指定玩家
+    await evt.send(await UniMessage().at(user_id).text(f"接对了！你有 {SCORE_BOARD[user_id]['score']} 分！").export())
+    LAST_CHAR = user_idiom[-1]
+    await evt.send(await UniMessage().text(f"下一个成语请以「{LAST_CHAR}」开头！").export())

konabot/plugins/idiomgame/base/path.py

@@ -0,0 +1,3 @@
+from pathlib import Path
+
+ASSETS = Path(__file__).parent.parent / "assets"

konabot/plugins/roll_dice/roll_dice.py

@@ -309,7 +309,7 @@ async def generate_dice_image(number: str) -> BytesIO:
     if(len(text) > 50):
         output = BytesIO()
         push_image = Image.open(ASSETS_PATH / "img" / "dice" / "stick.png")
-        push_image.save(output,format='PNG')
+        push_image.save(output,format='GIF')
         output.seek(0)
         return output