mirror of
https://github.com/meisnate12/Plex-Meta-Manager
synced 2024-11-10 15:04:21 +00:00
79 lines
3.6 KiB
Python
79 lines
3.6 KiB
Python
|
import logging, re, requests
|
||
|
|
from modules import util
|
||
|
|
from modules.util import Failed
|
||
|
|
from retrying import retry
|
||
|
|
|
||
|
|
logger = logging.getLogger("Plex Meta Manager")
|
||
|
|
|
||
|
|
class SonarrAPI:
|
||
|
|
def __init__(self, tvdb, params, language):
|
||
|
|
self.url_params = {"apikey": "{}".format(params["token"])}
|
||
|
|
self.base_url = "{}/api{}".format(params["url"], "/v3/" if params["version"] == "v3" else "/")
|
||
|
|
try:
|
||
|
|
response = requests.get("{}system/status".format(self.base_url), params=self.url_params)
|
||
|
|
result = response.json()
|
||
|
|
except Exception as e:
|
||
|
|
util.print_stacktrace()
|
||
|
|
raise Failed("Sonarr Error: Could not connect to Sonarr at {}".format(params["url"]))
|
||
|
|
if "error" in result and result["error"] == "Unauthorized":
|
||
|
|
raise Failed("Sonarr Error: Invalid API Key")
|
||
|
|
if "version" not in result:
|
||
|
|
raise Failed("Sonarr Error: Unexpected Response Check URL")
|
||
|
|
response = requests.get("{}{}".format(self.base_url, "qualityProfile" if params["version"] == "v3" else "profile"), params=self.url_params)
|
||
|
|
self.quality_profile_id = None
|
||
|
|
profiles = ""
|
||
|
|
for profile in response.json():
|
||
|
|
if len(profiles) > 0:
|
||
|
|
profiles += ", "
|
||
|
|
profiles += profile["name"]
|
||
|
|
if profile["name"] == params["quality_profile"]:
|
||
|
|
self.quality_profile_id = profile["id"]
|
||
|
|
if not self.quality_profile_id:
|
||
|
|
raise Failed("Sonarr Error: quality_profile: {} does not exist in sonarr. Profiles available: {}".format(params["quality_profile"], profiles))
|
||
|
|
self.tvdb = tvdb
|
||
|
|
self.language = language
|
||
|
|
self.url = params["url"]
|
||
|
|
self.version = params["version"]
|
||
|
|
self.token = params["token"]
|
||
|
|
self.root_folder_path = params["root_folder_path"]
|
||
|
|
self.add = params["add"]
|
||
|
|
self.search = params["search"]
|
||
|
|
|
||
|
|
def add_tvdb(self, tvdb_ids):
|
||
|
|
logger.info("")
|
||
|
|
logger.debug("TVDb IDs: {}".format(tvdb_ids))
|
||
|
|
add_count = 0
|
||
|
|
for tvdb_id in tvdb_ids:
|
||
|
|
try:
|
||
|
|
show = self.tvdb.get_series(self.language, tvdb_id=tvdb_id)
|
||
|
|
except Failed as e:
|
||
|
|
logger.error(e)
|
||
|
|
continue
|
||
|
|
|
||
|
|
titleslug = re.sub(r"([^\s\w]|_)+", "", show.title).replace(" ", "-").lower()
|
||
|
|
|
||
|
|
url_json = {
|
||
|
|
"title": show.title,
|
||
|
|
"{}".format("qualityProfileId" if self.version == "v3" else "profileId"): self.quality_profile_id,
|
||
|
|
"languageProfileId": 1,
|
||
|
|
"tvdbId": int(tvdb_id),
|
||
|
|
"titleslug": titleslug,
|
||
|
|
"language": self.language,
|
||
|
|
"monitored": True,
|
||
|
|
"rootFolderPath": self.root_folder_path,
|
||
|
|
"seasons" : [],
|
||
|
|
"images": [{"covertype": "poster", "url": show.poster_path}],
|
||
|
|
"addOptions": {"searchForMissingEpisodes": self.search}
|
||
|
|
}
|
||
|
|
response = self.send_post("{}series".format(self.base_url), url_json)
|
||
|
|
if response.status_code < 400:
|
||
|
|
logger.info("Added to Sonarr | {:<6} | {}".format(tvdb_id, show.title))
|
||
|
|
add_count += 1
|
||
|
|
else:
|
||
|
|
logger.error("Sonarr Error: ({}) {}: ({}) {}".format(tvdb_id, show.title, response.status_code, response.json()[0]["errorMessage"]))
|
||
|
|
logger.info("{} Show{} added to Sonarr".format(add_count, "s" if add_count > 1 else ""))
|
||
|
|
|
||
|
|
@retry(stop_max_attempt_number=6, wait_fixed=10000)
|
||
|
|
def send_post(self, url, url_json):
|
||
|
|
return requests.post(url, json=url_json, params=self.url_params)
|