在python中的枚举转换器

Question

我有一个枚举

class Nationality:
        Poland='PL'
        Germany='DE'
        France='FR'
        ...
        Spain='ES'

我有2个方法原型:

# I want somethink like in c#        
def convert_country_code_to_country_name(country_code):
        print Enum.Parse(typeof(Nationality),country_code)

#this a second solution ,but it has a lot of ifs

def convert_country_code_to_country_name(country_code):
        if country_code=='DE':
                print Nationality.Germany #btw how to print here 'Germany', instead 'DE'

这就是我想要调用这个方法的方法:

convert_country_code_to_country_name('DE') # I want here to  print 'Germany'

如何在python中实现它？

Answer 1

最好的解决方案是从一开始就创建一个字典.你的枚举在Python中没有意义,它只是不必要的复杂.看起来你正在尝试编写Java代码,这与Python代码看起来完全相反.

Answer 2

我的方法是这样的（也许不完美，但你明白了）：

class Nationality:
        Poland = 'PL'
        Germany = 'DE'
        France = 'FR'

        def convertToCodeFromName(name):
                return getattr(Nationality, name)

        def convertToNameFromCode(code):
                lookFor = None

                for member in dir(Nationality):
                        if (getattr(Nationality, member) == code):
                                lookFor = member
                                break
                return lookFor

print(Nationality.convertToCodeFromName("Poland"))
print(Nationality.convertToNameFromCode("PL"))

希望这可以帮助。