dispatch.py

from dataclasses import dataclass


@dataclass
class Potato:
    value: str


class EnglishPotato(Potato): ...


class SweetPotato(Potato): ...


def give_me_something(arg: int) -> EnglishPotato | SweetPotato:
    value = "batata"
    if arg > 10:
        return SweetPotato(value)
    return EnglishPotato(value)


result = give_me_something(90)

actions = {
    EnglishPotato: lambda: print("Deep Fry it"),
    SweetPotato: lambda: print("Make bread"),
}

actions[type(result)]()

isintance_check.py

from dataclasses import dataclass


@dataclass
class Potato:
    value: str


class EnglishPotato(Potato): ...


class SweetPotato(Potato): ...


def give_me_something(arg: int) -> EnglishPotato | SweetPotato:
    value = "batata"
    if arg > 10:
        return SweetPotato(value)
    return EnglishPotato(value)


result = give_me_something(90)


if isinstance(result, EnglishPotato):
    print("Deep Fry it")
elif isinstance(result, SweetPotato):
    print("Make bread")

match_on_class.py

from dataclasses import dataclass


@dataclass
class Potato:
    __match_args__ = ("value",)
    value: str


class EnglishPotato(Potato): ...
class SweetPotato(Potato): ...


def give_me_something(arg: int) -> EnglishPotato | SweetPotato:
    value = "batata"
    if arg > 10:
        return SweetPotato(value)
    return EnglishPotato(value)


match result := give_me_something(90):
    case EnglishPotato(value):
        print("Deep Fry it")
    case SweetPotato(value):
        print("Make bread")