]> git.ipfire.org Git - thirdparty/psycopg.git/commitdiff
refactor: use `yield from` in connections attempts
authorDaniele Varrazzo <daniele.varrazzo@gmail.com>
Thu, 30 Nov 2023 09:11:46 +0000 (10:11 +0100)
committerDaniele Varrazzo <daniele.varrazzo@gmail.com>
Fri, 1 Dec 2023 12:56:18 +0000 (13:56 +0100)
psycopg/psycopg/conninfo.py

index 293ba99febfe449dc5e8e6667e496808963be569..bdb5526d3ea3f000e2589dac5c7664bbdb687fe7 100644 (file)
@@ -294,11 +294,9 @@ def conninfo_attempts(params: ConnDict) -> Iterator[ConnDict]:
     if params.get("load_balance_hosts", "disable") == "random":
         attempts = list(_split_attempts(_inject_defaults(params)))
         shuffle(attempts)
-        for attempt in attempts:
-            yield attempt
+        yield from attempts
     else:
-        for attempt in _split_attempts(_inject_defaults(params)):
-            yield attempt
+        yield from _split_attempts(_inject_defaults(params))
 
 
 async def conninfo_attempts_async(params: ConnDict) -> AsyncIterator[ConnDict]: