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.
39 lines
1.4 KiB
39 lines
1.4 KiB
import backoff
|
|
|
|
from helpers.misc import backoff_handler, dict_merge
|
|
from media.pvr import PVR
|
|
from misc.log import logger
|
|
|
|
log = logger.get_logger(__name__)
|
|
|
|
|
|
class Radarr(PVR):
|
|
def get_objects(self):
|
|
return self._get_objects('api/movie')
|
|
|
|
@backoff.on_predicate(backoff.expo, lambda x: x is None, max_tries=4, on_backoff=backoff_handler)
|
|
def add_movie(self, movie_tmdb_id, movie_title, movie_year, movie_title_slug, profile_id, root_folder,
|
|
min_availability_temp, search_missing=False):
|
|
payload = self._prepare_add_object_payload(movie_title, movie_title_slug, profile_id, root_folder)
|
|
|
|
# replace radarr minimum_availability if supplied
|
|
if min_availability_temp == 'announced':
|
|
minimum_availability = 'announced'
|
|
elif min_availability_temp == 'in_cinemas':
|
|
minimum_availability = 'inCinemas'
|
|
elif min_availability_temp == 'predb':
|
|
minimum_availability = 'preDB'
|
|
else:
|
|
minimum_availability = 'released'
|
|
|
|
payload = dict_merge(payload, {
|
|
'tmdbId': movie_tmdb_id,
|
|
'year': movie_year,
|
|
'minimumAvailability': minimum_availability,
|
|
'addOptions': {
|
|
'searchForMovie': search_missing
|
|
}
|
|
})
|
|
|
|
return self._add_object('api/movie', payload, identifier_field='tmdbId', identifier=movie_tmdb_id)
|