如何“投射” Python类型提示

Question

我有一个客户端库，它的功能看起来像这样

def connect(timeout: float):

但是，该客户端的文档指出，如果未指定timeout或“无”，它将有效地使用默认超时。 我已经测试过这种情况，并且可以正常指定timeout的None值

然后，我的代码使用以下内容包装此connect方法：

def my_connect(timeout: float = None):
    connect(timeout=timeout)

当我对它运行mypy来验证类型提示时，它抱怨： connect的参数timeout期望float但是得到Optional[float] 。

我宁愿不必在代码中包含其他逻辑，也connect根据my_connect方法中是否使用默认值None来更改我调用connect方法的方式，因为这将需要其他单元测试以及整个原因我首先需要超时是因为单元测试会阻塞，而不会在否定的测试用例上超时。

建议的处理方式是什么？ 即使库创建者明确指定了一个非可选的float，是否有某种方法可以强制接受该参数？ 该库是专有的并且仅二进制文件，因此我没有能力更改签名，并且我真的很想尝试在不更改任何功能逻辑的情况下进行此工作，仅更改类型提示。

Answer 1

我已经为此实现了一个解决方案，它似乎可以通过所有检查，并且在功能上似乎等效。

同样，图书馆的签名是

def connect(timeout: float):

包装是：

def my_connect(timeout: float = None):

事实证明，如果缺少该值或None ，则connect方法默认为socket._GLOBAL_DEFAULT_TIMEOUT 。 在文档中这是正确的，我只是错过了。

解决方案是简单地更改包装器上的默认值以使用相同的socket._GLOBAL_DEFAULT_TIMEOUT而不是None。

例如

import socket
...
def my_connect(timeout: float = socket._GLOBAL_DEFAULT_TIMEOUT):

如果值显式为None ，则不会将float转换为Optional[float] ，从而导致隐式类型提示，从而正确地将参数键入float 。 这样就可以完全通过类型提示检查。