Add SonarrResourceNotFound Exception (#13)

* add SonarrResourceNotFound exception

* Update __init__.py

* Update client.py

* Update test_client.py

* Update client.py

* Update client.py

* Update client.py

Author: ctalkington

sonarr/__init__.py

@@ -3,5 +3,6 @@
     SonarrAccessRestricted,
     SonarrConnectionError,
     SonarrError,
+    SonarrResourceNotFound,
 )
 from .sonarr import Client, Sonarr  # noqa

sonarr/client.py

@@ -9,7 +9,12 @@
 from yarl import URL
 
 from .__version__ import __version__
-from .exceptions import SonarrAccessRestricted, SonarrConnectionError, SonarrError
+from .exceptions import (
+    SonarrAccessRestricted,
+    SonarrConnectionError,
+    SonarrError,
+    SonarrResourceNotFound,
+)
 
 
 class Client:
@@ -94,6 +99,9 @@ async def _request(
                 "Access restricted. Please ensure valid API Key is provided", {}
             )
 
+        if response.status == 404:
+            raise SonarrResourceNotFound("Resource not found")
+
         content_type = response.headers.get("Content-Type", "")
 
         if (response.status // 100) in [4, 5]:

sonarr/exceptions.py

@@ -14,6 +14,12 @@ class SonarrConnectionError(SonarrError):
 
 
 class SonarrAccessRestricted(SonarrError):
-    """Sonarr access restricted."""
+    """Sonarr access restricted exception."""
+
+    pass
+
+
+class SonarrResourceNotFound(SonarrError):
+    """Sonarr resource not found exception."""
 
     pass

tests/test_client.py

@@ -4,7 +4,12 @@
 import pytest
 from aiohttp import ClientSession
 from sonarr import Client
-from sonarr.exceptions import SonarrAccessRestricted, SonarrConnectionError, SonarrError
+from sonarr.exceptions import (
+    SonarrAccessRestricted,
+    SonarrConnectionError,
+    SonarrError,
+    SonarrResourceNotFound,
+)
 
 API_KEY = "MOCK_API_KEY"
 HOST = "192.168.1.89"
@@ -169,7 +174,7 @@ async def test_http_error404(aresponses):
 
     async with ClientSession() as session:
         client = Client(HOST, API_KEY, session=session)
-        with pytest.raises(SonarrError):
+        with pytest.raises(SonarrResourceNotFound):
             assert await client._request("system/status")