def completed_media_task(): nefarious_settings = NefariousSettings.get() transmission_client = get_transmission_client(nefarious_settings) incomplete_kwargs = dict(collected=False, transmission_torrent_hash__isnull=False) movies = WatchMovie.objects.filter(**incomplete_kwargs) tv_seasons = WatchTVSeason.objects.filter(**incomplete_kwargs) tv_episodes = WatchTVEpisode.objects.filter(**incomplete_kwargs) incomplete_media = list(movies) + list(tv_episodes) + list(tv_seasons) for media in incomplete_media: try: torrent = transmission_client.get_torrent( media.transmission_torrent_hash) except KeyError: # media's torrent reference no longer exists so remove the reference logger_background.info( "Media's torrent no longer present, removing reference: {}". format(media)) media.transmission_torrent_hash = None media.save() else: # download is complete if torrent.progress == 100: # flag media as completed logger_background.info('Media completed: {}'.format(media)) # special handling for tv seasons if isinstance(media, WatchTVSeason): # mark season request complete for season_request in WatchTVSeasonRequest.objects.filter( watch_tv_show=media.watch_tv_show, season_number=media.season_number): season_request.collected = True season_request.save() # get the sub path (ie. "movies/", "tv/') so we can move the data from staging sub_path = (nefarious_settings.transmission_movie_download_dir if isinstance(media, WatchMovie) else nefarious_settings.transmission_tv_download_dir ).lstrip('/') # get the path and updated name for the data new_path, new_name = get_media_new_path_and_name( media, torrent.name, len(torrent.files()) == 1) relative_path = os.path.join( sub_path, # i.e "movies" or "tv" new_path or '', ) # move the data to a new location transmission_session = transmission_client.session_stats() transmission_move_to_path = os.path.join( transmission_session.download_dir, relative_path, ) logger_background.info('Moving torrent data to "{}"'.format( transmission_move_to_path)) torrent.move_data(transmission_move_to_path) # rename the data logger_background.info( 'Renaming torrent file from "{}" to "{}"'.format( torrent.name, new_name)) transmission_client.rename_torrent_path( torrent.id, torrent.name, new_name) # save media as collected media.collected = True media.collected_date = timezone.now() media.save() # send websocket message media was updated media_type, data = websocket.get_media_type_and_serialized_watch_media( media) websocket.send_message(websocket.ACTION_UPDATED, media_type, data) # send complete message through webhook webhook.send_message('{} was downloaded'.format(media)) # define the import path import_path = os.path.join( settings.INTERNAL_DOWNLOAD_PATH, relative_path, # new_path will be None if the torrent is already a directory so fall back to the new name new_path or new_name, ) # post-tasks post_tasks = [ # queue import of media to save the actual media paths import_library_task.si( 'movie' if isinstance(media, WatchMovie) else 'tv', # media type media.user_id, # user id import_path, ), ] # conditionally add subtitles task to post-tasks if nefarious_settings.should_save_subtitles() and isinstance( media, (WatchMovie, WatchTVEpisode)): post_tasks.append( download_subtitles_task.si(media_type.lower(), media.id)) # queue post-tasks chain(*post_tasks)()
def completed_media_task(): nefarious_settings = NefariousSettings.get() transmission_client = get_transmission_client(nefarious_settings) incomplete_kwargs = dict(collected=False, transmission_torrent_hash__isnull=False) movies = WatchMovie.objects.filter(**incomplete_kwargs) tv_seasons = WatchTVSeason.objects.filter(**incomplete_kwargs) tv_episodes = WatchTVEpisode.objects.filter(**incomplete_kwargs) incomplete_media = list(movies) + list(tv_episodes) + list(tv_seasons) for media in incomplete_media: try: torrent = transmission_client.get_torrent( media.transmission_torrent_hash) except KeyError: # media's torrent reference no longer exists so remove the reference logging.info( "Media's torrent no longer present, removing reference: {}". format(media)) media.transmission_torrent_hash = None media.save() else: # download is complete if torrent.progress == 100: # flag media as completed logging.info('Media completed: {}'.format(media)) media.collected = True media.collected_date = datetime.utcnow() media.save() # special handling for tv seasons if isinstance(media, WatchTVSeason): # mark season request complete for season_request in WatchTVSeasonRequest.objects.filter( watch_tv_show=media.watch_tv_show, season_number=media.season_number): season_request.collected = True season_request.save() # delete any individual episodes now that we have the whole season for episode in WatchTVEpisode.objects.filter( watch_tv_show=media.watch_tv_show, season_number=media.season_number): episode.delete() # get the sub path (ie. "movies/", "tv/') so we can move the data from staging sub_path = (nefarious_settings.transmission_movie_download_dir if isinstance(media, WatchMovie) else nefarious_settings.transmission_tv_download_dir ).lstrip('/') # get the path and updated name for the data new_path, new_name = get_media_new_path_and_name( media, torrent.name, len(torrent.files()) == 1) # move the data transmission_session = transmission_client.session_stats() move_to_path = os.path.join( transmission_session.download_dir, sub_path, new_path or '', ) logging.info( 'Moving torrent data to "{}"'.format(move_to_path)) torrent.move_data(move_to_path) # rename the data logging.info('Renaming torrent file from "{}" to "{}"'.format( torrent.name, new_name)) transmission_client.rename_torrent_path( torrent.id, torrent.name, new_name)
def test_rename(self): for test_media, test_torrent, test_path, test_name, is_single_file in self.tests: path, name = get_media_new_path_and_name(test_media, test_torrent, is_single_file) self.assertEquals(test_name, name) self.assertEquals(test_path, path)