Type annotations for Enum value

Question

I have this piece of code:

import enum


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


def get_color_return_something(some_col         


        

Answer 1

def get_color_return_something(some_color: Color):
    pass

Answer 2

The following will work with Pyton 3.9/PyCharm

from enum import Enum
from typing import Optional, Union


class Color(Enum):
    RED: int = 1
    GREEN: int = 2


def guess_color(x: Union[Color.RED, Color.GREEN]) -> Optional[ValueError]:
    if x == Color.RED:
        print("Gotcha!")
    else:
        return ValueError(f"It's not {Color.RED}")


guess_color(Color.RED)

Answer 3

Type hinting the Color class should work:

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