Skip to content
Snippets Groups Projects
Commit 843a010c authored by vlorentz's avatar vlorentz
Browse files

Fix mypy error

Without the return, 'psycopg2.connect(db_or_conninfo)' is actually reachable
when calling next() twice on the generator
parent c232fb09
No related branches found
No related tags found
No related merge requests found
......@@ -79,12 +79,12 @@ def connect_to_conninfo(
# Database name
db_or_conninfo = f"dbname={db_or_conninfo}"
try:
db = psycopg2.connect(db_or_conninfo)
except psycopg2.Error:
logger.exception("Failed to connect to `%s`", db_or_conninfo)
else:
yield db
try:
db = psycopg2.connect(db_or_conninfo)
except psycopg2.Error:
logger.exception("Failed to connect to `%s`", db_or_conninfo)
else:
yield db
def swh_db_version(db_or_conninfo: Union[str, pgconnection]) -> Optional[int]:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment