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

Fix most mypy warnings

Browse Source
This commit is contained in:
Nat Noordanus
2020-06-07 19:10:41 +02:00
parent 3185c67098
commit c8229e53a7
4 changed files with 15 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
betterproto/grpc/grpclib_client.py
View File

@@ -3,9 +3,10 @@ import asyncio
import grpclib.const
from typing import (
Any,
AsyncIterable,
AsyncIterator,
Collection,
Iterator,
Iterable,
Mapping,
Optional,
Tuple,
@@ -23,7 +24,7 @@ if TYPE_CHECKING:
_Value = Union[str, bytes]
_MetadataLike = Union[Mapping[str, _Value], Collection[Tuple[str, _Value]]]
_MessageSource = Union[Iterator["IProtoMessage"], AsyncIterator["IProtoMessage"]]
_MessageSource = Union[Iterable["IProtoMessage"], AsyncIterable["IProtoMessage"]]
class ServiceStub(ABC):
@@ -160,7 +161,7 @@ class ServiceStub(ABC):
@staticmethod
async def _send_messages(stream, messages: _MessageSource):
if hasattr(messages, "__aiter__"):
if isinstance(messages, AsyncIterable):
async for message in messages:
await stream.send_message(message)
else: