nonebot2/nonebot/internal/driver/combine.py

from typing import TYPE_CHECKING, Type, Union, TypeVar, overload

from .abstract import Mixin, Driver

D = TypeVar("D", bound="Driver")

if TYPE_CHECKING:

    class CombinedDriver(Driver, Mixin):
        ...


@overload
def combine_driver(driver: Type[D]) -> Type[D]:
    ...


@overload
def combine_driver(driver: Type[D], *mixins: Type[Mixin]) -> Type["CombinedDriver"]:
    ...


def combine_driver(
    driver: Type[D], *mixins: Type[Mixin]
) -> Union[Type[D], Type["CombinedDriver"]]:
    """将一个驱动器和多个混入类合并。"""
    # check first
    if not issubclass(driver, Driver):
        raise TypeError("`driver` must be subclass of Driver")
    if not all(issubclass(m, Mixin) for m in mixins):
        raise TypeError("`mixins` must be subclass of Mixin")

    if not mixins:
        return driver

    def type_(self: "CombinedDriver") -> str:
        return (
            driver.type.__get__(self)  # type: ignore
            + "+"
            + "+".join(x.type.__get__(self) for x in mixins)  # type: ignore
        )

    return type(
        "CombinedDriver", (*mixins, driver), {"type": property(type_)}
    )  # type: ignore