pub/python-betterproto
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Include AsyncIterator import for both clients and servers (#264)

Browse Source 
Co-authored-by: Robin Lambertz <github@roblab.la>
This commit is contained in:
lazytype
2021-11-05 10:22:15 -04:00
committed by GitHub
parent 421fdba309
commit c424b6f8db
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/betterproto/grpc/grpclib_server.py
View File

@@ -1,5 +1,5 @@
from abc import ABC from abc import ABC
from collections import AsyncIterable from collections.abc import AsyncIterable
from typing import Callable, Any, Dict from typing import Callable, Any, Dict
import grpclib import grpclib

4
src/betterproto/plugin/models.py
View File

@@ -653,7 +653,9 @@ class ServiceMethodCompiler(ProtoContentBase):
self.output_file.typing_imports.add("AsyncIterable") self.output_file.typing_imports.add("AsyncIterable")
self.output_file.typing_imports.add("Iterable") self.output_file.typing_imports.add("Iterable")
self.output_file.typing_imports.add("Union") self.output_file.typing_imports.add("Union")
if self.server_streaming:
# Required by both client and server
if self.client_streaming or self.server_streaming:
self.output_file.typing_imports.add("AsyncIterator") self.output_file.typing_imports.add("AsyncIterator")
super().__post_init__() # check for unset fields super().__post_init__() # check for unset fields