Issue #1: add PackageNotFound exception

RickBarretto · May 21, 2022 · fd97b09 · fd97b09
1 parent 9db113a
commit fd97b09
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/core/fetch_api.py b/src/core/fetch_api.py
@@ -4,6 +4,7 @@
 from requests.exceptions import HTTPError
 
 from src.core.utils.osv_model import OsvModel, OsvUrl
+from src.core.utils.exceptions import PackageNotFound
 
 
 def fetch_api(osv_model: OsvModel) -> dict:
@@ -17,3 +18,8 @@ def fetch_api(osv_model: OsvModel) -> dict:
     json = response.json()
 
     return json
+
+
+def raise_for_not_found(json: dict):
+    if not json:
+        raise PackageNotFound
diff --git a/src/core/utils/exceptions.py b/src/core/utils/exceptions.py
@@ -0,0 +1,2 @@
+class PackageNotFound(Exception):
+    pass