Automatically add future annotations import (#58)

Signed-off-by: Anuraag Agrawal <anuraaga@gmail.com>

Author: anuraaga

conformance/test/_util.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import asyncio
 import sys
 

conformance/test/client.py

@@ -1,12 +1,13 @@
+from __future__ import annotations
+
 import argparse
 import asyncio
 import ssl
 import sys
 import time
 import traceback
-from collections.abc import AsyncIterator, Iterator
 from tempfile import NamedTemporaryFile
-from typing import Literal, TypeVar
+from typing import TYPE_CHECKING, Literal, TypeVar
 
 import httpx
 from _util import create_standard_streams
@@ -29,14 +30,18 @@
     UnaryRequest,
     UnimplementedRequest,
 )
-from google.protobuf.any_pb2 import Any
 from google.protobuf.message import Message
 
 from connectrpc.client import ResponseMetadata
 from connectrpc.code import Code
 from connectrpc.errors import ConnectError
 from connectrpc.request import Headers
 
+if TYPE_CHECKING:
+    from collections.abc import AsyncIterator, Iterator
+
+    from google.protobuf.any_pb2 import Any
+
 
 def _convert_code(error: Code) -> ConformanceCode:
     match error:

conformance/test/gen/connectrpc/conformance/v1/service_connect.py

@@ -1,15 +1,13 @@
 # Generated by https://github.com/connectrpc/connect-python.  DO NOT EDIT!
 # source: connectrpc/conformance/v1/service.proto
+from __future__ import annotations
 
-from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping
-from typing import Protocol
+from typing import TYPE_CHECKING, Protocol
 
 from connectrpc.client import ConnectClient, ConnectClientSync
 from connectrpc.code import Code
 from connectrpc.errors import ConnectError
-from connectrpc.interceptor import Interceptor, InterceptorSync
 from connectrpc.method import IdempotencyLevel, MethodInfo
-from connectrpc.request import Headers, RequestContext
 from connectrpc.server import (
     ConnectASGIApplication,
     ConnectWSGIApplication,
@@ -19,6 +17,18 @@
 
 from . import service_pb2 as connectrpc_dot_conformance_dot_v1_dot_service__pb2
 
+if TYPE_CHECKING:
+    from collections.abc import (
+        AsyncGenerator,
+        AsyncIterator,
+        Iterable,
+        Iterator,
+        Mapping,
+    )
+
+    from connectrpc.interceptor import Interceptor, InterceptorSync
+    from connectrpc.request import Headers, RequestContext
+
 
 class ConformanceService(Protocol):
     async def unary(

conformance/test/server.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import argparse
 import asyncio
 import os
@@ -7,7 +9,6 @@
 import ssl
 import sys
 import time
-from collections.abc import AsyncIterator, Iterator
 from contextlib import ExitStack, closing
 from tempfile import NamedTemporaryFile
 from typing import TYPE_CHECKING, Literal, TypeVar, get_args
@@ -43,14 +44,17 @@
 
 from connectrpc.code import Code
 from connectrpc.errors import ConnectError
-from connectrpc.request import RequestContext
 
 if TYPE_CHECKING:
+    from collections.abc import AsyncIterator, Iterator
+
     from google.protobuf.message import Message
 
+    from connectrpc.request import RequestContext
+
 
 # TODO: Use google.protobuf.any.pack on upgrade to protobuf==6.
-def pack(msg: "Message") -> Any:
+def pack(msg: Message) -> Any:
     any_msg = Any()
     any_msg.Pack(msg)
     return any_msg

conformance/test/test_client.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import subprocess
 import sys
 from pathlib import Path

conformance/test/test_server.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import os
 import subprocess
 import sys

example/example/_eliza.py

@@ -1,6 +1,11 @@
+from __future__ import annotations
+
 import random
 import re
-from collections.abc import Sequence
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from collections.abc import Sequence
 
 # Ported from https://github.com/connectrpc/examples-go
 # Originally from https://github.com/mattshiel/eliza-go

example/example/eliza_client.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import asyncio
 
 from example.eliza_connect import ElizaServiceClient

example/example/eliza_client_sync.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from example.eliza_connect import ElizaServiceClientSync
 from example.eliza_pb2 import IntroduceRequest, SayRequest
 

example/example/eliza_connect.py

@@ -1,15 +1,13 @@
 # Generated by https://github.com/connectrpc/connect-python.  DO NOT EDIT!
 # source: example/eliza.proto
+from __future__ import annotations
 
-from collections.abc import AsyncGenerator, AsyncIterator, Iterable, Iterator, Mapping
-from typing import Protocol
+from typing import TYPE_CHECKING, Protocol
 
 from connectrpc.client import ConnectClient, ConnectClientSync
 from connectrpc.code import Code
 from connectrpc.errors import ConnectError
-from connectrpc.interceptor import Interceptor, InterceptorSync
 from connectrpc.method import IdempotencyLevel, MethodInfo
-from connectrpc.request import Headers, RequestContext
 from connectrpc.server import (
     ConnectASGIApplication,
     ConnectWSGIApplication,
@@ -19,6 +17,18 @@
 
 import example.eliza_pb2 as example_dot_eliza__pb2
 
+if TYPE_CHECKING:
+    from collections.abc import (
+        AsyncGenerator,
+        AsyncIterator,
+        Iterable,
+        Iterator,
+        Mapping,
+    )
+
+    from connectrpc.interceptor import Interceptor, InterceptorSync
+    from connectrpc.request import Headers, RequestContext
+
 
 class ElizaService(Protocol):
     async def say(