如何让 Pylance 忽略 None 的可能性？

Question

我喜欢 Pylance 类型检查。

但是，如果我有一个变量var: Union[None, T]，其中T 实现foo，pylance 将在以下位置抛出错误：

var.foo() 因为None 类型没有实现foo。

有没有办法解决这个问题？一种告诉 Pylance 的方法“这个变量有时是 None，但在这种情况下，我 100% 确定它会被分配

Answer 1

有很多方法可以强制类型检查器接受这一点。

---

1. 使用assert：

   from typing import Union
   
   def do_something(var: Union[T, None]):
       assert var is not None
       var.foo()
   

   
   

2. 引发其他异常：

   from typing import Union
   
   def do_something(var: Union[T, None]):
       if var is None:
           raise RuntimeError("NO")
       var.foo()
   

   
   

3. 使用if 声明：

   from typing import Union
   
   def do_something(var: Union[T, None]):
       if var is not None:
           var.foo()
   

   
   

4. 使用typing.cast，该函数在运行时不执行任何操作，但会强制类型检查器接受变量属于特定类型：

   from typing import Union, cast
   
   def do_something(var: Union[T, None]):
       var = cast(T, var)
       var.foo()
   

   
   

5. 关闭该行的类型检查器：

   from typing import Union
   
   def do_something(var: Union[T, None]):
       var.foo()  # type: ignore
   

   
   

---

还要注意，虽然类型检查器如何解释类​​型注释没有区别（两者在语义上相同），但您也可以将 typing.Union[T, None] 写为 typing.Optional[T]，这可以说是稍微好一点的语法. In Python >=3.10（或者更早，如果您的代码顶部有 from __future__ import annotations），您甚至可以使用 | 运算符编写 Union 类型，即 T | None。