Type annotations for Enum value

python python-3.x type-hinting typing type-annotation
Yuval Pruss picture Yuval Pruss · Oct 3, 2018 · Viewed 21.8k times · Source

I have this piece of code:

import enum


class Color(enum.Enum):
    RED = '1'
    BLUE = '2'
    GREEN = '3'


def get_color_return_something(some_color):
    pass

How do I properly add type annotations to the some_color varaible in this function, if I suppose to receive a value from the Color enum (for example: Color.RED)?

Answer

ibarrond picture ibarrond · Oct 3, 2018

Type hinting the Color class should work:

def get_color_return_something(some_color: Color):
    print(some_color.value)

Related questions

Python 3 type hinting for decorator

Considere the following code: from typing import Callable, Any TFunc = Callable[..., Any] def get_authenticated_user(): return "John" def require_auth() -> Callable[TFunc, TFunc]: def decorator(func: TFunc) -> TFunc: def wrapper(*args, **kwargs) -> Any: …

Read More
What are type hints in Python 3.5?

One of the most talked about features in Python 3.5 is type hints. An example of type hints is mentioned in this article and this one while also mentioning to use type hints responsibly. Can someone explain more about them and …

Read More
Adding default parameter value with type hint in Python

If I have a function like this: def foo(name, opts={}): pass And I want to add type hints to the parameters, how do I do it? The way I assumed gives me a syntax error: def foo(name: str, …

Read More