python - 使用 Marshmallow 序列化对象时如何嵌套一些父对象字段

Question

我正在使用 marshmallow 序列化具有如下平面结构的 sqlalchemy 对象:

class Company(Base):
   __tablename__ = 'Company'
   id = sa.Column(sa.Integer, primary_key=True)
   MainStreetAddress = sa.Column(sa.String)
   MainCity = sa.Column(sa.String)
   MainState = sa.Column(sa.String)
   MainZip = sa.Column(sa.String)
   AltStreetAddress = sa.Column(sa.String)    
   AltCity = sa.Column(sa.String)
   AltState = sa.Column(sa.String)
   AltZip2 = sa.Column(sa.String)

序列化时，我希望 Marshmallow 将相关地址字段分组，如下所示:

{'id':1,
 'main_address':{'street_address':'101 main    st','city':'springfield','state':'IL','zip':'11231'},
 'alt_address':{'street_address':'102 main st','city':'new bedford','state':'MA','zip':'07630'}}

有没有办法让棉花糖在不使用@post_dump的情况下做到这一点？我知道我可以使用 Nested() 嵌套其他模式，但在这种情况下，我只想从父模式本身嵌套某些列。

Answer 1

也许...在您的 Schema 类中添加一个字段。方法:

class Company(Base):
    __tablename__ = 'Company'
    id = sa.Column(sa.Integer, primary_key=True)
    MainStreetAddress = sa.Column(sa.String)
    MainCity = sa.Column(sa.String)
    MainState = sa.Column(sa.String)
    MainZip = sa.Column(sa.String)
    AltStreetAddress = sa.Column(sa.String)    
    AltCity = sa.Column(sa.String)
    AltState = sa.Column(sa.String)
    AltZip2 = sa.Column(sa.String)

class CompanySchema(Schema):
    id = fields.Integer()
    main_address = fields.Method('make_main_address')
    alt_address = fields.Method('make_alt_address')

    def make_main_address(self, obj)
        return {'main_address':{'street_address':obj.MainStreetAddress,'city':obj.MainCity ,'state':obj.MainState,'zip':obj.MainZip}}

    def make_alt_address(self, obj)
        return {'alt_address':{'street_address':obj.AltStreetAddress,'city':obj.AltCity ,'state':obj.AltState,'zip':obj.AltZip2}}

看看Marshmallow Documentation