阅读 Eli Bendersky 的文章关于通过 Python 协程实现状态机后,我想...
我成功地完成了第一部分(但没有使用
async def
s 或 yield from
s,我基本上只是移植了代码 - 所以非常欢迎任何改进)。
但是我需要一些有关协程类型注释的帮助:
#!/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]
),但我不确定它们在这里是如何指定的。
非常欢迎任何帮助。
我自己找到了答案。
我进行了搜索,但在 Python 3.5.2 的
官方类型文档中没有找到关于
Generator
的 3 个类型参数的文档 - 除了真正神秘的提及...
class typing.Generator(Iterator[T_co], Generic[T_co, T_contra, V_co])
幸运的是,最初的 PEP484(这一切的开始)更有帮助:
“生成器函数的返回类型可以通过typing.py模块提供的泛型类型Generator[yield_type, send_type, return_type]来注释:
def echo_round() -> Generator[int, float, str]:
res = yield
while res:
res = yield round(res)
return 'OK'
基于此,我能够注释我的生成器,并看到
mypy
确认我的分配:
from typing import Callable, Generator
# A protocol decoder:
#
# - yields Nothing
# - expects ints to be `send` in his yield waits
# - and doesn't return anything.
ProtocolDecodingCoroutine = Generator[None, int, None]
# A frame consumer (passed as an argument to a protocol decoder):
#
# - yields Nothing
# - expects List[int] to be `send` in his waiting yields
# - and doesn't return anything.
FrameConsumerCoroutine = Generator[None, List[int], None]
def unwrap_protocol(header: int=0x61,
footer: int=0x62,
dle :int=0xAB,
after_dle_func: Callable[[int], int]=lambda x: x,
target: FrameConsumerCoroutine=None) -> ProtocolDecodingCoroutine:
...
def frame_receiver() -> FrameConsumerCoroutine:
...
我通过以下方式测试了我的作业:交换类型的顺序 - 然后正如预期的那样,
mypy
抱怨并要求正确的类型(如上所示)。
完整的代码可以从这里访问。
我会将这个问题留待几天,以防有人想插话 - 特别是在使用 Python 3.5 的新协程样式(
async def
等)方面 - 我希望能得到关于它们究竟如何使用的提示会用在这里。
如果您有一个使用
yield
的简单函数,那么您可以使用 Iterator
类型来注释其结果,而不是使用 Generator
:
from collections.abc import Iterator # Python >=3.9
def count_up() -> Iterator[int]:
for x in range(10):
yield x
如果功能是
async
,您需要使用AsyncIterator
代替:
from collections.abc import AsyncIterator # Python >=3.5
async def count_up() -> AsyncIterator[int]:
for x in range(10):
yield x
在Python中<3.9 you must import
Iterator
不同:
from typing import Iterator # Python <3.9
在撰写本文时,Python 文档也明确提到了如何处理异步情况(接受的答案中已经提到了非异步示例)。
引用那里:
async def echo_round() -> AsyncGenerator[int, float]:
sent = yield 0
while sent >= 0.0:
rounded = await round(sent)
sent = yield rounded
(第一个参数是yield-type,第二个参数是send-type)或简单情况(其中send-type是None)
async def infinite_stream(start: int) -> AsyncIterator[int]:
while True:
yield start
start = await increment(start)