Skip to content
Advertisement

How do I use type hint for multiple classes in a dictionary but they inherit from same parent class?

I want to use type hint with string as a key and different classes as a value in a dictionary like below.

configurations: Dict[str, what_class??] = {
    "dev": Dev,
    "product": Product,
    "test": Test,
}

The classes Dev, Product, Test inherit from a base class “Parent”.

class Parent:
    pass

class Dev(Parent):
    pass

class Product(Parent):
    pass

class Test(Parent):
    pass

What is the best way to do the type hint for this?

Do I need to set just “Any” as their type?

Advertisement

Answer

Depends on how accurate you want your hints to be,

  1. Use the parent class
configurations: Dict[str, Parent] = {
    "dev": Dev,
    "product": Product,
    "test": Test,
}
  1. Specify the classes in a union
from typing import Union
configurations: Dict[str, Union[Dev, Product, Test]] = {
    "dev": Dev,
    "product": Product,
    "test": Test,
}
  1. Create a TypedDict type for this specific dict
from typing import TypedDict

class EnvDict(TypedDict):
    dev: Dev
    product: Product
    test: Test
    
configurations: EnvDict = {
    "dev": Dev,
    "product": Product,
    "test": Test,
}
User contributions licensed under: CC BY-SA
9 People found this is helpful
Advertisement