More specfic annotations on callbacks

diff --git a/httpcore/connection.py b/httpcore/connection.py
index d5caadeeef23d1a051e61f641911a8584b46ea13..a9c9890e8005bd2d51b2c27133a33e32fe25d84e 100644 (file)
--- a/httpcore/connection.py
+++ b/httpcore/connection.py
@@ -11,6 +11,9 @@ from .http11 import HTTP11Connection
 from .models import Client, Origin, Request, Response
 from .streams import Protocol, connect
 
+# Callback signature: async def callback(conn: HTTPConnection) -> None
+ReleaseCallback = typing.Callable[["HTTPConnection"], typing.Awaitable[None]]
+
 
 class HTTPConnection(Client):
     def __init__(
@@ -18,12 +21,12 @@ class HTTPConnection(Client):
         origin: typing.Union[str, Origin],
         ssl: SSLConfig = DEFAULT_SSL_CONFIG,
         timeout: TimeoutConfig = DEFAULT_TIMEOUT_CONFIG,
-        pool_release_func: typing.Callable = None,
+        release_func: typing.Optional[ReleaseCallback] = None,
     ):
         self.origin = Origin(origin) if isinstance(origin, str) else origin
         self.ssl = ssl
         self.timeout = timeout
-        self.pool_release_func = pool_release_func
+        self.release_func = release_func
         self.h11_connection = None  # type: typing.Optional[HTTP11Connection]
         self.h2_connection = None  # type: typing.Optional[HTTP2Connection]
 
@@ -59,10 +62,10 @@ class HTTPConnection(Client):
         port = self.origin.port
         ssl_context = await ssl.load_ssl_context() if self.origin.is_ssl else None
 
-        if self.pool_release_func is None:
+        if self.release_func is None:
             on_release = None
         else:
-            on_release = functools.partial(self.pool_release_func, self)
+            on_release = functools.partial(self.release_func, self)
 
         reader, writer, protocol = await connect(hostname, port, ssl_context, timeout)
         if protocol == Protocol.HTTP_2:

diff --git a/httpcore/connection_pool.py b/httpcore/connection_pool.py
index 709ad0f71384633b8cac7baca021369073f6e764..4ba83ec3dc99758f750b3a61680dfc2d39ba2b87 100644 (file)
--- a/httpcore/connection_pool.py
+++ b/httpcore/connection_pool.py
@@ -126,7 +126,7 @@ class ConnectionPool(Client):
                 origin,
                 ssl=self.ssl,
                 timeout=self.timeout,
-                pool_release_func=self.release_connection,
+                release_func=self.release_connection,
             )
 
         self.active_connections.add(connection)

diff --git a/httpcore/http11.py b/httpcore/http11.py
index 3280e1def8ff15a84f379db9b059b92fd66b51c6..3fda6d1f0d30e0a41c4d6cbb65257c865a69c5c5 100644 (file)
--- a/httpcore/http11.py
+++ b/httpcore/http11.py
@@ -19,6 +19,11 @@ H11Event = typing.Union[
 
 OptionalTimeout = typing.Optional[TimeoutConfig]
 
+# Callback signature: async def callback() -> None
+# In practice the callback will be a functools partial, which binds
+# the `ConnectionPool.release_connection(conn: HTTPConnection)` method.
+OnReleaseCallback = typing.Callable[[], typing.Awaitable[None]]
+
 
 class HTTP11Connection(Client):
     READ_NUM_BYTES = 4096
@@ -29,11 +34,11 @@ class HTTP11Connection(Client):
         writer: BaseWriter,
         origin: Origin,
         timeout: TimeoutConfig = DEFAULT_TIMEOUT_CONFIG,
-        on_release: typing.Callable = None,
+        on_release: typing.Optional[OnReleaseCallback] = None,
     ):
-        self.origin = origin
         self.reader = reader
         self.writer = writer
+        self.origin = origin
         self.timeout = timeout
         self.on_release = on_release
         self.h11_state = h11.Connection(our_role=h11.CLIENT)

diff --git a/httpcore/http2.py b/httpcore/http2.py
index e48c4de8e0d1d64993de75f7345027938c21ee81..c60490ce1203b6526e9fb0b9bbb6c317bb439d22 100644 (file)
--- a/httpcore/http2.py
+++ b/httpcore/http2.py
@@ -23,9 +23,9 @@ class HTTP2Connection(Client):
         timeout: TimeoutConfig = DEFAULT_TIMEOUT_CONFIG,
         on_release: typing.Callable = None,
     ):
-        self.origin = origin
         self.reader = reader
         self.writer = writer
+        self.origin = origin
         self.timeout = timeout
         self.on_release = on_release
         self.h2_state = h2.connection.H2Connection()