Overhaul + Sonarr support!

1 files changed, 57 insertions, 0 deletions

diff --git a/code/utils/content_add.py b/code/utils/content_add.py
new file mode 100644
index 0000000..fda43af
--- /dev/null
+++ b/code/utils/content_add.py
@@ -0,0 +1,57 @@
+import requests
+
+
+def add_content(
+    content_info: dict,
+    service: str,
+    host: str,
+    headers: str,
+    folder_path: str,
+    profile_id: str,
+):
+    """
+    Add content to Sonarr or Radarr
+
+    Args:
+        content_info (dict): The content information
+        service (str): The service to add the content to
+        host (str): The host URL
+        headers (str): The headers for the request
+        folder_path (str): The folder path to download the content to
+        profile_id (str): The profile ID to download the content in
+
+    Returns:
+        str: The ID of the content or False
+    """
+    # Get the content data based on ID
+    data = requests.get(
+        url=(
+            f"{host}/api/v3/movie/lookup/tmdb?tmdbId={content_info['contentId']}"
+            if service == "radarr"
+            else f"{host}/api/v3/series/lookup?term=tvdb:{content_info['contentId']}"
+        ),
+        headers=headers,
+    ).json()[0]
+
+    data["monitored"] = True
+    data["qualityProfileId"] = profile_id
+    data["rootFolderPath"] = folder_path
+    # Search for the content on add
+    data["addOptions"] = {
+        (
+            "searchForMovie"
+            if service == "radarr"
+            else "searchForMissingEpisodes"
+        ): True
+    }
+    # Send the request to add the content
+    response = requests.post(
+        f"{host}/api/v3/{'movie' if service == 'radarr' else 'series'}",
+        headers=headers,
+        json=data,
+    )
+
+    if response.status_code == 201:
+        return response.json()["id"]
+    else:
+        return False