feat✨: CSGO开箱

zhenxun/plugins/open_cases/models/buff_prices.py

@@ -0,0 +1,22 @@
+from tortoise import fields
+
+from zhenxun.services.db_context import Model
+
+# 1.狂牙武器箱
+
+
+class BuffPrice(Model):
+
+    id = fields.IntField(pk=True, generated=True, auto_increment=True)
+    """自增id"""
+    case_id = fields.IntField()
+    """箱子id"""
+    skin_name = fields.CharField(255, unique=True)
+    """皮肤名称"""
+    skin_price = fields.FloatField()
+    """皮肤价格"""
+    update_date = fields.DatetimeField()
+
+    class Meta:
+        table = "buff_prices"
+        table_description = "Buff价格数据表"

zhenxun/plugins/open_cases/models/buff_skin.py

@@ -0,0 +1,113 @@
+from tortoise import fields
+from tortoise.contrib.postgres.functions import Random
+
+from zhenxun.services.db_context import Model
+
+
+class BuffSkin(Model):
+
+    id = fields.IntField(pk=True, generated=True, auto_increment=True)
+    """自增id"""
+    case_name: str = fields.CharField(255)  # type: ignore
+    """箱子名称"""
+    name: str = fields.CharField(255)  # type: ignore
+    """武器/手套/刀名称"""
+    skin_name: str = fields.CharField(255)  # type: ignore
+    """皮肤名称"""
+    is_stattrak = fields.BooleanField(default=False)
+    """是否暗金(计数)"""
+    abrasion = fields.CharField(255)
+    """磨损度"""
+    color = fields.CharField(255)
+    """颜色(品质)"""
+    skin_id = fields.CharField(255, null=True, unique=True)
+    """皮肤id"""
+
+    img_url = fields.CharField(255)
+    """图片url"""
+    steam_price = fields.FloatField(default=0)
+    """steam价格"""
+    weapon_type = fields.CharField(255)
+    """枪械类型"""
+    buy_max_price = fields.FloatField(default=0)
+    """最大求购价格"""
+    buy_num = fields.IntField(default=0)
+    """求购数量"""
+    sell_min_price = fields.FloatField(default=0)
+    """售卖最低价格"""
+    sell_num = fields.IntField(default=0)
+    """出售个数"""
+    sell_reference_price = fields.FloatField(default=0)
+    """参考价格"""
+
+    create_time = fields.DatetimeField(auto_add_now=True)
+    """创建日期"""
+    update_time = fields.DatetimeField(auto_add=True)
+    """更新日期"""
+
+    class Meta:
+        table = "buff_skin"
+        table_description = "Buff皮肤数据表"
+        # unique_together = ("case_name", "name", "skin_name", "abrasion", "is_stattrak")
+
+    def __eq__(self, other: "BuffSkin"):
+
+        return self.skin_id == other.skin_id
+
+    def __hash__(self):
+
+        return hash(self.case_name + self.name + self.skin_name + str(self.is_stattrak))
+
+    @classmethod
+    async def random_skin(
+        cls,
+        num: int,
+        color: str,
+        abrasion: str,
+        is_stattrak: bool = False,
+        case_name: str | None = None,
+    ) -> list["BuffSkin"]:  # type: ignore
+        """随机皮肤
+
+        参数:
+            num: 数量
+            color: 品质
+            abrasion: 磨损度
+            is_stattrak: 是否暗金
+            case_name: 箱子名称
+
+        返回:
+            list["BuffSkin"]: 皮肤列表
+        """
+        query = cls
+        if case_name:
+            query = query.filter(case_name__contains=case_name)
+        query = query.filter(abrasion=abrasion, is_stattrak=is_stattrak, color=color)
+        skin_list = await query.annotate(rand=Random()).limit(num)  # type:ignore
+        num_ = num
+        cnt = 0
+        while len(skin_list) < num:
+            cnt += 1
+            num_ = num - len(skin_list)
+            skin_list += await query.annotate(rand=Random()).limit(num_)
+            if cnt > 10:
+                break
+        return skin_list  # type: ignore
+
+    @classmethod
+    async def _run_script(cls):
+        return [
+            "ALTER TABLE buff_skin ADD img_url varchar(255);",  # 新增img_url
+            "ALTER TABLE buff_skin ADD skin_id varchar(255);",  # 新增skin_id
+            "ALTER TABLE buff_skin ADD steam_price float DEFAULT 0;",  # 新增steam_price
+            "ALTER TABLE buff_skin ADD weapon_type varchar(255);",  # 新增type
+            "ALTER TABLE buff_skin ADD buy_max_price float DEFAULT 0;",  # 新增buy_max_price
+            "ALTER TABLE buff_skin ADD buy_num Integer DEFAULT 0;",  # 新增buy_max_price
+            "ALTER TABLE buff_skin ADD sell_min_price float DEFAULT 0;",  # 新增sell_min_price
+            "ALTER TABLE buff_skin ADD sell_num Integer DEFAULT 0;",  # 新增sell_num
+            "ALTER TABLE buff_skin ADD sell_reference_price float DEFAULT 0;",  # 新增sell_reference_price
+            "ALTER TABLE buff_skin DROP COLUMN skin_price;",  # 删除skin_price
+            "alter table buff_skin drop constraint if EXISTS uid_buff_skin_case_na_c35c93;",  # 删除唯一约束
+            "UPDATE buff_skin set case_name='手套' where case_name='手套武器箱'",
+            "UPDATE buff_skin set case_name='左轮' where case_name='左轮武器箱'",
+        ]

zhenxun/plugins/open_cases/models/buff_skin_log.py

