如何在数据类模块的 asdict 函数中使用枚举值

Question

我有一个数据类，其字段 template 类型为枚举。使用 asdict 函数时，它将我的数据类转换为字典。是否可以使用FoobarEnum 的value 属性来返回字符串值而不是Enum 对象？

我最初的想法是使用asdict 函数的dict_factory=dict 参数并提供我自己的工厂，但我不知道该怎么做。

from dataclasses import dataclass, asdict
from enum import Enum


@dataclass
class Foobar:
  name: str
  template: "FoobarEnum"


class FoobarEnum(Enum):
  FIRST = "foobar"
  SECOND = "baz"


foobar = Foobar(name="John", template=FoobarEnum.FIRST)

print(asdict(foobar))

当前输出：

{'name': 'John', 'template': <FoobarEnum.FIRST: 'foobar'>}

目标：

{'name': 'John', 'template': 'foobar'}

Answer 1

这不能用标准库来完成，除非我不知道一些元类枚举黑客。 Enum.name 和 Enum.value 是内置的，不应更改。

使用数据类default_factory 的方法也行不通。因为调用default_factory 是为了为数据类成员生成默认值，而不是自定义对成员的访问。

您可以将 Enum 成员或 Enum.value 作为数据类成员，这就是 asdict() 将返回的内容。

如果您想保留一个 Enum 成员 - 而不仅仅是 Enum.value- 作为数据类成员，并且有一个函数将其转换为返回 Enum.value 而不是 Enum 成员的字典，那么正确的方法是正在实现您自己的方法以将数据类作为字典返回。

from dataclasses import dataclass
from enum import Enum


class FoobarEnum(Enum):
    FIRST = "foobar"
    SECOND = "baz"


@dataclass
class Foobar:
    name: str
    template: FoobarEnum

    def as_dict(self):
        return {
            'name': self.name,
            'template': self.template.value
        }


# Testing.
print(Foobar(name="John", template=FoobarEnum.FIRST).as_dict())
# {'name': 'John', 'template': 'foobar'}

Answer 2

你可以实现__deepcopy__方法来实现你想要的：

class FoobarEnum(Enum):
  FIRST = "foobar"
  SECOND = "baz"

  def __deepcopy__(self, memo):
      return self.value

asdict 函数处理数据类、元组、列表和字典。如果是它调用的任何其他类型：

copy.deepcopy(obj)

不过，像这样覆盖 __deepcopy__ 可能不是最好的主意。

Answer 3

添加__post_init__，如下所示。

from dataclasses import dataclass, asdict
from enum import Enum
from typing import Union


@dataclass
class Foobar:
    name:str
    template: Union["FoobarEnum", str]
    def __post_init__(self):
        self.template = self.template.value

class FoobarEnum(Enum):
    FIRST = "foobar"
    SECOND = "baz"

foobar = Foobar(name="John", template=FoobarEnum.FIRST)
print(asdict(foobar))

根据用例，您可以同时继承 str 和 Enum 或拥有一个仅初始化字段。

Answer 4

你试过了吗？

import json

def dumps(object):
    def default(o):
        if isinstance(o, Enum):
            # use enum value when JSON deserialize the enum
            return o.__dict__['_value_'] 
        else:
            return o.__dict__
    return json.dumps(object, default=default)

print(json.dumps(YOUR_OBJECT_CONTAINS_ENUMS, default=default))

Answer 5

from dataclasses import dataclass, asdict
from enum import Enum


class FoobarEnum(Enum):
    FIRST = "foobar"
    SECOND = "baz"


@dataclass
class Foobar:
    name: str
    template: FoobarEnum


def my_dict(data):

    return {
        field: value.value if isinstance(value, Enum) else value
        for field, value in data
    }


foobar = Foobar(name="John", template=FoobarEnum.FIRST)

data = {'name': 'John', 'template': 'foobar'}

assert asdict(foobar, dict_factory=my_dict) == data

Answer 6

其实你可以做到的。 asdict 有关键字参数dict_factory 允许您在那里处理您的数据：

from dataclasses import dataclass, asdict
from enum import Enum


@dataclass
class Foobar:
  name: str
  template: "FoobarEnum"


class FoobarEnum(Enum):
  FIRST = "foobar"
  SECOND = "baz"


def custom_asdict_factory(data):

    def convert_value(obj):
        if isinstance(obj, Enum):
            return obj.value
        return obj

    return dict((k, convert_value(v)) for k, v in data)


foobar = Foobar(name="John", template=FoobarEnum.FIRST)

print(asdict(foobar, dict_factory=custom_asdict_factory))
# {'name': 'John', 'template': 'foobar'}