就像标题所说的那样,我正在尝试为一个我定义了其类的对象编写一个自定义解码器,其中包含其他我定义了该类的对象。 “外部”类是Edge,其定义如下:

class Edge:
    def __init__(self, actor, movie):
        self.actor = actor
        self.movie = movie

    def __eq__(self, other):
        if (self.movie == other.movie) & (self.actor == other.actor):
            return True
        else:
            return False

    def __str__(self):
        print("Actor: ", self.actor, " Movie: ", self.movie)

    def get_actor(self):
        return self.actor

    def get_movie(self):
        return self.movie


演员和电影的“内部”类定义如下:

class Movie:
    def __init__(self, title, gross, soup, year):
        self.title = title
        self.gross = gross
        self.soup = soup
        self.year = year

    def __eq__(self, other):
        if self.title == other.title:
            return True
        else:
            return False

    def __repr__(self):
        return self.title

    def __str__(self):
        return self.title

    def get_gross(self):
        return self.gross

    def get_soup(self):
        return self.soup

    def get_title(self):
        return self.title

    def get_year(self):
        return self.year

class Actor:
    def __init__(self, name, age, soup):
        self.name = name
        self.age = age
        self.soup = soup

    def __eq__(self, other):
        if self.name == other.name:
            return True
        else:
            return False

    def __repr__(self):
        return self.name

    def __str__(self):
        return self.name

    def get_age(self):
        return self.age

    def get_name(self):
        return self.name

    def get_soup(self):
        return self.soup


(汤只是该电影/演员的维基百科页面上的beautifulsoup对象,可以忽略)。
我还为边缘类编写了客户编码器:

class EdgeEncoder(json.JSONEncoder):
    def default(self, o):
        if isinstance(o, Edge):
            return {
                    "Actor": {
                             "Name": o.get_actor().get_name(),
                             "Age": o.get_actor().get_age()
                             },
                    "Movie": {
                             "Title": o.get_movie().get_title(),
                             "Gross": o.get_movie().get_gross(),
                             "Year": o.get_movie().get_year()
                             }
                    }
        return json.JSONEncoder.default(self, o)


我已经测试过,并且可以将边缘列表正确序列化为JSON文件。现在,当我尝试编写边缘解码器时出现了我的问题。我已经使用github页面here作为参考,但是我的编码器偏离了他,并且我想知道是否有必要进行更改。我是否需要像在JSON序列化中那样将对象的类型显式编码为其自身的键值对,或者是否有某种方法可以通过边缘序列化来获取“ Actor”和“ Movie”键?同样,有没有办法获取“名称”。 “年龄”等,这样我就可以重建Actor / Movie对象,然后使用这些对象重建边缘?有没有更好的方法来编码对象呢?我也尝试过遵循this教程,但是我发现对象dict的使用对于其编码器造成了混乱,而且我不确定如何将该方法扩展到包含自定义对象的自定义对象。

最佳答案

您引用的编码器/解码器示例(here)可以轻松扩展,以允许JSON输入/输出中使用不同类型的对象。

但是,如果您只希望一个简单的解码器与您的编码器匹配(仅在JSON文件中编码Edge对象),请使用以下解码器:

class EdgeDecoder(json.JSONDecoder):
    def __init__(self, *args, **kwargs):
        json.JSONDecoder.__init__(self, object_hook=self.object_hook, *args, **kwargs)
    def object_hook(self, dct):
        if 'Actor' in dct:
            actor = Actor(dct['Actor']['Name'], dct['Actor']['Age'], '')
            movie = Movie(dct['Movie']['Title'], dct['Movie']['Gross'], '', dct['Movie']['Year'])
            return Edge(actor, movie)
        return dct


使用问题中的代码定义类MovieActorEdgeEdgeEncoder,以下代码将输出测试文件,然后将其读回:

filename='test.json'
movie = Movie('Python', 'many dollars', '', '2000')
actor = Actor('Casper Van Dien', 49, '')
edge = Edge(actor, movie)
with open(filename, 'w') as jsonfile:
    json.dump(edge, jsonfile, cls=EdgeEncoder)
with open(filename, 'r') as jsonfile:
    edge1 = json.load(jsonfile, cls=EdgeDecoder)
assert edge1 == edge

关于json - 如何为复杂对象编写自定义JSON解码器?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/48991911/

10-12 00:32
查看更多