python - 具有已知键输入的 Python 3 字典

Question

我正在使用 Python 3 打字功能来更好地自动完成。

很多时候，我有返回带有特定键的键/值（字典）的函数。超级简单的例子：

def get_info(name):
    name_first_letter = name[0]
    return {'my_name': name, 'first_letter': name_first_letter}

我想给这个函数添加类型提示，告诉其他使用这个函数的人会发生什么。

我可以做类似的事情：

 class NameInfo(object):
     def __init__(self, name, first_letter):
         self.name = name
         self.first_letter = first_letter

然后将函数签名更改为：

def get_info(name) -> NameInfo:

但是每个字典都需要太多的代码。

在这种情况下，最佳做法是什么？

Answer 1

正如 Blckknght、您和 Stanislav Ivanov 在评论中指出的那样，您可以使用NamedTuple：

from typing import NamedTuple


class NameInfo(NamedTuple):
    name: str
    first_letter: str


def get_info(name: str) -> NameInfo:
    return NameInfo(name=name, first_letter=name[0])

从 Python 3.8 开始，您可以使用TypedDict与您想要的更相似的：

from typing import TypedDict


class NameInfo(TypedDict):
    name: str
    first_letter: str


def get_info(name: str) -> NameInfo:
    return {'name': name, 'first_letter': name[0]}