fix: catch ConnectionResetError

ninoseki · Oct 5, 2023 · 942e7cd · 942e7cd
1 parent 05dc3e4
commit 942e7cd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/abuse_whois/api/endpoints/whois.py b/abuse_whois/api/endpoints/whois.py
@@ -21,3 +21,5 @@ async def whois(query: schemas.Query) -> schemas.Contacts:
         raise HTTPException(status.HTTP_404_NOT_FOUND, detail=str(e)) from e
     except errors.AbuseWhoisError as e:
         raise HTTPException(status.HTTP_500_INTERNAL_SERVER_ERROR, detail=str(e)) from e
+    except ConnectionResetError as e:
+        raise HTTPException(status.HTTP_500_INTERNAL_SERVER_ERROR, detail=str(e)) from e
diff --git a/abuse_whois/main.py b/abuse_whois/main.py
@@ -24,7 +24,7 @@ def whois(
     try:
         contacts = asyncio.run(get_abuse_contacts(address))
         print(contacts.model_dump_json(by_alias=True))  # noqa: T201
-    except (errors.AbuseWhoisError, asyncio.TimeoutError) as e:
+    except (errors.AbuseWhoisError, asyncio.TimeoutError, ConnectionResetError) as e:
         print(json.dumps({"error": str(e)}))  # noqa: T201