handle PackageNotFoundError

whatsonpypi/cli.py

@@ -20,7 +20,6 @@ def main(package):
     Example usages:
 
     $ whatsonpypi django
-
     """
     try:
         click.secho(get_query_response(package=package), fg='green', bold=True)

whatsonpypi/client.py

@@ -4,7 +4,7 @@
 from requests import Request, Session, hooks
 
 from .constants import PYPI_BASE_URL
-from .exceptions import PackageAbsentException
+from .exceptions import PackageNotProvidedError, PackageNotFoundError
 
 
 class WoppResponse:
@@ -48,7 +48,6 @@ def request(self, timeout=3.1, package=None, version=None):
         :return: response serialized by WoppResponse object
         """
         url = self._build_url(package, version)
-
         req_kwargs = {
             'method': 'GET',
             'url': url,
@@ -63,6 +62,10 @@ def request(self, timeout=3.1, package=None, version=None):
             prepared_request,
             timeout=timeout,
         )
+
+        if response.status_code == 404:
+            raise PackageNotFoundError("Sorry, but that package couldn't be found on PyPI.")
+
         # serialize response
         wopp_response = WoppResponse(int(response.status_code), response.json())
         return wopp_response
@@ -76,7 +79,7 @@ def _build_url(self, package=None, version=None):
         :return: fully qualified URL
         """
         if package is None:
-            raise PackageAbsentException('A package name is needed to proceed.')
+            raise PackageNotProvidedError('A package name is needed to proceed.')
 
         if version is not None:
             url = "{}/{}/{}/json".format(self.base_url, package, version)

whatsonpypi/exceptions.py

@@ -14,7 +14,13 @@ class WoppException(Exception):
     """
 
 
-class PackageAbsentException(WoppException):
+class PackageNotProvidedError(WoppException):
     """
     Raised when no package is available for the client to request
     """
+
+
+class PackageNotFoundError(WoppException):
+    """
+    Raised when a package is not found on PyPI
+    """

whatsonpypi/whatsonpypi.py

@@ -16,6 +16,7 @@ def get_query_response(package=None, version=None):
     """
     client = WoppClient(request_hooks={'response': clean_response})
     response = client.request(package=package, version=version)
+
     # returns version by default
     return response.latest_version