Merge pull request #52 from nat-n/fix_type_imports

Only import types from grpclib when type checking

Author: nat-n

betterproto/__init__.py

@@ -27,14 +27,16 @@
     TYPE_CHECKING,
 )
 
-import grpclib.client
+
 import grpclib.const
 import stringcase
 
 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 @@ def __init__(
     def __resolve_request_kwargs(
         self,
         timeout: Optional[float],
-        deadline: Optional[grpclib.metadata.Deadline],
+        deadline: Optional["Deadline"],
         metadata: Optional[_MetadataLike],
     ):
         return {
@@ -1103,7 +1105,7 @@ async def _unary_unary(
         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 @@ async def _unary_stream(
         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."""