zhenxun_bot/plugins/gold_redbag/model.py

from typing import List

from tortoise import fields

from services.db_context import Model


class RedbagUser(Model):

    id = fields.IntField(pk=True, generated=True, auto_increment=True)
    """自增id"""
    user_qq = fields.BigIntField()
    """用户id"""
    group_id = fields.BigIntField()
    """群聊id"""
    send_redbag_count = fields.IntField(default=0)
    """发送红包次数"""
    get_redbag_count = fields.IntField(default=0)
    """开启红包次数"""
    spend_gold = fields.IntField(default=0)
    """发送红包花费金额"""
    get_gold = fields.IntField(default=0)
    """开启红包获取金额"""

    class Meta:
        table = "redbag_users"
        table_description = "红包统计数据表"
        unique_together = ("user_qq", "group_id")

    @classmethod
    async def add_redbag_data(
        cls, user_qq: int, group_id: int, i_type: str, money: int
    ):
        """
        说明:
            添加收发红包数据
        参数:
            :param user_qq: qq号
            :param group_id: 群号
            :param i_type: 收或发
            :param money: 金钱数量
        """

        user, _ = await cls.get_or_create(user_qq=user_qq, group_id=group_id)
        if i_type == "get":
            user.get_redbag_count = user.get_redbag_count + 1
            user.get_gold = user.get_gold + money
        else:
            user.send_redbag_count = user.send_redbag_count + 1
            user.spend_gold = user.spend_gold + money
        await user.save(
            update_fields=[
                "get_redbag_count",
                "get_gold",
                "send_redbag_count",
                "spend_gold",
            ]
        )