python dict 和 object 相互转换

python dict 和 object 的相互转换

dict.py 借助 dict, isinstance 来实现对象与字典之间的相互转换

def as_dict(obj):
    if not hasattr(obj, "__dict__"):
        return obj
    result = {}
    for key, val in obj.__dict__.items():
        if key.startswith("_"):
            continue
        element = []
        if isinstance(val, list):
            for item in val:
                element.append(as_dict(item))
        else:
            element = as_dict(val)
        result[key] = element
    return result


def as_obj(d):
    if isinstance(d, list):
        d = [as_obj(x) for x in d]
    if not isinstance(d, dict):
        return d

    class C:
        pass

    obj = C()
    for k in d:
        obj.__dict__[k] = as_obj(d[k])
    return obj

test_dict.py

import dict


class Book:
    def __init__(self, name, price, author):
        self.name = name
        self.price = price
        self.author = author


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


def test_as_dict():
    book_dict = dict.as_dict(Book(name="Python crash course", price=29.50, author=Author(name="Phur", age=32)))
    assert book_dict["name"] == "Python crash course" and book_dict["author"]["age"] == 32


def test_as_obj():
    book = dict.as_obj({"name": "Python crash course", "price": 29.50, "author": {"name": "Phur", "age": 32}})
    assert book.name == "Python crash course" and book.author.age == 32