67 lines
2.7 KiB
Python
67 lines
2.7 KiB
Python
# -*- 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
|
||
|