FastAPI中SQLModel的可编辑计算字段实现方案问询

要实现基于字段a自动生成b的默认值，同时允许手动覆盖b，最直接的方案是重写模型的__init__方法，在初始化阶段处理逻辑：

from sqlmodel import SQLModel, Field

class MyModel(SQLModel):
    a: str
    b: str | None = Field(default=None)

    def __init__(self, **data):
        # 当未传入b时，根据a的值计算默认值
        if "b" not in data or data["b"] is None:
            a_value = data["a"]
            data["b"] = "option1" if len(a_value) > 5 else "option2"
        # 调用父类初始化方法完成模型创建
        super().__init__(**data)

测试验证 #

# 自动计算b的场景
my_model = MyModel(a="abcde")
print(my_model)
# 输出: MyModel(a='abcde', b='option2')

# 手动指定b的场景
my_model = MyModel(a="some_value", b="option3")
print(my_model)
# 输出: MyModel(a='some_value', b='option3')

方案说明 #

• 将b的默认值设为None，用于区分用户是否主动传入了字段值
• 在__init__中检查data字典里是否有b的有效输入，没有则基于a计算赋值
• 完全兼容SQLModel的原有功能，既支持自动生成默认值，也允许手动覆盖

内容的提问来源于stack exchange，提问作者yael