首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >python,其中键和值依赖于相同的泛型类型

python,其中键和值依赖于相同的泛型类型
EN

Stack Overflow用户
提问于 2022-06-24 10:59:36
回答 1查看 181关注 0票数 2

我想要定义一个dict,它的键和值共享相同的泛型类型,并且对它有一些约束。以下是这种情况的例子。但是,将mypy应用于以下代码会导致错误:

代码语言:javascript
运行
复制
tmp.py:8: error: Type variable "tmp.AT" is unbound
tmp.py:8: note: (Hint: Use "Generic[AT]" or "Protocol[AT]" base class to bind "AT" inside a class)
tmp.py:8: note: (Hint: Use "AT" in function signature to bind "AT" inside a function)
Found 1 error in 1 file (checked 1 source file)

怎么解决这个问题?我需要这样一个dict的原因是我想在dict的键和值之间添加一个类型约束。

tmp.py:

代码语言:javascript
运行
复制
from typing import Dict, Generic, TypeVar, Type, List

class A: pass
class B(A): pass
class C(A): pass

AT = TypeVar("AT", bound=A)
d: Dict[Type[AT], List[AT]] = {}
EN

回答 1

Stack Overflow用户

发布于 2022-06-26 01:25:26

我最终通过定义从Dict继承的自定义dict来解决这个问题。

代码语言:javascript
运行
复制
from typing import Dict, TypeVar, Type, List
from dataclasses import dataclass

AT = TypeVar("AT")

class ConstrainedDict(Dict):
    def __getitem__(self, k: Type[AT]) -> List[AT]:
        return super().__getitem__(k)

class Foo: pass
class Bar: pass

d: ConstrainedDict
a: List[Foo] = d[Foo]  # expected to pass
b: List[Foo] = d[Bar]  # expected to cause error

我所期望的结果是:

代码语言:javascript
运行
复制
tmp.py:15: error: Invalid index type "Type[Bar]" for "ConstrainedDict"; expected type "Type[Foo]"
Found 1 error in 1 file (checked 1 source file)
票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/72743074

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档