python-3.x - 从 Optional[T] 获取 Python 类型 T 的变量

Question

两个python方法定义为:

   def function() -> Optional[ast.Module]:
      pass

   def something(t: ast.Module) -> None:
      pass

在调用者中:

   var = function()
   if var:
     do_something(var)

类型检查器提示我不能将 Optional[t] 传递给 t。我想象有某种类型的类型转换机制，但它逃脱了我。

准确地说，这是一个单元测试，看起来像:

    self.assertIsNotNone(var)
    do_something(var)

关于可能的重复 Optional type annotation. Using value after checking if is None? ,

new_var: ast.Module = var

得到相同类型的错误。

是否需要“如果”？

Answer 1

必须明确排除值与 None 相同的情况，而不仅仅是任何 false'y 值。

var: Optional[...] = function()
if var is not None:    # reject None case in this branch
    do_something(var)  # `var` is the unwrapped type

反过来也是可能的:

var: Optional[...] = function()
if var is None:    # handle None case in this branch
   ...
else:              # no more None case in this branch
    do_something(var)  # `var` is the unwrapped type

---

请注意，类型检查器无法检查类型保护的任意函数。诸如 self.assertIsNotNone(var) 之类的隐式拒绝不等同于针对None 的显式检查。需要额外的明确排除。

self.assertIsNotNone(var)
assert var is not None    # reject None case after this point
do_something(var)

当然，总是可以cast离开选项。由于这会禁用对变量类型的验证，因此通常只能作为最后的手段。

do_something(cast(ast.Module, var))