mirror of
https://github.com/Sonarr/Sonarr.git
synced 2024-12-14 11:23:42 +02:00
560706ae8f
PostDownload Importing setup, need to setup job for period scan.
201 lines
11 KiB
C#
201 lines
11 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 IRepository _repository;
|
|
private readonly TvDbProvider _tvDbProvider;
|
|
|
|
public EpisodeProvider(IRepository repository, TvDbProvider tvDbProviderProvider)
|
|
{
|
|
_repository = repository;
|
|
_tvDbProvider = tvDbProviderProvider;
|
|
}
|
|
|
|
public EpisodeProvider()
|
|
{
|
|
}
|
|
|
|
public virtual void AddEpisode(Episode episode)
|
|
{
|
|
_repository.Add(episode);
|
|
}
|
|
|
|
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 IQueryable<Episode> GetEpisodeBySeries(long seriesId)
|
|
{
|
|
return _repository.All<Episode>().Where(e => e.SeriesId == seriesId);
|
|
}
|
|
|
|
public virtual IList<Episode> GetEpisodesBySeason(long seriesId, int seasonNumber)
|
|
{
|
|
return _repository.Find<Episode>(e => e.SeriesId == seriesId && e.SeasonNumber == seasonNumber);
|
|
}
|
|
|
|
public virtual List<Episode> GetEpisodes(EpisodeParseResult parseResult)
|
|
{
|
|
if (parseResult.Series == null)
|
|
{
|
|
Logger.Debug("Episode Parse Result is Invalid, skipping");
|
|
return new List<Episode>();
|
|
}
|
|
|
|
var episodes = new List<Episode>();
|
|
|
|
foreach (var ep in parseResult.EpisodeNumbers)
|
|
{
|
|
var episode = GetEpisode(parseResult.Series.SeriesId, parseResult.SeasonNumber, ep);
|
|
|
|
if (episode == null)
|
|
return new List<Episode>();
|
|
|
|
episodes.Add(episode);
|
|
}
|
|
|
|
return episodes;
|
|
}
|
|
|
|
public virtual IList<Episode> EpisodesWithoutFiles(bool includeSpecials)
|
|
{
|
|
if (includeSpecials)
|
|
return _repository.All<Episode>().Where(e => e.EpisodeFileId == 0 && e.AirDate <= DateTime.Today && e.AirDate > new DateTime(1899, 12, 31)).ToList();
|
|
|
|
return _repository.All<Episode>().Where(e => e.EpisodeFileId == 0 && e.AirDate <= DateTime.Today && e.AirDate > new DateTime(1899, 12, 31) && e.SeasonNumber > 0).ToList();
|
|
}
|
|
|
|
public virtual IList<Episode> EpisodesByFileId(int episodeFileId)
|
|
{
|
|
return _repository.All<Episode>().Where(e => e.EpisodeFileId == episodeFileId).ToList();
|
|
}
|
|
|
|
public virtual void RefreshEpisodeInfo(Series series)
|
|
{
|
|
Logger.Info("Starting episode info refresh for series:{0}", series.SeriesId);
|
|
int successCount = 0;
|
|
int failCount = 0;
|
|
var tvDbSeriesInfo = _tvDbProvider.GetSeries(series.SeriesId, true);
|
|
|
|
var updateList = new List<Episode>();
|
|
var newList = new List<Episode>();
|
|
|
|
foreach (var episode in tvDbSeriesInfo.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}", tvDbSeriesInfo.SeriesName, episode.SeasonNumber, episode.EpisodeNumber);
|
|
|
|
//first check using tvdbId, this should cover cases when and episode number in a season is changed
|
|
var episodeToUpdate = _repository.Single<Episode>(c => c.TvDbEpisodeId == episode.Id);
|
|
|
|
//not found, try using season/episode number
|
|
if (episodeToUpdate == null)
|
|
{
|
|
episodeToUpdate = GetEpisode(series.SeriesId, episode.SeasonNumber, episode.EpisodeNumber);
|
|
}
|
|
|
|
//Episode doesn't exist locally
|
|
if (episodeToUpdate == null)
|
|
{
|
|
episodeToUpdate = new Episode();
|
|
newList.Add(episodeToUpdate);
|
|
}
|
|
else
|
|
{
|
|
updateList.Add(episodeToUpdate);
|
|
}
|
|
|
|
episodeToUpdate.SeriesId = series.SeriesId;
|
|
episodeToUpdate.TvDbEpisodeId = episode.Id;
|
|
episodeToUpdate.AirDate = episode.FirstAired.Date;
|
|
episodeToUpdate.EpisodeNumber = episode.EpisodeNumber;
|
|
episodeToUpdate.SeasonNumber = episode.SeasonNumber;
|
|
episodeToUpdate.Title = episode.EpisodeName;
|
|
episodeToUpdate.Overview = episode.Overview;
|
|
|
|
successCount++;
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
Logger.FatalException(
|
|
String.Format("An error has occurred while updating episode info for series {0}", series.SeriesId), e);
|
|
failCount++;
|
|
}
|
|
}
|
|
|
|
_repository.AddMany(newList);
|
|
_repository.UpdateMany(updateList);
|
|
|
|
Logger.Debug("Finished episode refresh for series:{0}. Successful:{1} - Failed:{2} ",
|
|
tvDbSeriesInfo.SeriesName, successCount, failCount);
|
|
}
|
|
|
|
public virtual void DeleteEpisode(int episodeId)
|
|
{
|
|
_repository.Delete<Episode>(episodeId);
|
|
}
|
|
|
|
public virtual void UpdateEpisode(Episode episode)
|
|
{
|
|
_repository.Update(episode);
|
|
}
|
|
|
|
public virtual bool IsIgnored(int seriesId, int seasonNumber)
|
|
{
|
|
return !_repository.Exists<Episode>(e => e.SeriesId == seriesId && e.SeasonNumber == seasonNumber && !e.Ignored);
|
|
}
|
|
|
|
public virtual IList<int> GetSeasons(int seriesId)
|
|
{
|
|
return _repository.All<Episode>().Where(e => e.SeriesId == seriesId).Select(s => s.SeasonNumber).Distinct().OrderBy(c => c).ToList();
|
|
}
|
|
|
|
public virtual void SetSeasonIgnore(long seriesId, int seasonNumber, bool isIgnored)
|
|
{
|
|
Logger.Info("Setting ignore flag on Series:{0} Season:{1} to {2}", seriesId, seasonNumber, isIgnored);
|
|
var episodes = GetEpisodesBySeason(seriesId, seasonNumber);
|
|
|
|
foreach (var episode in episodes)
|
|
{
|
|
episode.Ignored = isIgnored;
|
|
}
|
|
|
|
_repository.UpdateMany(episodes);
|
|
|
|
Logger.Info("Ignore flag for Series:{0} Season:{1} successfully set to {2}", seriesId, seasonNumber, isIgnored);
|
|
}
|
|
}
|
|
} |