✨ 添加金币排行

Co-Authored-By: HibiKier <45528451+HibiKier@users.noreply.github.com>

zhenxun/builtin_plugins/shop/__init__.py

@@ -4,12 +4,16 @@ from nonebot_plugin_session import EventSession
 from nonebot_plugin_userinfo import UserInfo, EventUserInfo
 from nonebot_plugin_alconna import (
     Args,
+    Query,
+    Option,
     UniMsg,
     Alconna,
     Arparma,
     Subcommand,
     UniMessage,
+    AlconnaQuery,
     on_alconna,
+    store_true,
 )
 
 from zhenxun.services.log import logger
@@ -18,7 +22,7 @@ from zhenxun.utils.exception import GoodsNotFound
 from zhenxun.utils.enum import BlockType, PluginType
 from zhenxun.configs.utils import BaseBlock, PluginExtraData
 
-from ._data_source import ShopManage
+from ._data_source import ShopManage, gold_rank
 
 __plugin_meta__ = PluginMetadata(
     name="商店",
@@ -30,6 +34,8 @@ __plugin_meta__ = PluginMetadata(
         我的道具
         使用道具 [名称/Id]
         购买道具 [名称/Id]
+        金币排行 ?[num=10]
+        金币总排行 ?[num=10]
     """.strip(),
     extra=PluginExtraData(
         author="HibiKier",
@@ -44,10 +50,12 @@ __plugin_meta__ = PluginMetadata(
 _matcher = on_alconna(
     Alconna(
         "商店",
+        Option("--all", action=store_true),
         Subcommand("my-cost", help_text="我的金币"),
         Subcommand("my-props", help_text="我的道具"),
         Subcommand("buy", Args["name", str]["num", int, 1], help_text="购买道具"),
         Subcommand("use", Args["name", str]["num?", int, 1], help_text="使用道具"),
+        Subcommand("gold-list", Args["num?", int], help_text="使用道具"),
     ),
     priority=5,
     block=True,
@@ -81,6 +89,20 @@ _matcher.shortcut(
     prefix=True,
 )
 
+_matcher.shortcut(
+    "金币排行",
+    command="商店",
+    arguments=["gold-list"],
+    prefix=True,
+)
+
+_matcher.shortcut(
+    r"金币总排行",
+    command="商店",
+    arguments=["--all", "gold-list"],
+    prefix=True,
+)
+
 
 @_matcher.assign("$main")
 async def _(session: EventSession, arparma: Arparma):
@@ -153,3 +175,22 @@ async def _(
         await MessageUtils.build_message(f"没有找到道具 {name} 或道具数量不足...").send(
             reply_to=True
         )
+
+
+@_matcher.assign("gold-list")
+async def _(
+    session: EventSession, arparma: Arparma, num: Query[int] = AlconnaQuery("num", 10)
+):
+    if session.id1:
+        gid = session.id3 or session.id2
+        if arparma.find("all"):
+            gid = None
+        result = await gold_rank(session.id1, gid, num.result, session.platform)
+        logger.info(
+            "查看金币排行",
+            arparma.header_result,
+            session=session,
+        )
+        await MessageUtils.build_message(result).send(reply_to=True)
+    else:
+        await MessageUtils.build_message("用户id为空...").send(reply_to=True)

zhenxun/builtin_plugins/shop/_data_source.py

@@ -12,18 +12,29 @@ from nonebot_plugin_alconna import UniMsg, UniMessage
 
 from zhenxun.services.log import logger
 from zhenxun.models.goods_info import GoodsInfo
+from zhenxun.utils.platform import PlatformUtils
+from zhenxun.models.friend_user import FriendUser
 from zhenxun.configs.path_config import IMAGE_PATH
 from zhenxun.models.user_console import UserConsole
 from zhenxun.models.user_gold_log import UserGoldLog
 from zhenxun.utils.enum import GoldHandle, PropHandle
 from zhenxun.models.user_props_log import UserPropsLog
+from zhenxun.models.group_member_info import GroupInfoUser
 from zhenxun.utils.image_utils import BuildImage, ImageTemplate, text2image
 
 ICON_PATH = IMAGE_PATH / "shop_icon"
 
+RANK_ICON_PATH = IMAGE_PATH / "_icon"
+
+PLATFORM_PATH = {
+    "dodo": RANK_ICON_PATH / "dodo.png",
+    "discord": RANK_ICON_PATH / "discord.png",
+    "kaiheila": RANK_ICON_PATH / "kook.png",
+    "qq": RANK_ICON_PATH / "qq.png",
+}
+
 
 class Goods(BaseModel):
-
     name: str
     """商品名称"""
     before_handle: list[Callable] = []
@@ -45,7 +56,6 @@ class Goods(BaseModel):
 
 
 class ShopParam(BaseModel):
-
     goods_name: str
     """商品名称"""
     user_id: int
@@ -70,8 +80,50 @@ class ShopParam(BaseModel):
     """UniMessage"""
 
 
-class ShopManage:
+async def gold_rank(user_id: str, group_id: str | None, num: int, platform: str):
+    query = UserConsole
+    if group_id:
+        uid_list = await GroupInfoUser.filter(group_id=group_id).values_list(
+            "user_id", flat=True
+        )
+        query = query.filter(user_id__in=uid_list)
+    user_list = await query.annotate().order_by("-gold").values_list("user_id", "gold")
+    user_id_list = [user[0] for user in user_list]
+    index = user_id_list.index(user_id) + 1
+    user_list = user_list[:num] if num < len(user_list) else user_list
+    friend_user = await FriendUser.filter(user_id__in=user_id_list).values_list(
+        "user_id", "user_name"
+    )
+    uid2name = {user[0]: user[1] for user in friend_user}
+    if diff_id := set(user_id_list).difference(set(uid2name.keys())):
+        group_user = await GroupInfoUser.filter(user_id__in=diff_id).values_list(
+            "user_id", "user_name"
+        )
+        for g in group_user:
+            uid2name[g[0]] = g[1]
+    column_name = ["排名", "-", "名称", "金币", "平台"]
+    data_list = []
+    for i, user in enumerate(user_list):
+        ava_bytes = await PlatformUtils.get_user_avatar(user[0], platform)
+        data_list.append(
+            [
+                f"{i+1}",
+                (ava_bytes, 30, 30) if platform == "qq" else "",
+                uid2name.get(user[0]),
+                user[1],
+                (PLATFORM_PATH.get(platform), 30, 30),
+            ]
+        )
+    if group_id:
+        title = "金币群组内排行"
+        tip = f"你的排名在本群第 {index} 位哦!"
+    else:
+        title = "金币全局排行"
+        tip = f"你的排名在全局第 {index} 位哦!"
+    return await ImageTemplate.table_page(title, tip, column_name, data_list)
 
+
+class ShopManage:
     uuid2goods: dict[str, Goods] = {}  # noqa: RUF012
 
     @classmethod