zhenxun_bot/plugins/shop/gold.py
2021-06-15 10:57:08 +08:00

29 lines
868 B
Python

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 util.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):
users = await BagUser.get_user_all(event.group_id)
all_user_data = [user.gold for user in users]
await gold_rank.finish(await init_rank(users, all_user_data, event.group_id))