zhenxun_bot/plugins/shop/gold.py

from nonebot import on_command
from nonebot.adapters.cqhttp import Bot, GroupMessageEvent
from nonebot.typing import T_State
from nonebot.adapters.cqhttp.permission import GROUP
from utils.data_utils import init_rank
from models.bag_user import BagUser


my_gold = on_command("我的金币", priority=5, block=True, permission=GROUP)

gold_rank = on_command("金币排行", priority=5, block=True, permission=GROUP)


@my_gold.handle()
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
    await my_gold.finish(await BagUser.get_my_total_gold(event.user_id, event.group_id))


@gold_rank.handle()
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
    all_users = await BagUser.get_all_users(event.group_id)
    all_user_id = [user.qq for user in all_users]
    all_user_data = [user.gold for user in all_users]
    await gold_rank.finish(
        "金币排行：\n" + await init_rank(all_user_id, all_user_data, event.group_id)
    )