代码之家 › 专栏 › 技术社区 › Q42

如何让Pyright从枚举检查中推断类型?

pyright python-typing enums python

Q42 · 技术社区 · 1 年前

类型是否可以与枚举相关联,以便Pyright可以从等式检查中推断出类型?(没有 cast() 或 isinstance() .)

from dataclasses import dataclass
from enum import Enum, auto

class Type(Enum):
    FOO = auto()
    BAR = auto()

@dataclass
class Foo:
    type: Type

@dataclass
class Bar:
    type: Type

item = next(i for i in (Foo(Type.FOO), Bar(Type.BAR)) if i.type == Type.BAR)
reveal_type(item)  # How to have this be `Bar` instead of `Foo | Bar`?

1 回复 | 直到 1 年前

InSync Christian C. Salvadó 1 年前

你想要一个 受歧视的工会 (也称为 tagged union ).

在受歧视的工会中,存在 鉴别器 (也称为a 标签字段 )其可用于区分成员。

您当前有一个联盟 Foo 和 Bar ,你想用 .type 属性。但是,此字段不能作为鉴别器,因为它对联盟的每个成员都没有不同。

_{(游乐场:

Pyright

,

Mypy

)}

for i in (Foo(Type.FOO), Bar(Type.BAR)):
    reveal_type(i)  # Foo | Bar

mischievous_foo = Foo(Type.BAR)  # This is valid
naughty_bar = Bar(Type.FOO)      # This too

for i in (mischievous_foo, naughty_bar):
    if i.type == Type.FOO:
        reveal_type(i)           # Runtime: Bar, not Foo

如果 Foo.type 只能是 Type.FOO 和 Bar.Type 是 Type.BAR ,那么在类型中反映这一点很重要:

(制作 type 此时,数据类字段不再有意义,但我假设就这个问题而言,它们只是数据类。)

@dataclass
class Foo:
    type: Literal[Type.FOO]

@dataclass
class Bar:
    type: Literal[Type.BAR]

As Literal[Type.FOO] 和 Literal[Type.BAR] 是不相交类型, i 然后通过检查类型来缩小范围 .type :

_{(游乐场:

Pyright

,

Mypy

)}

for i in (Foo(Type.FOO), Bar(Type.BAR)):
    if i.type == Type.FOO:
        reveal_type(i)           # Foo

Foo(Type.BAR)  # error
Bar(Type.FOO)  # error

…即使在发电机中,是的:

item = next(i for i in (Foo(Type.FOO), Bar(Type.BAR)) if i.type == Type.BAR)
reveal_type(item)                # Bar

推荐文章

H.v.M. · 为最后一个枚举值指定不同的变量值

4 年前

JimmyDiJim · 在声明后定义枚举变量

8 年前

Eli K · 语法错误,插入“{”以完成枚举体(在类的末尾)

8 年前

Vivek Shukla · 在中间添加枚举

8 年前

Sebastian · 在IntelliJ中通过Ctrl+left查找枚举的用法

8 年前

Mohamad Moustafa · 创建一个接受枚举变量并返回字符串指针的函数

8 年前

Michel Michels · 用于在SQL Server中保存选定月份并在WPF中显示的数据结构[已关闭]

8 年前

roman_ds · 为什么将对象指针前缀添加到值中?(Swift 4)

8 年前

LucaRoverelli · Swift 4中带默认大小写的可编码枚举

8 年前

DannyD · 如何访问typescript中的枚举名称

8 年前