upload project source code

后端源码/yifan.action-ai.cn/app/api/v1/module_common/file/schema.py

@@ -0,0 +1,66 @@
+# -*- coding: utf-8 -*-
+
+from pydantic import BaseModel, ConfigDict, Field, field_validator, model_validator
+from pydantic.alias_generators import to_camel
+
+
+class ImportFieldModel(BaseModel):
+    model_config = ConfigDict(alias_generator=to_camel, from_attributes=True)
+    
+    base_column: str | None = Field(description='数据库字段名', default=None)
+    excel_column: str | None = Field(description='excel字段名', default=None)
+    default_value: str | None = Field(description='默认值', default=None)
+    is_required: bool | None = Field(description='是否必传', default=None)
+    selected: bool | None = Field(description='是否勾选', default=None)
+
+    @model_validator(mode='before')
+    @classmethod
+    def _normalize(cls, data):
+        if isinstance(data, dict):
+            for key in ('base_column', 'excel_column', 'default_value'):
+                val = data.get(key)
+                if isinstance(val, str):
+                    val = val.strip()
+                    if val == '':
+                        val = None
+                    data[key] = val
+            # is_required 兼容转换
+            val = data.get('is_required')
+            if isinstance(val, str):
+                lowered = val.strip().lower()
+                if lowered in {'true', '1', 'y', 'yes'}:
+                    data['is_required'] = True
+                elif lowered in {'false', '0', 'n', 'no'}:
+                    data['is_required'] = False
+        return data
+
+    @model_validator(mode='after')
+    def _validate(self):
+        if self.selected and not (self.base_column and self.base_column.strip()):
+            raise ValueError('选中字段必须提供数据库字段名')
+        if self.is_required and not (self.excel_column and self.excel_column.strip()):
+            raise ValueError('必传字段必须提供excel字段名')
+        return self
+
+
+class ImportModel(BaseModel):
+    model_config = ConfigDict(alias_generator=to_camel, from_attributes=True)
+
+    table_name: str | None = Field(description='表名', default=None)
+    sheet_name: str | None = Field(description='Sheet名', default=None)
+    filed_info: list[ImportFieldModel] | None = Field(description='字段关联表', default=None)
+    file_name: str | None = Field(description='文件名', default=None)
+
+    @model_validator(mode='after')
+    def _validate(self):
+        # excel_column 不重复（忽略 None）
+        if self.filed_info:
+            seen = set()
+            for f in self.filed_info:
+                if f.excel_column:
+                    key = f.excel_column.strip()
+                    if key in seen:
+                        raise ValueError('excel字段名存在重复')
+                    seen.add(key)
+        return self
+