在使用Mypy进行Python静态类型检查时，函数内的类型精炼和非空值推断常常不符合预期，导致类型检查报错或者遗漏潜在问题。理解Mypy的类型推断规则，掌握正确的写法可以让类型检查更精准。
Mypy类型精炼的基本逻辑
Mypy的类型精炼（Type Narrowing）指的是根据代码中的条件判断，缩小变量的可能类型范围。比如当一个变量被标注为Optional[str]时，如果在条件判断中确认它不为None，Mypy应该能推断后续代码中该变量的类型为str。
基础的精炼场景不需要额外操作，Mypy会自动处理简单的条件判断：
from typing import Optional
def process_name(name: Optional[str]) -> str:
if name is not None:
# 此处Mypy会自动推断name为str类型
return name.upper()
return "default"
函数内类型精炼常见问题
1. 条件判断后类型未缩小
如果条件判断的逻辑比较复杂，或者判断逻辑被封装到单独的函数中，Mypy可能无法自动完成类型精炼。比如下面的例子：
from typing import Optional
def is_valid_name(name: Optional[str]) -> bool:
return name is not None and len(name) > 0
def get_name_length(name: Optional[str]) -> int:
if is_valid_name(name):
# 此处Mypy仍然认为name是Optional[str]，会报错
return len(name)
return 0
这是因为Mypy默认不会分析is_valid_name函数的返回值和输入参数的类型关联，所以无法推断条件成立时name的非空属性。
2. 非空值推断失效
当变量经过多次赋值或者条件分支较多时，Mypy可能无法正确跟踪变量的非空状态，导致后续使用非空值时出现类型错误提示。
解决方法
使用TypeGuard自定义类型守卫
Python 3.10及以上版本支持TypeGuard，可以给判断函数标注返回类型，明确告诉Mypy当函数返回True时，输入参数的类型范围。
from typing import Optional, TypeGuard
def is_valid_name(name: Optional[str]) -> TypeGuard[str]:
return name is not None and len(name) > 0
def get_name_length(name: Optional[str]) -> int:
if is_valid_name(name):
# 此处Mypy正确推断name为str类型
return len(name)
return 0
如果是Python 3.10以下版本，可以使用TypeGuard的替代写法，通过typing_extensions库导入：
from typing import Optional
from typing_extensions import TypeGuard
def is_valid_name(name: Optional[str]) -> TypeGuard[str]:
return name is not None and len(name) > 0
使用assert语句明确非空
如果函数内已经通过逻辑确认变量非空，可以使用assert语句，Mypy会识别assert后的变量类型：
from typing import Optional
def process_value(value: Optional[int]) -> int:
# 假设此处逻辑已经确保value不为None
assert value is not None
# 此处Mypy推断value为int类型
return value * 2
避免过复杂的分支逻辑
尽量将类型判断的逻辑放在使用变量的同一作用域内，减少跨函数的类型判断逻辑，能让Mypy更准确地跟踪类型变化。如果必须在多个函数间传递类型状态，优先使用TypeGuard标注判断函数。
验证效果
完成上述修改后，运行Mypy检查代码，之前的类型错误提示会消失，同时如果后续代码中错误地将变量当作None处理，Mypy也会正确报错。可以通过下面的示例验证：
from typing import Optional, TypeGuard
def is_non_empty_str(value: Optional[str]) -> TypeGuard[str]:
return value is not None and len(value) > 0
def handle_input(data: Optional[str]) -> None:
if is_non_empty_str(data):
print(data.strip()) # 正确，data被推断为str
else:
print("输入为空")
# 如果此处错误地使用data的方法，Mypy会报错
# print(data.upper()) # 取消注释会提示错误