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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 2 deletions
Show Stats 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