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

Only import types from grpclib when type checking

Browse Source
This commit is contained in:
Nat Noordanus 2020-05-22 18:41:29 +02:00
parent 3546f55146
commit 33fb83faad
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, TYPE_CHECKING,
) )
import grpclib.client
import grpclib.const import grpclib.const
import stringcase import stringcase
@ -35,6 +35,8 @@ from .casing import safe_snake_case
if TYPE_CHECKING: if TYPE_CHECKING:
from grpclib._protocols import IProtoMessage 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): if not (sys.version_info.major == 3 and sys.version_info.minor >= 7):
# Apply backport of datetime.fromisoformat from 3.7 # Apply backport of datetime.fromisoformat from 3.7
@ -1013,10 +1015,10 @@ class ServiceStub(ABC):
def __init__( def __init__(
self, self,
channel: grpclib.client.Channel, channel: 'Channel',
*, *,
timeout: Optional[float] = None, timeout: Optional[float] = None,
deadline: Optional[grpclib.metadata.Deadline] = None, deadline: Optional['Deadline'] = None,
metadata: Optional[_MetadataLike] = None, metadata: Optional[_MetadataLike] = None,
) -> None: ) -> None:
self.channel = channel self.channel = channel
@ -1027,7 +1029,7 @@ class ServiceStub(ABC):
def __resolve_request_kwargs( def __resolve_request_kwargs(
self, self,
timeout: Optional[float], timeout: Optional[float],
deadline: Optional[grpclib.metadata.Deadline], deadline: Optional['Deadline'],
metadata: Optional[_MetadataLike], metadata: Optional[_MetadataLike],
): ):
return { return {
@ -1043,7 +1045,7 @@ class ServiceStub(ABC):
response_type: Type[T], response_type: Type[T],
*, *,
timeout: Optional[float] = None, timeout: Optional[float] = None,
deadline: Optional[grpclib.metadata.Deadline] = None, deadline: Optional['Deadline'] = None,
metadata: Optional[_MetadataLike] = None, metadata: Optional[_MetadataLike] = None,
) -> T: ) -> T:
"""Make a unary request and return the response.""" """Make a unary request and return the response."""
@ -1066,7 +1068,7 @@ class ServiceStub(ABC):
response_type: Type[T], response_type: Type[T],
*, *,
timeout: Optional[float] = None, timeout: Optional[float] = None,
deadline: Optional[grpclib.metadata.Deadline] = None, deadline: Optional['Deadline'] = None,
metadata: Optional[_MetadataLike] = None, metadata: Optional[_MetadataLike] = None,
) -> AsyncGenerator[T, None]: ) -> AsyncGenerator[T, None]:
"""Make a unary request and return the stream response iterator.""" """Make a unary request and return the stream response iterator."""