237 lines
14 KiB
C#
237 lines
14 KiB
C#
using System;
|
||
using System.Collections.Generic;
|
||
using System.IO;
|
||
using System.Linq;
|
||
using NLog;
|
||
using NzbDrone.Core.Model;
|
||
using NzbDrone.Core.Repository;
|
||
using SubSonic.Repository;
|
||
|
||
namespace NzbDrone.Core.Providers
|
||
{
|
||
public class EpisodeProvider
|
||
{
|
||
private static readonly Logger Logger = LogManager.GetCurrentClassLogger();
|
||
private readonly SeasonProvider _seasonsProvider;
|
||
private readonly IRepository _repository;
|
||
private readonly TvDbProvider _tvDbProvider;
|
||
|
||
public EpisodeProvider(IRepository repository, SeasonProvider seasonProviderProvider, TvDbProvider tvDbProviderProvider)
|
||
{
|
||
_repository = repository;
|
||
_tvDbProvider = tvDbProviderProvider;
|
||
_seasonsProvider = seasonProviderProvider;
|
||
}
|
||
|
||
public EpisodeProvider()
|
||
{
|
||
}
|
||
|
||
public virtual Episode GetEpisode(long id)
|
||
{
|
||
return _repository.Single<Episode>(id);
|
||
}
|
||
|
||
public virtual Episode GetEpisode(int seriesId, int seasonNumber, int episodeNumber)
|
||
{
|
||
return
|
||
_repository.Single<Episode>(
|
||
c => c.SeriesId == seriesId && c.SeasonNumber == seasonNumber && c.EpisodeNumber == episodeNumber);
|
||
}
|
||
|
||
public virtual Episode GetEpisode(int seriesId, DateTime date)
|
||
{
|
||
return
|
||
_repository.Single<Episode>(
|
||
c => c.SeriesId == seriesId && c.AirDate == date.Date);
|
||
}
|
||
|
||
public virtual IList<Episode> GetEpisodeBySeries(long seriesId)
|
||
{
|
||
return _repository.Find<Episode>(e => e.SeriesId == seriesId);
|
||
}
|
||
|
||
public virtual IList<Episode> GetEpisodeBySeason(long seasonId)
|
||
{
|
||
return _repository.Find<Episode>(e => e.SeasonId == seasonId);
|
||
}
|
||
|
||
public virtual IList<Episode> GetEpisodeByParseResult(EpisodeParseResult parseResult)
|
||
{
|
||
var seasonEpisodes = _repository.All<Episode>().Where(e =>
|
||
e.SeriesId == parseResult.SeriesId &&
|
||
e.SeasonNumber == parseResult.SeasonNumber).ToList();
|
||
|
||
//Has to be done separately since subsonic doesn't support contain method
|
||
return seasonEpisodes.Where(c => parseResult.Episodes.Contains(c.EpisodeNumber)).ToList();
|
||
|
||
}
|
||
|
||
public virtual IList<Episode> EpisodesWithoutFiles(bool includeSpecials)
|
||
{
|
||
if (includeSpecials)
|
||
return _repository.All<Episode>().Where(e => e.EpisodeFileId == 0 && e.AirDate <= DateTime.Today).ToList();
|
||
|
||
return _repository.All<Episode>().Where(e => e.EpisodeFileId == 0 && e.AirDate <= DateTime.Today && e.SeasonNumber > 0).ToList();
|
||
}
|
||
|
||
/// <summary>
|
||
/// Comprehensive check on whether or not this episode is needed.
|
||
/// </summary>
|
||
/// <param name = "parsedReport">Episode that needs to be checked</param>
|
||
/// <returns></returns>
|
||
public virtual bool IsNeeded(EpisodeParseResult parsedReport)
|
||
{
|
||
//Todo: Fix this so it properly handles multi-epsiode releases (Currently as long as the first episode is needed we download it)
|
||
//Todo: for small releases this is less of an issue, but for Full Season Releases this could be an issue if we only need the first episode (or first few)
|
||
foreach (var episode in parsedReport.Episodes)
|
||
{
|
||
var episodeInfo = GetEpisode(parsedReport.SeriesId, parsedReport.SeasonNumber, episode);
|
||
|
||
if (episodeInfo == null)
|
||
{
|
||
Logger.Debug("Episode S{0:00}E{1:00} doesn't exist in db. adding it now.", parsedReport.SeasonNumber, episode);
|
||
//Todo: How do we want to handle this really? Episode could be released before information is on TheTvDB
|
||
//(Parks and Rec did this a lot in the first season, from experience)
|
||
//Keivan: Should automatically add the episode to db with minimal information. then update the description/title when available.
|
||
//Mark: Perhaps we should only add the epsiode if its the latest season, sometimes people name things completely wrong (duh!)
|
||
episodeInfo = new Episode
|
||
{
|
||
SeriesId = parsedReport.SeriesId,
|
||
AirDate = DateTime.Now.Date,
|
||
EpisodeNumber = episode,
|
||
SeasonNumber = parsedReport.SeasonNumber,
|
||
Title = String.Empty,
|
||
Overview = String.Empty,
|
||
Language = "en"
|
||
};
|
||
|
||
_repository.Add(episodeInfo);
|
||
|
||
}
|
||
|
||
var file = episodeInfo.EpisodeFile;
|
||
|
||
if (file != null)
|
||
{
|
||
Logger.Debug("File is {0} Proper:{1}", file.Quality, file.Proper);
|
||
|
||
//There will never be a time when the episode quality is less than what we have and we want it... ever.... I think.
|
||
if (file.Quality > parsedReport.Quality)
|
||
{
|
||
Logger.Trace("file has better quality. skipping");
|
||
continue;
|
||
}
|
||
|
||
//If not null we need to see if this episode has the quality as the download (or if it is better)
|
||
if (file.Quality == parsedReport.Quality && file.Proper == parsedReport.Proper)
|
||
{
|
||
Logger.Trace("Same quality/proper. existing proper. skipping");
|
||
continue;
|
||
}
|
||
|
||
//Now we need to handle upgrades and actually pay attention to the Cut-off Value
|
||
if (file.Quality < parsedReport.Quality)
|
||
{
|
||
if (episodeInfo.Series.QualityProfile.Cutoff <= file.Quality)
|
||
{
|
||
Logger.Trace("Quality is past cut-off skipping.");
|
||
continue;
|
||
}
|
||
}
|
||
}
|
||
|
||
Logger.Debug("Episode {0} is needed", parsedReport);
|
||
return true; //If we get to this point and the file has not yet been rejected then accept it
|
||
}
|
||
|
||
Logger.Debug("Episode {0} is not needed", parsedReport);
|
||
return false;
|
||
}
|
||
|
||
public virtual void RefreshEpisodeInfo(int seriesId)
|
||
{
|
||
Logger.Info("Starting episode info refresh for series:{0}", seriesId);
|
||
int successCount = 0;
|
||
int failCount = 0;
|
||
var targetSeries = _tvDbProvider.GetSeries(seriesId, true);
|
||
|
||
var updateList = new List<Episode>();
|
||
var newList = new List<Episode>();
|
||
|
||
Logger.Debug("Updating season info for series:{0}", targetSeries.SeriesName);
|
||
targetSeries.Episodes.Select(e => new { e.SeasonId, e.SeasonNumber })
|
||
.Distinct().ToList()
|
||
.ForEach(s => _seasonsProvider.EnsureSeason(seriesId, s.SeasonId, s.SeasonNumber));
|
||
|
||
foreach (var episode in targetSeries.Episodes)
|
||
{
|
||
try
|
||
{
|
||
//DateTime throws an error in SQLServer per message below:
|
||
//SqlDateTime overflow. Must be between 1/1/1753 12:00:00 AM and 12/31/9999 11:59:59 PM.
|
||
//So lets hack it so it works for SQLServer (as well as SQLite), perhaps we can find a better solution
|
||
//Todo: Fix this hack
|
||
if (episode.FirstAired < new DateTime(1753, 1, 1))
|
||
episode.FirstAired = new DateTime(1753, 1, 1);
|
||
|
||
Logger.Trace("Updating info for [{0}] - S{1}E{2}", targetSeries.SeriesName, episode.SeasonNumber, episode.EpisodeNumber);
|
||
var newEpisode = new Episode
|
||
{
|
||
AirDate = episode.FirstAired.Date,
|
||
TvDbEpisodeId = episode.Id,
|
||
EpisodeNumber = episode.EpisodeNumber,
|
||
Language = episode.Language.Abbriviation,
|
||
Overview = episode.Overview,
|
||
SeasonId = episode.SeasonId,
|
||
SeasonNumber = episode.SeasonNumber,
|
||
SeriesId = seriesId,
|
||
Title = episode.EpisodeName,
|
||
LastInfoSync = DateTime.Now
|
||
|
||
};
|
||
|
||
var existingEpisode = GetEpisode(episode.SeriesId, episode.SeasonNumber, episode.EpisodeNumber);
|
||
|
||
if (existingEpisode != null)
|
||
{
|
||
//TODO: Write test for this, possibly make it future proof, we should only copy fields that come from tvdb
|
||
newEpisode.EpisodeId = existingEpisode.EpisodeId;
|
||
newEpisode.EpisodeFileId = existingEpisode.EpisodeFileId;
|
||
newEpisode.LastDiskSync = existingEpisode.LastDiskSync;
|
||
newEpisode.Status = existingEpisode.Status;
|
||
updateList.Add(newEpisode);
|
||
}
|
||
else
|
||
{
|
||
newList.Add(newEpisode);
|
||
}
|
||
|
||
successCount++;
|
||
}
|
||
catch (Exception e)
|
||
{
|
||
Logger.FatalException(
|
||
String.Format("An error has occurred while updating episode info for series {0}", seriesId), e);
|
||
failCount++;
|
||
}
|
||
}
|
||
|
||
_repository.AddMany(newList);
|
||
_repository.UpdateMany(updateList);
|
||
|
||
Logger.Debug("Finished episode refresh for series:{0}. Successful:{1} - Failed:{2} ",
|
||
targetSeries.SeriesName, successCount, failCount);
|
||
}
|
||
|
||
public virtual void DeleteEpisode(int episodeId)
|
||
{
|
||
_repository.Delete<Episode>(episodeId);
|
||
}
|
||
|
||
public virtual void UpdateEpisode(Episode episode)
|
||
{
|
||
_repository.Update(episode);
|
||
}
|
||
}
|
||
} |