Я хочу аннотировать функцию, чтобы указать, что она может возвращать значения только из IntEnum
from enum import IntEnum
import random
class GoodIntegers(IntEnum):
THE_ANSWER_TO_LIFE_THE_UNIVERSE_AND_EVERYTHING = 42
DEADBEEF = 0xdeadbeef
LEET = 1337
def f() -> GoodIntegers:
return random.choice([42, 1337, 0xdeadbeef])
Но mypy не поверит мне, что мой источник целых чисел поступает правильно:
$ mypy test.py
test.py:12: error: Incompatible return value type (got "int", expected "GoodIntegers") [return-value]
Found 1 error in 1 file (checked 1 source file)
Есть ли способ сообщить mypy, что я действительно доверяю своему источнику целых чисел, который выдает мне только значения, соответствующие значению в моем IntEnum?
В моем реалистичном примере источником целых чисел являются отдельные элементы байтовой строки:
class OpCodes(IntEnum):
...
def get_ops_from_script(script: bytes) -> List[Tuple[OpCodes, Optional[bytes]]:
ops = []
while n < len(script):
op: GoodIntegers = script[n]
...
# depending on the opcode some data may be extracted from the script
# then n is incremented to find the next opcode
ops.append((op, data))
return ops
Подробнее здесь: https://stackoverflow.com/questions/775 ... an-intenum
Как напечатать возвращаемое значение в IntEnum ⇐ Python
-
- Похожие темы
- Ответы
- Просмотры
- Последнее сообщение