歧视联盟 Python

Question

赏金将在 6 天后到期。 此问题的答案有资格获得+400声望赏金。 mnowotka想让更多人关注这个问题：

我希望赏金获得者将提供一个尊重问题中提供的约束的解决方案。 具体来说，“为什么你做的完全不同”的答案将不会被接受，除非他们清楚地展示了一种替代方法，并提供了代码示例和关于优缺点的讨论。

假设我有一个基类 class 和两个派生类。 我还有一个工厂方法，它返回其中一个类的 object。 问题是，mypy 或 IntelliJ 无法确定 object 是哪种类型。 他们知道两者都可以，但不知道究竟是哪一个。 有什么方法可以帮助 mypy/IntelliJ 解决这个问题，而无需在conn变量名称旁边放置类型提示？

import abc
import enum
import typing


class BaseConnection(abc.ABC):
    @abc.abstractmethod
    def sql(self, query: str) -> typing.List[typing.Any]:
        ...


class PostgresConnection(BaseConnection):

    def sql(self, query: str) -> typing.List[typing.Any]:
        return "This is a postgres result".split()

    def only_postgres_things(self):
        pass


class MySQLConnection(BaseConnection):

    def sql(self, query: str) -> typing.List[typing.Any]:
        return "This is a mysql result".split()

    def only_mysql_things(self):
        pass


class ConnectionType(enum.Enum):
    POSTGRES = 1
    MYSQL = 2


def connect(conn_type: ConnectionType) -> typing.Union[PostgresConnection, MySQLConnection]:
    if conn_type is ConnectionType.POSTGRES:
        return PostgresConnection()
    if conn_type is ConnectionType.MYSQL:
        return MySQLConnection()


conn = connect(ConnectionType.POSTGRES)
conn.only_postgres_things()

看看 IntelliJ 是如何处理这个的：

正如您所看到的两种方法：当我希望 IntelliJ/mypy 从我传递给connect function 的类型中找出它时，建议使用only_postgres_things和only_mysql_things 。

Answer 1

您可以尝试将typing.overload与typing.Literal结合使用，如下所示：

@typing.overload
def connect(type_: typing.Literal[ConnectionType.POSTGRES]) -> PostgresConnection:
    ...

@typing.overload
def connect(type_: typing.Literal[ConnectionType.MYSQL]) -> MySQLConnection:
    ...

def connect(type_):
    if type_ is ConnectionType.POSTGRES:
        return PostgresConnection()
    if type_ is ConnectionType.MYSQL:
        return MySQLConnection()

我用type_替换了type ，这样你就不会隐藏内置函数，并且使用is而不是==来比较枚举值是惯用的。

Answer 2

由于您的ConnectionType class 的目的显然是使您的 API 更具可读性和用户友好性，而不是使用Enum的任何特定功能，因此您实际上不必将其设为Enum class。

相反，您可以创建一个常规的 class，并将每个连接类型分配给一个用户友好名称的 class 变量，这样您就可以使用类型变量键入connect function 的返回值，并使用该类型的类型键入参数多变的。 使用类型别名使类型变量的类型更具可读性：

class ConnectionTypes:
    POSTGRES = PostgresConnection
    MYSQL = MySQLConnection

Connection = typing.TypeVar('Connection', PostgresConnection, MySQLConnection)
# or make it bound to the base class:
# Connection = typing.TypeVar('Connection', bound=BaseConnection)
ConnectionType: typing.TypeAlias = type[Connection]

def connect(type_: ConnectionType) -> Connection:
    if type_ is ConnectionType.POSTGRES:
        return PostgresConnection()
    if type_ is ConnectionType.MYSQL:
        return MySQLConnection()