mirror of https://github.com/l3uddz/traktarr
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
2.1 KiB
60 lines
2.1 KiB
import os.path
|
|
|
|
import backoff
|
|
import requests
|
|
from helpers.misc import backoff_handler, dict_merge
|
|
|
|
from helpers import str as misc_str
|
|
from media.pvr import PVR
|
|
from misc.log import logger
|
|
|
|
log = logger.get_logger(__name__)
|
|
|
|
|
|
class Sonarr(PVR):
|
|
def get_objects(self):
|
|
return self._get_objects('api/series')
|
|
|
|
@backoff.on_predicate(backoff.expo, lambda x: x is None, max_tries=4, on_backoff=backoff_handler)
|
|
def get_tags(self):
|
|
tags = {}
|
|
try:
|
|
# make request
|
|
req = requests.get(
|
|
os.path.join(misc_str.ensure_endswith(self.server_url, "/"), 'api/tag'),
|
|
headers=self.headers,
|
|
timeout=60,
|
|
allow_redirects=False
|
|
)
|
|
log.debug("Request URL: %s", req.url)
|
|
log.debug("Request Response: %d", req.status_code)
|
|
|
|
if req.status_code == 200:
|
|
resp_json = req.json()
|
|
log.debug("Found Sonarr Tags: %d", len(resp_json))
|
|
for tag in resp_json:
|
|
tags[tag['label']] = tag['id']
|
|
return tags
|
|
else:
|
|
log.error("Failed to retrieve all tags, request response: %d", req.status_code)
|
|
except Exception:
|
|
log.exception("Exception retrieving tags: ")
|
|
return None
|
|
|
|
@backoff.on_predicate(backoff.expo, lambda x: x is None, max_tries=4, on_backoff=backoff_handler)
|
|
def add_series(self, series_tvdbid, series_title, series_title_slug, profile_id, root_folder, tag_ids=None,
|
|
search_missing=False):
|
|
payload = self._prepare_add_object_payload(series_title, series_title_slug, profile_id, root_folder)
|
|
|
|
payload = dict_merge(payload, {
|
|
'tvdbId': series_tvdbid,
|
|
'tags': [] if not tag_ids or not isinstance(tag_ids, list) else tag_ids,
|
|
'seasons': [],
|
|
'seasonFolder': True,
|
|
'addOptions': {
|
|
'searchForMissingEpisodes': search_missing
|
|
}
|
|
})
|
|
|
|
return self._add_object('api/series', payload, identifier_field='tvdbId', identifier=series_tvdbid)
|