棉花糖结果定制

Question

我有带有 jsonb 字段的 sqlalchemy 模型和该模型的棉花糖模式：

class Settings(db.Model):
    id = db.Column(UUID, primary_key=True,
                   server_default=text("uuid_generate_v4()"))
    settings = db.Column(JSONB)

class SettingsSchema(ma.ModelSchema):
    class Meta:
        model = Settings

我在 json 中将其序列化为：

settings = Settings(settings={'qwerty': 'test'})
settings_schema = SettingsSchema(many=False, exclude=('id', ))
body = settings_schema.dump(settings).data

和 json 模型视图如下所示：

{
  "settings": {
    "qwerty": "test"
  }
}

当我将此架构用作嵌套时，我得到的结果类似于

{
    "id": 123456,
    "some_field": "some_field_value",
    "other_field": "other_field_value",
    "settings": {
        "settings": {
            "qwerty": "test"
        }
    }
}

这个"settings.settings"看起来很难看。

有没有一种方法对棉花糖说我只需要值作为转储结果，我的意思是

{
    "qwerty": "test"
}

可能我需要用元类做点什么，但我不知道该怎么做。

Answer 1

我明白了。我应该在我的架构中使用@post_dump：

class SettingsSchema(ma.ModelSchema):

    class Meta:
        model = WorkspaceSettings

    @post_dump(pass_many=True)
    def unwrap_settings(self, data, many):

        if many:
            return [item['settings'] or {} for item in data]
        else:
            return data['settings'] or {}

棉花糖结果定制

Marshmallow result customization

python

sqlalchemy

psycopg2

marshmallow