在Python中将字符串转换为Enum [英] Convert string to Enum in Python

问题描述

我想知道将字符串转换(反序列化)为Python的Enum类的正确方法是什么.似乎 getattr(YourEnumType，str)可以完成这项工作，但是我不确定它是否足够安全.

I wonder what's the correct way of converting (deserializing) a string to a Python's Enum class. Seems like getattr(YourEnumType, str) does the job, but I'm not sure if it's safe enough.

更具体地说，我想将'debug'字符串转换为这样的Enum对象:

Just to be more specific, I would like to convert a 'debug'string to an Enum object like this:

class BuildType(Enum):
    debug = 200
    release = 400

推荐答案

此功能已内置到枚举[1]中:

This functionality is already built in to Enum [1]:

>>> from enum import Enum
>>> class Build(Enum):
...   debug = 200
...   build = 400
... 
>>> Build['debug']
<Build.debug: 200>

成员名称区分大小写，因此，如果要转换用户输入，则需要确保大小写匹配:

The member names are case sensitive, so if user-input is being converted you need to make sure case matches:

an_enum = input('Which type of build?')
build_type = Build[an_enum.lower()]

[1]官方文档: 枚举编程访问权限

[1] Official docs: Enum programmatic access

这篇关于在Python中将字符串转换为Enum的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！