clean

encode · Jun 15, 2024 · 6e4b9aa · 6e4b9aa
1 parent b6fd02c
commit 6e4b9aa
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 12 deletions.
diff --git a/httpcore/_async/connection_pool.py b/httpcore/_async/connection_pool.py
@@ -5,7 +5,6 @@
     AsyncIterable,
     AsyncIterator,
     Iterable,
-    Iterator,
     List,
     Optional,
     Type,
@@ -272,12 +271,15 @@ def _assign_requests_to_connections(self) -> List[AsyncConnectionInterface]:
                 continue
 
             origin = pool_request.request.url.origin
-            available_connections_iter: Iterator[AsyncConnectionInterface] = (
-                connection
-                for connection in self._connections
-                if connection.can_handle_request(origin) and connection.is_available()
+            available_connection = next(
+                (
+                    connection
+                    for connection in self._connections
+                    if connection.can_handle_request(origin)
+                    and connection.is_available()
+                ),
+                None,
             )
-            available_connection = next(available_connections_iter, None)
 
             # There are three cases for how we may be able to handle the request:
             #

diff --git a/httpcore/_sync/connection_pool.py b/httpcore/_sync/connection_pool.py
@@ -5,7 +5,6 @@
     Iterable,
     Iterator,
     Iterable,
-    Iterator,
     List,
     Optional,
     Type,
@@ -272,12 +271,15 @@ def _assign_requests_to_connections(self) -> List[ConnectionInterface]:
                 continue
 
             origin = pool_request.request.url.origin
-            available_connections_iter: Iterator[ConnectionInterface] = (
-                connection
-                for connection in self._connections
-                if connection.can_handle_request(origin) and connection.is_available()
+            available_connection = next(
+                (
+                    connection
+                    for connection in self._connections
+                    if connection.can_handle_request(origin)
+                    and connection.is_available()
+                ),
+                None,
             )
-            available_connection = next(available_connections_iter, None)
 
             # There are three cases for how we may be able to handle the request:
             #