feat✨: 更新内置插件

2026-10-14 08:30:01 +08:00 · 2024-02-25 03:18:34 +08:00
parent 131200a28e
commit eb0572ea77
83 changed files with 7588 additions and 450 deletions
@@ -0,0 +1,162 @@
+import uuid
+from typing import Dict
+
+from tortoise import fields
+from typing_extensions import Self
+
+from zhenxun.services.db_context import Model
+
+
+class GoodsInfo(Model):
+    __tablename__ = "goods_info"
+
+    id = fields.IntField(pk=True, generated=True, auto_increment=True)
+    """自增id"""
+    uuid = fields.CharField(255, null=True)
+    """uuid"""
+    goods_name = fields.CharField(255, unique=True)
+    """商品名称"""
+    goods_price = fields.IntField()
+    """价格"""
+    goods_description = fields.TextField()
+    """描述"""
+    goods_discount = fields.FloatField(default=1)
+    """折扣"""
+    goods_limit_time = fields.BigIntField(default=0)
+    """限时"""
+    daily_limit = fields.IntField(default=0)
+    """每日限购"""
+    is_passive = fields.BooleanField(default=False)
+    """是否为被动道具"""
+    icon = fields.TextField(null=True)
+    """图标路径"""
+
+    class Meta:
+        table = "goods_info"
+        table_description = "商品数据表"
+
+    @classmethod
+    async def add_goods(
+        cls,
+        goods_name: str,
+        goods_price: int,
+        goods_description: str,
+        goods_discount: float = 1,
+        goods_limit_time: int = 0,
+        daily_limit: int = 0,
+        is_passive: bool = False,
+        icon: str | None = None,
+    ):
+        """添加商品
+
+        参数:
+            goods_name: 商品名称
+            goods_price: 商品价格
+            goods_description: 商品简介
+            goods_discount: 商品折扣
+            goods_limit_time: 商品限时
+            daily_limit: 每日购买限制
+            is_passive: 是否为被动道具
+            icon: 图标
+        """
+        if not await cls.exists(goods_name=goods_name):
+            await cls.create(
+                uuid=uuid.uuid1(),
+                goods_name=goods_name,
+                goods_price=goods_price,
+                goods_description=goods_description,
+                goods_discount=goods_discount,
+                goods_limit_time=goods_limit_time,
+                daily_limit=daily_limit,
+                is_passive=is_passive,
+                icon=icon,
+            )
+
+    @classmethod
+    async def delete_goods(cls, goods_name: str) -> bool:
+        """删除商品
+
+        参数:
+            goods_name: 商品名称
+
+        返回:
+            bool: 是否删除成功
+        """
+        if goods := await cls.get_or_none(goods_name=goods_name):
+            await goods.delete()
+            return True
+        return False
+
+    @classmethod
+    async def update_goods(
+        cls,
+        goods_name: str,
+        goods_price: int | None = None,
+        goods_description: str | None = None,
+        goods_discount: float | None = None,
+        goods_limit_time: int | None = None,
+        daily_limit: int | None = None,
+        is_passive: bool | None = None,
+        icon: str | None = None,
+    ):
+        """更新商品信息
+
+        参数:
+            goods_name: 商品名称
+            goods_price: 商品价格
+            goods_description: 商品简介
+            goods_discount: 商品折扣
+            goods_limit_time: 商品限时时间
+            daily_limit: 每日次数限制
+            is_passive: 是否为被动
+            icon: 图标
+        """
+        if goods := await cls.get_or_none(goods_name=goods_name):
+            await cls.update_or_create(
+                goods_name=goods_name,
+                defaults={
+                    "goods_price": goods_price or goods.goods_price,
+                    "goods_description": goods_description or goods.goods_description,
+                    "goods_discount": goods_discount or goods.goods_discount,
+                    "goods_limit_time": (
+                        goods_limit_time
+                        if goods_limit_time is not None
+                        else goods.goods_limit_time
+                    ),
+                    "daily_limit": (
+                        daily_limit if daily_limit is not None else goods.daily_limit
+                    ),
+                    "is_passive": (
+                        is_passive if is_passive is not None else goods.is_passive
+                    ),
+                    "icon": icon or goods.icon,
+                },
+            )
+
+    @classmethod
+    async def get_all_goods(cls) -> list[Self]:
+        """
+        获得全部有序商品对象
+        """
+        query = await cls.all()
+        id_lst = [x.id for x in query]
+        goods_lst = []
+        for _ in range(len(query)):
+            min_id = min(id_lst)
+            goods_lst.append([x for x in query if x.id == min_id][0])
+            id_lst.remove(min_id)
+        return goods_lst
+
+    @classmethod
+    async def _run_script(cls):
+        if goods_list := await cls.filter(uuid__isnull=True).all():
+            for goods in goods_list:
+                goods.uuid = uuid.uuid1()
+            await cls.bulk_update(goods_list, ["uuid"], 10)
+        return [
+            "ALTER TABLE goods_info ADD uuid VARCHAR(255);",
+            "ALTER TABLE goods_info ADD daily_limit Integer DEFAULT 0;",
+            "ALTER TABLE goods_info ADD is_passive boolean DEFAULT False;",
+            "ALTER TABLE goods_info ADD icon VARCHAR(255);",
+            "ALTER TABLE goods_info DROP daily_purchase_limit;",  # 删除 daily_purchase_limit 字段
+        ]