python - 打字:定义一个只能是某些字符串的类型？

Question

如何使用打字模块来创建可以是某些字符串的类型？

例如。假设我需要一个类型 CondOperator ，可以是以下任何字符串:

['=', '>', '<', '>=', '<=', '<>', '!=']

我希望 CondOperator = String['=', '>', '<', '>=', '<=', '<>', '!='] ，但没有 String在 typing .所以这行不通。

如何定义这样的类型？

Answer 1

Python 3.8 引入了 Literal输入 typing .也可通过 typing_extensions 访问包在以前版本的 Python 中。与 Literal ，您可以将您的类型定义为:

from typing import Literal

CondOperator = Literal['=', '>', '<', '>=', '<=', '<>', '!=']

mypy docs在 Literal是一本好书。