[WIP] Readarr Integration

butlarr/config/services.py

@@ -1,6 +1,7 @@
 from . import CONFIG
 from ..services.radarr import Radarr
 from ..services.sonarr import Sonarr
+from ..services.readarr import Readarr
 
 APIS = CONFIG["apis"]
 SERVICES = []
@@ -26,5 +27,13 @@
                 api_key=api_config["api_key"]
             )
         )
+    elif service_type == "Readarr":
+        SERVICES.append(
+            Readarr(
+                commands=commands,
+                api_host=api_config["api_host"],
+                api_key=api_config["api_key"]
+            )
+        )
     else:
         assert False, "Unsupported service type!"

butlarr/services/__init__.py

@@ -28,6 +28,7 @@ class ArrVariants(Enum):
     UNSUPPORTED = None
     RADARR = "movie"
     SONARR = "series"
+    READARR = "book"
 
 
 class ArrService(TelegramHandler):
@@ -71,20 +72,20 @@ def request(self, endpoint: str, *, action=Action.GET, params={}, fallback=None)
             return r.json()
         return r
 
-    def detect_api(self, api_host):
+    def detect_api(self, api_host, version="v3"):
         status = None
         # Detect version and api_url
         try:
-            self.api_url = f"{api_host.rstrip('/')}/api/v3"
+            self.api_url = f"{api_host.rstrip('/')}/api/{version}"
             status = self.request("system/status")
             if not status:
                 self.api_url = f"{api_host.rstrip('/')}/api"
                 status = self.request("system/status")
-                assert not status, "By default only v3 ArrServices are supported"
+                assert not status, f"By default only a {version} service is supported"
         finally:
             if status is None:
                 logger.error(
-                    "Could not reach compatible api. Is the service down? Is your API key correct?"
+                    f"Could not reach compatible api at {api_host}. Is the service down? Is your API key correct?"
                 )
                 exit(1)
             assert (
@@ -150,12 +151,6 @@ def add(
         self,
         *,
         item=None,
-        root_folder_path: str = "",
-        quality_profile_id: str = "",
-        language_profile_id: str = "",
-        min_availability="released",
-        tags: List[str] = [],
-        monitored=True,
         options={},
     ):
         assert item, "Missing required arg! You need to provide a item!"
@@ -165,12 +160,6 @@ def add(
             action=Action.POST,
             params={
                 **item,
-                "qualityProfileId": quality_profile_id,
-                "languageProfileId": language_profile_id,
-                "rootFolderPath": root_folder_path,
-                "tags": tags,
-                "monitored": monitored,
-                "minimumAvailability": min_availability,
                 **options,
             },
         )

butlarr/services/radarr.py

@@ -354,10 +354,14 @@ async def clbk_update(self, update, context, args, state):
     async def clbk_add(self, update, context, args, state):
         result = self.add(
             item=state.items[state.index],
-            quality_profile_id=state.quality_profile.get("id"),
-            root_folder_path=state.root_folder.get("path"),
-            tags=state.tags,
-            options={"addOptions": {"searchForMovie": args[1] == "search"}},
+            options={
+                "minimumAvailability": "released",
+                "monitored": True,
+                "qualityProfileId": state.quality_profile.get("id"),
+                "rootFolderPath": state.quality_profile.get("id"),
+                "tags": state.tags,
+                "addOptions": {"searchForMovie": args[1] == "search"},
+            },
         )
         if not result:
             return Response(caption="Seems like something went wrong...")