以下对我来说不起作用,使用Python 3.4.7,ruamel.yaml版本0.15.35:
import sys
import enum
import ruamel.yaml
from ruamel.yaml import yaml_object
yaml = ruamel.yaml.YAML()
@yaml_object(yaml)
class Speed(enum.IntEnum):
Reverse = 0
Neutral = 1
Low = 2
Drive = 3
Park = 999
print("Neutral:", repr(Speed.Neutral))
yaml.dump(Speed.Neutral, sys.stdout)
我得到一个完全合理的repr
:
Neutral: <Speed.Neutral: 1>
但是.dump()
提出:
ruamel.yaml.representer.RepresenterError: cannot represent an object: <enum 'Speed'>
如果不支持enum
,我可以做些什么来扩展我使用的enum
类(或者我创建的子类enum.IntEnum
),例如一个dunder方法?
enum
不支持开箱即用,主要是因为默认的dump
方法是安全的,因此不支持任意Python对象。这种安全性也不包括标准库中的enum
类型。
你应该做的是在to_yaml
中描述的classmethod
类中加入Speed
ruamel.yaml documentation:
import sys
import enum
import ruamel.yaml
from ruamel.yaml import yaml_object
yaml = ruamel.yaml.YAML()
@yaml_object(yaml)
class Speed(enum.IntEnum):
Reverse = 0
Neutral = 1
Low = 2
Drive = 3
Park = 999
@classmethod
def to_yaml(cls, representer, node):
return representer.represent_scalar(
u'!Speed',
'{}-{}'.format(node._name_, node._value_)
)
yaml.dump(Speed.Neutral, sys.stdout)
这使:
!Speed Neutral-1
...
您当然可以根据自己的喜好调整字符串表示(并添加一个from_yaml
以便能够将输出加载回来)。
请注意,您不能像文档示例中那样添加yaml_tag
,因为这会干扰enum
值。
我认为你需要通过值传递枚举。
yaml.dump(Speed.Neutral.value, sys.stdout)
扩展答案添加@classmethod,在传递时调用.value
http://yaml.readthedocs.io/en/latest/dumpcls.html
import sys
import enum
import ruamel.yaml
from ruamel.yaml import yaml_object
yaml = ruamel.yaml.YAML()
@yaml_object(yaml)
class Speed(enum.IntEnum):
Reverse = 0
Neutral = 1
Low = 2
Drive = 3
Park = 999
@classmethod
def to_yaml(cls, representer, node):
return representer.represent_scalar(u'!enum:', u'{.value}'.format(node))