用户生成资源容量管控

video-gen-api/app/enums/__init__.py

@@ -10,3 +10,4 @@ from app.enums.generation_task import *
 from app.enums.generation_status import *
 from app.enums.sms import *
 from app.enums.notification import *
+from app.enums.resource_capacity import *

video-gen-api/app/enums/resource_capacity.py

@@ -0,0 +1,68 @@
+from __future__ import annotations
+
+from enum import Enum
+
+
+class ResourceCapacityUnitEnum(str, Enum):
+    """生成资源容量单位。"""
+
+    MB = "MB"
+    GB = "GB"
+    TB = "TB"
+
+    @property
+    def bytes_multiplier(self) -> int:
+        return RESOURCE_CAPACITY_UNIT_BYTES[self]
+
+    @property
+    def label(self) -> str:
+        return RESOURCE_CAPACITY_UNIT_LABELS[self]
+
+
+MB_BYTES = 1048576
+GB_BYTES = 1073741824
+TB_BYTES = 1099511627776
+
+RESOURCE_CAPACITY_UNIT_BYTES: dict[ResourceCapacityUnitEnum, int] = {
+    ResourceCapacityUnitEnum.MB: MB_BYTES,
+    ResourceCapacityUnitEnum.GB: GB_BYTES,
+    ResourceCapacityUnitEnum.TB: TB_BYTES,
+}
+
+RESOURCE_CAPACITY_UNIT_LABELS: dict[ResourceCapacityUnitEnum, str] = {
+    ResourceCapacityUnitEnum.MB: "MB（1048576 字节）",
+    ResourceCapacityUnitEnum.GB: "GB（1073741824 字节）",
+    ResourceCapacityUnitEnum.TB: "TB（1099511627776 字节）",
+}
+
+
+class ResourceCapacitySourceEnum(str, Enum):
+    """最终容量配置来源。"""
+
+    USER = "user"
+    GLOBAL = "global"
+    DISABLED = "disabled"
+
+
+class ResourceCapacityConfigKeyEnum(str, Enum):
+    """SystemConfig 中使用的固定配置 key。"""
+
+    RESOURCE_CAPACITY_LIMIT_CONFIG = "resource_capacity_limit_config"
+
+
+class ResourceCapacityErrorCodeEnum(str, Enum):
+    """资源容量管控错误码。"""
+
+    RESOURCE_CAPACITY_EXCEEDED = "RESOURCE_CAPACITY_EXCEEDED"
+
+
+class ResourceCapacityOperationEnum(str, Enum):
+    """管理后台资源容量配置操作日志动作。"""
+
+    UPDATE_GLOBAL_CONFIG = "更新全局资源容量配置"
+    CREATE_USER_CONFIG = "新增用户资源容量配置"
+    UPDATE_USER_CONFIG = "更新用户资源容量配置"
+    DELETE_USER_CONFIG = "删除用户资源容量配置"
+
+
+RESOURCE_CAPACITY_EXCEEDED_MESSAGE = "您个人的空间容量已超额，请删除素材资源释放容量或购买更高额度的容量套餐"