时间戳

首页 > 解决方案 > 将自定义参数传递给棉花糖模式和预处理方法

python - 将自定义参数传递给棉花糖模式和预处理方法

问题描述

我是marshamallow的新用户，并尝试使用 Schema 在 Python 中验证灵活的 JSON/dict 记录。实例化marshamallow Schema时有什么方法可以传递自定义参数？另外，如何将自定义参数传递给架构的 pre_load 方法？我的要求如下


    from marshmallow import Schema, fields, post_load, EXCLUDE, validate, \
    validates, pre_load

    class UserSchema(Schema):
      name = fields.Str()
      joined_on = fields.AwareDateTime()

      @pre_load
      def sanitize(self, data, **kwargs):
          tzinfo = kwargs.get('tzinfo')
          data['joined_on'] = tzinfo.localize(data['joined_on'])
          return data

     schema = UserSchema()
     user = schema.load({"name": "Tim", "joined_on": datetime.datetime(2019, 10, 23)}, tzinfo=pytz.utc)

标签： pythonmarshmallow

解决方案

自定义参数不能直接传递给棉花糖load函数，但它们可以作为键值对传递给函数的数据参数，与装饰器的参数load一起。pass_original=Truepost_load

解决方案：

import pytz


from marshmallow import (
    fields,
    post_load,
    EXCLUDE,
    Schema,
  )


class UserSchema(Schema):
    name = fields.Str()
    joined_on = fields.AwareDateTime(required=True)

    @post_load(pass_original=True)
    def sanitize(self, data, original_data, **_):
        tzinfo = original_data.get('tzinfo', pytz.utc)
        data['joined_on'] = tzinfo.localize(data['joined_on'])
        return data


schema = UserSchema()
user = schema.load(
    {
        "name": "Tim",
        "joined_on": datetime.datetime(2019, 10, 23), 
        "tzinfo"=pytz.timezone("Asia/Kolkata")
    },
    unknown=EXCLUDE
)

推荐阅读