python类型注解_适当的类型注释的Python函数与yield

阅读Eli Bendersky的文章

on implementing state machines via Python coroutines后,我想…

看看他在Python3下运行的例子

>并为生成器添加适当的类型注释

我成功地做了第一部分(但没有使用异步defs或yield froms,我基本上只是移植代码 – 所以任何改进是最受欢迎的).

但是我需要帮助协同类型注释：

#!/usr/bin/env python3

from typing import Callable, Generator

def unwrap_protocol(header: int=0x61,

footer: int=0x62,

dle: int=0xAB,

after_dle_func: Callable[[int], int]=lambda x: x,

target: Generator=None) -> Generator:

""" Simplified protocol unwrapping co-routine."""

#

# Outer loop looking for a frame header

#

while True:

byte = (yield)

frame = [] # type: List[int]

if byte == header:

#

# Capture the full frame

#

while True:

byte = (yield)

if byte == footer:

target.send(frame)

break

elif byte == dle:

byte = (yield)

frame.append(after_dle_func(byte))

else:

frame.append(byte)

def frame_receiver() -> Generator:

""" A simple co-routine "sink" for receiving full frames."""

while True:

frame = (yield)

print('Got frame:', ''.join('%02x' % x for x in frame))

bytestream = bytes(

bytearray((0x70, 0x24,

0x61, 0x99, 0xAF, 0xD1, 0x62,

0x56, 0x62,

0x61, 0xAB, 0xAB, 0x14, 0x62,

0x7)))

frame_consumer = frame_receiver()

next(frame_consumer) # Get to the yield

unwrapper = unwrap_protocol(target=frame_consumer)

next(unwrapper) # Get to the yield

for byte in bytestream:

unwrapper.send(byte)

这运行正常…

$./decoder.py

Got frame: 99afd1

Got frame: ab14

…以及类型检查：

$mypy --disallow-untyped-defs decoder.py

$

但是我很确定我可以做的不仅仅是在类型规范中使用Generator基类(就像我对Callable一样).我知道它需要3个类型的参数(Generator [A,B,C]),但我不知道这里是如何被指定的.

任何帮助最受欢迎.