feat✨: CSGO开箱

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='左轮武器箱'",
+        ]