如何在 Flask-Marshmallow 中将整数字段设置为可选?

时间:2021-03-02 16:39:18

标签: python flask marshmallow marshmallow-sqlalchemy

我有以下模型及其对应的架构。如果我尝试将该字段留空,则 sold_price 字段会导致问题。我提出了错误“不是有效整数”,这在技术上是正确的,但我需要将该字段设为可选/允许为空。

# Models.py
class Product(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    sold_price = db.Column(db.Integer)

# Marshmallow
class ProductSchema(SQLAlchemyAutoSchema):
    class Meta:
        model = Product
    id = auto_field(dump_only=True)
    sold_price = auto_field()

我设法找到了一个次优的解决方案,但我觉得必须有更好的方法。次优解决方案会影响自定义字段:

class OptionalInteger(fields.Field):

    def _serialize(self, value, attr, obj, **kwargs):
        if value is None:
            return ''
        return value

    def _deserialize(self, value, attr, data, **kwargs):
        if value:
            try:
                return int(value)
            except ValueError as error:
                raise ValidationError("Must a integer") from error
        return value

是否有任何其他想法可以使 sell_price (int) 字段成为可选字段?

1 个答案:

答案 0 :(得分:0)

我的猜测是您输入数据中的字段没有丢失,而是 null。这是典型的 HTML 表单,其中字段留空。

默认情况下,DB字段不可为空,所以生成的字段

  • 不是必需的 (required = False)
  • 允许 None (allow_none = True)

这发生在这里:https://github.com/marshmallow-code/marshmallow-sqlalchemy/blob/6e43a8357a012fb08ee1ec32e67c07679a97b917/src/marshmallow_sqlalchemy/convert.py#L264-L266

您可以通过打印来检查

print(ProductSchema().fields['sold_price'].required)
print(ProductSchema().fields['sold_price'].allow_none)

如果没有显示验证错误的代码示例,我无法解释您的问题。