如何同时实现Literal穷举检查、无Linter警告与精确返回类型推断？

核心需求 #

• 对Literal类型的参数进行穷举检查，确保所有可能的取值都被覆盖
• 避免Ruff等Linter工具发出的「返回语句过多」警告
• 保留精确的返回类型推断能力，例如让类型检查器自动推断出返回类型为Literal[42, "string"]

现有实现方案对比 #

版本1——基于match语句（推断精准，但触发Linter警告） #

from typing import Literal, assert_never, reveal_type

def foo(x: Literal["a", "b"]):
    match x:
        case "a":
            return 42
        case "b":
            return "string"
        case _:
            assert_never(x)

reveal_type(foo)  # Pyright推断为: (x: Literal['a', 'b']) -> Literal[42, 'string']

该版本通过match语句完成穷举检查，配合assert_never确保无遗漏，类型推断精准，但多分支返回会触发Linter的「返回语句过多」警告。

版本2——基于字典映射（代码简洁，但推断效果差） #

from typing import Literal, reveal_type

def bar(x: Literal["a", "b"]):
    mapping = {"a": 42, "b": "string"}
    return mapping[x]

reveal_type(bar)  # Pyright推断为: (x: Literal['a', 'b']) -> Unknown

该版本仅用单个return语句，规避了Linter警告，但类型检查器无法从字典映射中推断出精确的返回类型，只能识别为Unknown。

问题 #

是否存在一种方案可以同时满足以下三点？

• 实现穷举检查（如通过match语句或等效方式）
• 使用单个返回语句（或其他方式避免「返回语句过多」警告）
• 自动推断出精确的返回类型？

可行解决方案 #

通过match语句给变量赋值，再单次返回变量的方式，可以同时满足所有需求：

from typing import Literal, assert_never, reveal_type

def baz(x: Literal["a", "b"]):
    match x:
        case "a":
            result = 42
        case "b":
            result = "string"
        case _:
            assert_never(x)
    return result

reveal_type(baz)  # Pyright推断为: (x: Literal['a', 'b']) -> Literal[42, 'string']

方案优势 #

• 穷举检查：通过match语句覆盖所有Literal取值，assert_never确保无遗漏分支
• Linter合规：仅使用单个return语句，不会触发「返回语句过多」警告
• 精确类型推断：类型检查器能通过match分支的赋值逻辑，自动推断出返回类型为Literal[42, "string"]

若需要更明确的类型标注，也可以显式声明result的类型，进一步强化类型约束：

def baz(x: Literal["a", "b"]):
    result: Literal[42, "string"]
    match x:
        case "a":
            result = 42
        case "b":
            result = "string"
        case _:
            assert_never(x)
    return result

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