Вложенный класс в десериализации JSON, используя зефир - PullRequest
Новая
       25

Вложенный класс в десериализации JSON, используя зефир

0 голосов
/ 09 марта 2019

Как бы я десериализовал вложенный JSON, используя зефир, чтобы я мог использовать точечную нотацию, такую ​​как app.data.person.lname?В настоящее время мой пример работает только на один уровень ниже, но я не могу получить фамилию (lname) из вложенной структуры person: 

from marshmallow import Schema, fields, post_load
import datetime as dt
import json

class Person(object):
    def __init__(self, fname, lname):
        self.fname = fname
        self.lname = lname

class PersonSchema(Schema):
    fname = fields.Str()
    lname = fields.Str()

class App(object):
    def __init__(self, appid, channel, person):
        self.appid = appid
        self.channel = channel
        self.person = person
        self.created_at = dt.datetime.now()

class AppSchema(Schema):
    appid = fields.Str()
    channel = fields.Str()
    person = fields.Nested(PersonSchema)
    created_at = fields.DateTime()

    @post_load
    def make_user(self, data):
        return App(**data)

json_data = """{
    "appid": "2309wfjwef",
    "channel": "retail",
    "person": {
        "fname": "John",
        "lname": "Doe"
        }
}"""

app_data = json.loads(json_data)

schema = AppSchema()
app = schema.load(app_data)

print(app.data.person.lname)

Получение: 

AttributeError: 'dict' object has no attribute 'lname'

1 Ответ

0 голосов
/ 10 марта 2019

Очевидно, вам нужно создать Person в PersonSchema: 

from marshmallow import Schema, fields, post_load
import datetime as dt
import json

class Person(object):
    def __init__(self, fname, lname):
        self.fname = fname
        self.lname = lname

class PersonSchema(Schema):
    fname = fields.Str()
    lname = fields.Str()

    @post_load
    def make_person(self, data):
        return Person(**data)


class App(object):
    def __init__(self, appid, channel, person):
        self.appid = appid
        self.channel = channel
        self.person = person
        self.created_at = dt.datetime.now()

class AppSchema(Schema):
    appid = fields.Str()
    channel = fields.Str()
    person = fields.Nested(PersonSchema)
    created_at = fields.DateTime()

    @post_load
    def make_app(self, data):
        return App(**data)

json_data = """{
    "appid": "2309wfjwef",
    "channel": "retail",
    "person": {
        "fname": "John",
        "lname": "Doe"
        }
}"""

app_data = json.loads(json_data)

schema = AppSchema()
app = schema.load(app_data)

print(app.data.person.fname)
...