@@ -0,0 +1,50 @@
+from tortoise import fields
+
+from zhenxun.services.db_context import Model
+
+
+class BuffSkinLog(Model):
+
+    id = fields.IntField(pk=True, generated=True, auto_increment=True)
+    """自增id"""
+    case_name = fields.CharField(255)
+    """箱子名称"""
+    name = fields.CharField(255)
+    """武器/手套/刀名称"""
+    skin_name = fields.CharField(255)
+    """皮肤名称"""
+    is_stattrak = fields.BooleanField(default=False)
+    """是否暗金(计数)"""
+    abrasion = fields.CharField(255)
+    """磨损度"""
+    color = fields.CharField(255)
+    """颜色(品质)"""
+
+    steam_price = fields.FloatField(default=0)
+    """steam价格"""
+    weapon_type = fields.CharField(255)
+    """枪械类型"""
+    buy_max_price = fields.FloatField(default=0)
+    """最大求购价格"""
+    buy_num = fields.IntField(default=0)
+    """求购数量"""
+    sell_min_price = fields.FloatField(default=0)
+    """售卖最低价格"""
+    sell_num = fields.IntField(default=0)
+    """出售个数"""
+    sell_reference_price = fields.FloatField(default=0)
+    """参考价格"""
+
+    create_time = fields.DatetimeField(auto_add_now=True)
+    """创建日期"""
+
+    class Meta:
+        table = "buff_skin_log"
+        table_description = "Buff皮肤更新日志表"
+
+    @classmethod
+    async def _run_script(cls):
+        return [
+            "UPDATE buff_skin_log set case_name='手套' where case_name='手套武器箱'",
+            "UPDATE buff_skin_log set case_name='左轮' where case_name='左轮武器箱'",
+        ]

zhenxun/plugins/open_cases/models/open_cases_log.py

@@ -0,0 +1,44 @@
+from tortoise import fields
+from tortoise.contrib.postgres.functions import Random
+
+from zhenxun.services.db_context import Model
+
+
+class OpenCasesLog(Model):
+
+    id = fields.IntField(pk=True, generated=True, auto_increment=True)
+    """自增id"""
+    user_id = fields.CharField(255)
+    """用户id"""
+    group_id = fields.CharField(255)
+    """群聊id"""
+    case_name = fields.CharField(255)
+    """箱子名称"""
+    name = fields.CharField(255)
+    """武器/手套/刀名称"""
+    skin_name = fields.CharField(255)
+    """皮肤名称"""
+    is_stattrak = fields.BooleanField(default=False)
+    """是否暗金(计数)"""
+    abrasion = fields.CharField(255)
+    """磨损度"""
+    abrasion_value = fields.FloatField()
+    """磨损数值"""
+    color = fields.CharField(255)
+    """颜色(品质)"""
+    price = fields.FloatField(default=0)
+    """价格"""
+    create_time = fields.DatetimeField(auto_add_now=True)
+    """创建日期"""
+
+    class Meta:
+        table = "open_cases_log"
+        table_description = "开箱日志表"
+
+    @classmethod
+    async def _run_script(cls):
+        return [
+            "ALTER TABLE open_cases_log RENAME COLUMN user_qq TO user_id;",  # 将user_qq改为user_id
+            "ALTER TABLE open_cases_log ALTER COLUMN user_id TYPE character varying(255);",
+            "ALTER TABLE open_cases_log ALTER COLUMN group_id TYPE character varying(255);",
+        ]

zhenxun/plugins/open_cases/models/open_cases_user.py

@@ -0,0 +1,60 @@
+from tortoise import fields
+
+from zhenxun.services.db_context import Model
+
+
+class OpenCasesUser(Model):
+
+    id = fields.IntField(pk=True, generated=True, auto_increment=True)
+    """自增id"""
+    user_id = fields.CharField(255)
+    """用户id"""
+    group_id = fields.CharField(255)
+    """群聊id"""
+    total_count = fields.IntField(default=0)
+    """总开启次数"""
+    blue_count = fields.IntField(default=0)
+    """蓝色"""
+    blue_st_count = fields.IntField(default=0)
+    """蓝色暗金"""
+    purple_count = fields.IntField(default=0)
+    """紫色"""
+    purple_st_count = fields.IntField(default=0)
+    """紫色暗金"""
+    pink_count = fields.IntField(default=0)
+    """粉色"""
+    pink_st_count = fields.IntField(default=0)
+    """粉色暗金"""
+    red_count = fields.IntField(default=0)
+    """紫色"""
+    red_st_count = fields.IntField(default=0)
+    """紫色暗金"""
+    knife_count = fields.IntField(default=0)
+    """金色"""
+    knife_st_count = fields.IntField(default=0)
+    """金色暗金"""
+    spend_money = fields.IntField(default=0)
+    """花费金币"""
+    make_money = fields.FloatField(default=0)
+    """赚取金币"""
+    today_open_total = fields.IntField(default=0)
+    """今日开箱数量"""
+    open_cases_time_last = fields.DatetimeField()
+    """最后开箱日期"""
+    knifes_name = fields.TextField(default="")
+    """已获取金色"""
+
+    class Meta:
+        table = "open_cases_users"
+        table_description = "开箱统计数据表"
+        unique_together = ("user_id", "group_id")
+
+    @classmethod
+    async def _run_script(cls):
+        return [
+            "alter table open_cases_users alter COLUMN make_money type float;",  # 将make_money字段改为float
+            "alter table open_cases_users alter COLUMN spend_money type float;",  # 将spend_money字段改为float
+            "ALTER TABLE open_cases_users RENAME COLUMN user_qq TO user_id;",  # 将user_qq改为user_id
+            "ALTER TABLE open_cases_users ALTER COLUMN user_id TYPE character varying(255);",
+            "ALTER TABLE open_cases_users ALTER COLUMN group_id TYPE character varying(255);",
+        ]