Validate against enum members in Python

Question

In Python, I have an input (called input_var below) that I would like to validate against a enum (called Color below). Is the following way the recommended Pythonic approach?

from enum import Enum
class Color(Enum):
    red = 1
    blue = 2
input_var = 'red'
if input_var in Color.__members__:
    print('Everything is fine and dandy.')

gtlambert · Accepted Answer · 2016-01-29 12:27:34Z

13

Use the built-in hasattr() function. hasattr(object, name) returns True if the string name is an attribute of object, else it returns False.

Demo

from enum import Enum

class Color(Enum):
    red = 1
    blue = 2

input_var = 'red'

if hasattr(Color, input_var):
    print('Everything is fine and dandy.')

Output

Everything is fine and dandy.

answered Jan 29, 2016 at 12:27

gtlambert

12k2 gold badges32 silver badges48 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Ryan Rapp Over a year ago

This will return true for hasattr(Color, 'mro'). You may want if hasattr(Color, input_var) and input_var != 'mro'

Collectives™ on Stack Overflow

Validate against enum members in Python

1 Answer 1

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related