mirror of https://github.com/Sonarr/Sonarr
Add TvDbEpisodeId to Episode. now we can index episodes before they showup in thetvdb
This commit is contained in:
parent
be74c67df8
commit
3beeff4e18
|
@ -1,15 +0,0 @@
|
||||||
using System;
|
|
||||||
|
|
||||||
namespace NzbDrone.Core.Model
|
|
||||||
{
|
|
||||||
public class NzbInfoModel
|
|
||||||
{
|
|
||||||
public string Title { get; set; }
|
|
||||||
public Uri Link { get; set; }
|
|
||||||
|
|
||||||
public bool IsPassworded()
|
|
||||||
{
|
|
||||||
return Title.EndsWith("(Passworded)", StringComparison.InvariantCultureIgnoreCase);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -180,7 +180,6 @@
|
||||||
<Compile Include="Model\EpisodeRenameModel.cs" />
|
<Compile Include="Model\EpisodeRenameModel.cs" />
|
||||||
<Compile Include="Model\EpisodeSortingType.cs" />
|
<Compile Include="Model\EpisodeSortingType.cs" />
|
||||||
<Compile Include="Model\EpisodeStatusType.cs" />
|
<Compile Include="Model\EpisodeStatusType.cs" />
|
||||||
<Compile Include="Model\NzbInfoModel.cs" />
|
|
||||||
<Compile Include="Model\SabnzbdPriorityType.cs" />
|
<Compile Include="Model\SabnzbdPriorityType.cs" />
|
||||||
<Compile Include="Model\SceneNameModel.cs" />
|
<Compile Include="Model\SceneNameModel.cs" />
|
||||||
<Compile Include="Model\SeasonParseResult.cs" />
|
<Compile Include="Model\SeasonParseResult.cs" />
|
||||||
|
|
|
@ -178,7 +178,7 @@ namespace NzbDrone.Core.Providers
|
||||||
var newEpisode = new Episode
|
var newEpisode = new Episode
|
||||||
{
|
{
|
||||||
AirDate = episode.FirstAired,
|
AirDate = episode.FirstAired,
|
||||||
EpisodeId = episode.Id,
|
TvDbEpisodeId = episode.Id,
|
||||||
EpisodeNumber = episode.EpisodeNumber,
|
EpisodeNumber = episode.EpisodeNumber,
|
||||||
Language = episode.Language.Abbriviation,
|
Language = episode.Language.Abbriviation,
|
||||||
Overview = episode.Overview,
|
Overview = episode.Overview,
|
||||||
|
@ -188,8 +188,11 @@ namespace NzbDrone.Core.Providers
|
||||||
Title = episode.EpisodeName
|
Title = episode.EpisodeName
|
||||||
};
|
};
|
||||||
|
|
||||||
if (_sonicRepo.Exists<Episode>(e => e.EpisodeId == newEpisode.EpisodeId))
|
var existingEpisode = GetEpisode(episode.SeriesId, episode.SeasonNumber, episode.EpisodeNumber);
|
||||||
|
|
||||||
|
if (existingEpisode != null)
|
||||||
{
|
{
|
||||||
|
newEpisode.EpisodeId = existingEpisode.EpisodeId;
|
||||||
updateList.Add(newEpisode);
|
updateList.Add(newEpisode);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
|
@ -214,72 +217,6 @@ namespace NzbDrone.Core.Providers
|
||||||
targetSeries.SeriesName, successCount, failCount);
|
targetSeries.SeriesName, successCount, failCount);
|
||||||
}
|
}
|
||||||
|
|
||||||
public virtual void RefreshEpisodeInfo(Season season)
|
|
||||||
{
|
|
||||||
Logger.Info("Starting episode info refresh for season {0} of series:{1}", season.SeasonNumber,
|
|
||||||
season.SeriesId);
|
|
||||||
int successCount = 0;
|
|
||||||
int failCount = 0;
|
|
||||||
var targetSeries = _tvDb.GetSeries(season.SeriesId, true);
|
|
||||||
|
|
||||||
var updateList = new List<Episode>();
|
|
||||||
var newList = new List<Episode>();
|
|
||||||
|
|
||||||
foreach (var episode in targetSeries.Episodes.Where(e => e.SeasonId == season.SeasonId))
|
|
||||||
{
|
|
||||||
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 series:{0} - episode:{1}", targetSeries.SeriesName,
|
|
||||||
episode.EpisodeNumber);
|
|
||||||
var newEpisode = new Episode
|
|
||||||
{
|
|
||||||
AirDate = episode.FirstAired,
|
|
||||||
EpisodeId = episode.Id,
|
|
||||||
EpisodeNumber = episode.EpisodeNumber,
|
|
||||||
Language = episode.Language.Abbriviation,
|
|
||||||
Overview = episode.Overview,
|
|
||||||
SeasonId = episode.SeasonId,
|
|
||||||
SeasonNumber = episode.SeasonNumber,
|
|
||||||
SeriesId = season.SeriesId,
|
|
||||||
Title = episode.EpisodeName
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
//TODO: Replace this db check with a local check. Should make things even faster
|
|
||||||
if (_sonicRepo.Exists<Episode>(e => e.EpisodeId == newEpisode.EpisodeId))
|
|
||||||
{
|
|
||||||
updateList.Add(newEpisode);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
newList.Add(newEpisode);
|
|
||||||
}
|
|
||||||
|
|
||||||
successCount++;
|
|
||||||
}
|
|
||||||
catch (Exception e)
|
|
||||||
{
|
|
||||||
Logger.FatalException(
|
|
||||||
String.Format("An error has occurred while updating episode info for season {0} of series {1}",
|
|
||||||
season.SeasonNumber, season.SeriesId), e);
|
|
||||||
failCount++;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
_sonicRepo.AddMany(newList);
|
|
||||||
_sonicRepo.UpdateMany(updateList);
|
|
||||||
|
|
||||||
Logger.Debug("Finished episode refresh for series:{0}. Successful:{1} - Failed:{2} ",
|
|
||||||
targetSeries.SeriesName, successCount, failCount);
|
|
||||||
}
|
|
||||||
|
|
||||||
public virtual void DeleteEpisode(int episodeId)
|
public virtual void DeleteEpisode(int episodeId)
|
||||||
{
|
{
|
||||||
_sonicRepo.Delete<Episode>(episodeId);
|
_sonicRepo.Delete<Episode>(episodeId);
|
||||||
|
|
|
@ -7,9 +7,11 @@ namespace NzbDrone.Core.Repository
|
||||||
{
|
{
|
||||||
public class Episode
|
public class Episode
|
||||||
{
|
{
|
||||||
[SubSonicPrimaryKey(false)]
|
[SubSonicPrimaryKey]
|
||||||
public virtual int EpisodeId { get; set; }
|
public virtual int EpisodeId { get; set; }
|
||||||
|
|
||||||
|
public int? TvDbEpisodeId { get; set; }
|
||||||
|
|
||||||
public virtual int SeriesId { get; set; }
|
public virtual int SeriesId { get; set; }
|
||||||
public virtual int EpisodeFileId { get; set; }
|
public virtual int EpisodeFileId { get; set; }
|
||||||
public virtual int SeasonId { get; set; }
|
public virtual int SeasonId { get; set; }
|
||||||
|
|
Loading…
Reference in New Issue