def blacklist_auto_retry(self, request, pk):
watch_media = self.get_object()
nefarious_settings = NefariousSettings.get()
# add to blacklist
logger_foreground.info('Blacklisting {}'.format(
watch_media.transmission_torrent_hash))
TorrentBlacklist.objects.get_or_create(
hash=watch_media.transmission_torrent_hash,
defaults=dict(name=str(watch_media), ))
# unset previous details
del_transmission_torrent_hash = watch_media.transmission_torrent_hash
watch_media.transmission_torrent_hash = None
watch_media.collected = False
watch_media.collected_date = None
watch_media.save()
# re-queue search
self._watch_media_task(watch_media_id=watch_media.id)
# remove torrent and delete data
logger_foreground.info('Removing blacklisted torrent hash: {}'.format(
del_transmission_torrent_hash))
transmission_client = get_transmission_client(
nefarious_settings=nefarious_settings)
transmission_client.remove_torrent([del_transmission_torrent_hash],
delete_data=True)
return Response(self.serializer_class(watch_media).data)
def post(self, request):
result = {
'success': True,
}
nefarious_settings = NefariousSettings.get()
tmdb_media = request.data.get('tmdb_media', {})
torrent_info = request.data.get('torrent', {})
torrent_url = torrent_info.get('MagnetUri') or torrent_info.get('Link')
if not torrent_url:
return Response({
'success': False,
'error': 'Missing torrent link'
})
media_type = request.data.get('media_type', MEDIA_TYPE_TV)
# validate tv
if media_type == MEDIA_TYPE_TV:
if 'season_number' not in request.data:
return Response({
'success': False,
'error': 'Missing season_number'
})
if not is_magnet_url(torrent_url):
torrent_url = swap_jackett_host(torrent_url, nefarious_settings)
try:
torrent_url = trace_torrent_url(torrent_url)
except Exception as e:
return Response({
'success': False,
'error': 'An unknown error occurred',
'error_detail': str(e)
})
logger_foreground.info('adding torrent: {}'.format(torrent_url))
# add torrent
transmission_client = get_transmission_client(nefarious_settings)
transmission_session = transmission_client.session_stats()
tmdb = get_tmdb_client(nefarious_settings)
# set download paths and associate torrent with watch instance
if media_type == MEDIA_TYPE_MOVIE:
tmdb_request = tmdb.Movies(tmdb_media['id'])
tmdb_movie = tmdb_request.info()
watch_media = WatchMovie(
user=request.user,
tmdb_movie_id=tmdb_movie['id'],
name=tmdb_movie['title'],
poster_image_url=nefarious_settings.get_tmdb_poster_url(
tmdb_movie['poster_path']),
release_date=parse_date(tmdb_movie['release_date'] or ''),
)
watch_media.save()
download_dir = os.path.join(
transmission_session.download_dir,
nefarious_settings.transmission_movie_download_dir.lstrip('/'))
result['watch_movie'] = WatchMovieSerializer(watch_media).data
else:
tmdb_request = tmdb.TV(tmdb_media['id'])
tmdb_show = tmdb_request.info()
watch_tv_show, _ = WatchTVShow.objects.get_or_create(
tmdb_show_id=tmdb_show['id'],
defaults=dict(
user=request.user,
name=tmdb_show['name'],
poster_image_url=nefarious_settings.get_tmdb_poster_url(
tmdb_show['poster_path']),
))
result['watch_tv_show'] = WatchTVShowSerializer(watch_tv_show).data
# single episode
if 'episode_number' in request.data:
tmdb_request = tmdb.TV_Episodes(tmdb_media['id'],
request.data['season_number'],
request.data['episode_number'])
tmdb_episode = tmdb_request.info()
watch_media = WatchTVEpisode(
user=request.user,
watch_tv_show=watch_tv_show,
tmdb_episode_id=tmdb_episode['id'],
season_number=request.data['season_number'],
episode_number=request.data['episode_number'],
release_date=parse_date(tmdb_episode['air_date'] or ''))
watch_media.save()
result['watch_tv_episode'] = WatchTVEpisodeSerializer(
watch_media).data
# entire season
else:
season_result = tmdb.TV_Seasons(tmdb_show['id'],
request.data['season_number'])
season_data = season_result.info()
# create the season request
watch_tv_season_request, _ = WatchTVSeasonRequest.objects.get_or_create(
watch_tv_show=watch_tv_show,
season_number=request.data['season_number'],
defaults=dict(
user=request.user,
collected=
True, # set collected since we're directly downloading a torrent
release_date=parse_date(season_data['air_date']
or '')),
)
# create the actual watch season instance
watch_media = WatchTVSeason(
user=request.user,
watch_tv_show=watch_tv_show,
season_number=request.data['season_number'],
release_date=parse_date(season_data['air_date'] or ''))
watch_media.save()
# return the season request vs the watch instance
result[
'watch_tv_season_request'] = WatchTVSeasonRequestSerializer(
watch_tv_season_request).data
download_dir = os.path.join(
transmission_session.download_dir,
nefarious_settings.transmission_tv_download_dir.lstrip('/'))
torrent = transmission_client.add_torrent(
torrent_url,
paused=settings.DEBUG,
download_dir=download_dir,
)
watch_media.transmission_torrent_hash = torrent.hashString
watch_media.save()
return Response(result)