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

Merge pull request #52 from nat-n/fix_type_imports

Browse Source 
Only import types from grpclib when type checking
This commit is contained in:
nat 2020-05-24 19:09:08 +02:00 committed by GitHub
commit 0fe557bd3c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
betterproto/__init__.py
View File

@ -27,7 +27,7 @@ from typing import (
TYPE_CHECKING,
)
import grpclib.client
import grpclib.const
import stringcase
@ -35,6 +35,8 @@ from .casing import safe_snake_case
if TYPE_CHECKING:
from grpclib._protocols import IProtoMessage
from grpclib.client import Channel
from grpclib.metadata import Deadline
if not (sys.version_info.major == 3 and sys.version_info.minor >= 7):
# Apply backport of datetime.fromisoformat from 3.7
@ -1073,10 +1075,10 @@ class ServiceStub(ABC):
def __init__(
self,
channel: grpclib.client.Channel,
channel: "Channel",
*,
timeout: Optional[float] = None,
deadline: Optional[grpclib.metadata.Deadline] = None,
deadline: Optional["Deadline"] = None,
metadata: Optional[_MetadataLike] = None,
) -> None:
self.channel = channel
@ -1087,7 +1089,7 @@ class ServiceStub(ABC):
def __resolve_request_kwargs(
self,
timeout: Optional[float],
deadline: Optional[grpclib.metadata.Deadline],
deadline: Optional["Deadline"],
metadata: Optional[_MetadataLike],
):
return {
@ -1103,7 +1105,7 @@ class ServiceStub(ABC):
response_type: Type[T],
*,
timeout: Optional[float] = None,
deadline: Optional[grpclib.metadata.Deadline] = None,
deadline: Optional["Deadline"] = None,
metadata: Optional[_MetadataLike] = None,
) -> T:
"""Make a unary request and return the response."""
@ -1126,7 +1128,7 @@ class ServiceStub(ABC):
response_type: Type[T],
*,
timeout: Optional[float] = None,
deadline: Optional[grpclib.metadata.Deadline] = None,
deadline: Optional["Deadline"] = None,
metadata: Optional[_MetadataLike] = None,
) -> AsyncGenerator[T, None]:
"""Make a unary request and return the stream response iterator."""