如何确保用户输入符合自定义Python TypeAlias并通过mypy检查？

一、原代码的mypy错误原因及修复 #

你原代码里的type(raw_input) == CustomType完全是错误逻辑——CustomType是静态类型别名（对应Literal['MyCustomType']），运行时不存在这个类型对象，raw_input的实际类型永远是str，所以这个条件永远为False，mypy能识别到这一点，因此抛出「Incompatible types in assignment」错误。

正确的做法是直接检查输入值是否属于CustomType允许的字面量集合，用get_args获取Literal的可选值：

from typing import TypeAlias, Literal, Union, get_args
from sys import argv

CustomType: TypeAlias = Literal['MyCustomType']
raw_input: Union[CustomType, str] = argv[1]
default_value: CustomType = 'MyCustomType'
# 替换类型检查为值检查
typed_input: CustomType = raw_input if raw_input in get_args(CustomType) else default_value

这样mypy能正确识别：当raw_input在允许值范围内时，它符合CustomType类型，否则使用默认值，不会再报错。

二、argparse的优雅类型绑定方案 #

你想直接写type=CustomType是行不通的——argparse的type参数需要运行时可调用的类型/函数，但Literal是静态类型提示，运行时没有实际的构造逻辑。下面是两种优雅的解决方式：

方案1：自定义类型转换函数（推荐） #

写一个校验并转换输入的函数，给它标注正确的返回类型，让mypy能自动识别argparse参数的类型：

from argparse import ArgumentParser, Namespace
from typing import TypeAlias, Literal, get_args, cast

CustomType: TypeAlias = Literal['MyCustomType']

def parse_custom_type(input_str: str) -> CustomType:
    if input_str in get_args(CustomType):
        # 用cast告诉mypy：此时input_str符合CustomType类型
        return cast(CustomType, input_str)
    raise ValueError(f"无效输入，必须是{get_args(CustomType)}中的一个")

parser = ArgumentParser(description="处理输入参数")
# 用自定义函数作为type，同时保留choices做输入提示
parser.add_argument("argument", type=parse_custom_type, choices=get_args(CustomType))

arguments = parser.parse_args()
# mypy会自动识别arguments.argument的类型为CustomType，无需手动标注
my_argument = arguments.argument

这个方案既实现了运行时的输入校验，又让mypy能正确推断参数类型，完全符合你的需求。

方案2：利用typing.cast简化标注 #

如果你不想写自定义函数，也可以在获取参数后用cast直接给mypy做类型断言：

from argparse import ArgumentParser, Namespace
from typing import TypeAlias, Literal, get_args, cast

CustomType: TypeAlias = Literal['MyCustomType']

parser = ArgumentParser(description="处理输入参数")
parser.add_argument("argument", choices=get_args(CustomType))

arguments = parser.parse_args()
# 用cast告诉mypy：该参数符合CustomType类型
my_argument: CustomType = cast(CustomType, arguments.argument)

这种方式更简洁，但依赖手动断言，适合简单场景。

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