Python默认参数为None时，如何让类型检查器识别变量为int？

问题场景 #

我有如下代码：

def other_function(x: int) -> None:
    pass

def foo(x: int = None) -> None:   # 不一致；如何优化？
    if x is None:
        x = 4
    other_function(x)

执行到other_function(x)行时，我确定x已经是int类型，但Neovim中的Pyright类型检查器仍认为x是Union[int, None]，因此不允许将x传入other_function。

我需要解决的问题是：有没有办法告知类型检查器“此时我确定x是int类型”？

我尝试过相关问题的解决方案及变体，但均未成功。

说明：我并不关注函数声明处的类型注解（这是相关问题的讨论点），而是希望让类型检查器知道，在执行if x is None:...这段逻辑后x是int类型。

可行解决方案 #

1. 使用typing.cast进行类型断言 #

通过cast函数明确向类型检查器声明变量的类型，它不会影响代码运行时的行为，仅作为检查器的提示：

from typing import cast

def other_function(x: int) -> None:
    pass

def foo(x: int = None) -> None:
    if x is None:
        x = 4
    # 断言x为int类型
    x = cast(int, x)
    other_function(x)

2. 调整分支逻辑让检查器自动推导 #

将函数调用放到条件分支内部，让检查器能明确每个分支中x的类型：

def other_function(x: int) -> None:
    pass

def foo(x: int = None) -> None:
    if x is None:
        other_function(4)
    else:
        other_function(x)

或者保持原有赋值逻辑，将调用放在分支里：

def foo(x: int = None) -> None:
    if x is None:
        x = 4
        other_function(x)
    else:
        other_function(x)

3. （可选）用Optional修正参数注解 #

虽然你提到不关注函数声明注解，但正确标注参数类型能让检查器自动识别条件后的变量类型，从根源解决问题：

from typing import Optional

def other_function(x: int) -> None:
    pass

def foo(x: Optional[int] = None) -> None:
    if x is None:
        x = 4
    other_function(x)

Optional[int]是Union[int, None]的别名，检查器会跟踪到if x is None分支后x的类型已变为int。

内容的提问来源于stack exchange，提问作者Laurent Claessens