Как создать динамически импортируемый шаблон стратегии в pydantic?

Кажется, что оптимального решения можно достичь гораздо проще, переопределив __new__ следующим образом:

class Strategy(BaseModel):
    strategy: str

    def __new__(cls, *args, **kwargs):
        qualifiers = kwargs.get("strategy").split(".")
        module = ".".join(qualifiers[:-1])
        attr = qualifiers[-1]
        module_ref = importlib.import_module(module)
        child_cls = getattr(module_ref, attr)
        return super().__new__(child_cls)

любой класс, наследуемый от Strategy, будет загружен в соответствии со значением его атрибута strategy. В примере модуля под названием my.pkg:

class Cone(Strategy):
    radius: float
    height: float

а затем откуда угодно:

>>> Strategy(strategy = "my.pkg.Cone", radius=3, height=5)
Cone(strategy='my.pkg.Cone', radius=3.0, height=5.0)

Хотя на первый взгляд это выглядит элегантным решением, я бы посоветовал не привязывать строку к фактическому имени класса и местоположению в пакете. Это усложнит обслуживание в будущем. Например, когда классы переименовываются или перемещаются, старые файлы не разрешаются, что приводит к ошибке.

Вместо этого я бы рекомендовал определить независимый тег. Это позволяет вам работать со стандартным шаблоном Pydantic: дискриминируемым объединением.

Вот минимальный пример:

from pydantic import BaseModel, Field
from typing import Annotated, Literal, Union

class Strategy(BaseModel):
    pass


# The subclasses can live in the same file or
# in different files, as long as they inherit from Strategy
class Cone(Strategy):
    strategy: Literal["my-cone"] = "my-cone"
    radius: float
    height: float


class Translate(Strategy):
    strategy: Literal["my-translate"] = "my-translate"
    x: float
    y: float
    z: float


class Rotate(Strategy):
    strategy: Literal["my-rotate"] = "my-rotate"
    quaternion: tuple[float, float, float, float]


def get_all_subclasses(cls):
    """Get subclasses"""
    for subclass in cls.__subclasses__():
        yield from get_all_subclasses(subclass)
        yield subclass


sub_classes = tuple(get_all_subclasses(Strategy))
AnyStrategy = Annotated[Union[sub_classes], Field(discriminator = "strategy")]


class Model(BaseModel):
    shape: AnyStrategy
    transformations: list[AnyStrategy]

data = {
  "shape": {
    "strategy": "my-cone",
    "radius": 5,
    "height": 10
  },
  "transformations": [
    {
      "strategy": "my-translate",
      "x": 10,
      "y": 10,
      "z": 10
    },
    {
      "strategy": "my-rotate",
      "quaternion": [0, 0, 0, 0]
    }
  ]
}

Model(**data)

Что возвращает:

Model(shape=Cone(strategy='my-cone', radius=5.0, height=10.0), transformations=[Translate(strategy='my-translate', x=10.0, y=10.0, z=10.0), Rotate(strategy='my-rotate', quaternion=(0.0, 0.0, 0.0, 0.0))])

Хотя это и облегчило сохранение тега strategy на ваших подклассах, я думаю, что это приводит к более стабильному и простому в обслуживании коду в долгосрочной перспективе.

Если вас действительно волнует решение с расположением пакетов и именами классов, вы можете автоматически генерировать их, используя вызываемые дискриминаторы.

Надеюсь, это поможет!