self-hosted/Sonarr
1
0
Fork 0
mirror of https://github.com/Sonarr/Sonarr.git synced 2024-12-16 11:37:58 +02:00
Code Issues Releases Activity

New: Prefer regular episodes over specials when absolute numbers conflict

Browse Source 
Closes #676
This commit is contained in:
Mark McDowall 2016-02-11 00:09:29 -08:00
parent 2396af4589
commit 1603b06431
2 changed files with 35 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/NzbDrone.Core.Test/TvTests/RefreshEpisodeServiceFixture.cs
View File

@ -365,5 +365,33 @@ public void should_not_update_air_date_when_multiple_episodes_air_on_the_same_da
_insertedEpisodes.Should().OnlyContain(e => e.AirDateUtc.Value.ToString("s") == episodes.First().AirDateUtc.Value.ToString("s")); _insertedEpisodes.Should().OnlyContain(e => e.AirDateUtc.Value.ToString("s") == episodes.First().AirDateUtc.Value.ToString("s"));
} }
[Test]
public void should_prefer_regular_season_when_absolute_numbers_conflict()
{
var episodes = Builder<Episode>.CreateListOfSize(2)
.Build()
.ToList();
episodes[0].AbsoluteEpisodeNumber = episodes[1].AbsoluteEpisodeNumber;
episodes[0].SeasonNumber = 0;
episodes[0].EpisodeNumber.Should().NotBe(episodes[1].EpisodeNumber);
var existingEpisode = new Episode
{
SeasonNumber = episodes[0].SeasonNumber,
EpisodeNumber = episodes[0].EpisodeNumber,
AbsoluteEpisodeNumber = episodes[1].AbsoluteEpisodeNumber
};
Mocker.GetMock<IEpisodeService>().Setup(c => c.GetEpisodeBySeries(It.IsAny<int>()))
.Returns(new List<Episode> { existingEpisode });
Subject.RefreshEpisodeInfo(GetAnimeSeries(), episodes);
_updatedEpisodes.First().SeasonNumber.Should().Be(episodes[1].SeasonNumber);
_updatedEpisodes.First().EpisodeNumber.Should().Be(episodes[1].EpisodeNumber);
_updatedEpisodes.First().AbsoluteEpisodeNumber.Should().Be(episodes[1].AbsoluteEpisodeNumber);
}
} }
} }

13
src/NzbDrone.Core/Tv/RefreshEpisodeService.cs
View File

@ -41,7 +41,7 @@ public void RefreshEpisodeInfo(Series series, IEnumerable<Episode> remoteEpisode
if (series.SeriesType == SeriesTypes.Anime) if (series.SeriesType == SeriesTypes.Anime)
{ {
dupeFreeRemoteEpisodes = MapAbsoluteEpisodeNumbers(series, dupeFreeRemoteEpisodes); dupeFreeRemoteEpisodes = MapAbsoluteEpisodeNumbers(dupeFreeRemoteEpisodes);
} }
foreach (var episode in OrderEpisodes(series, dupeFreeRemoteEpisodes)) foreach (var episode in OrderEpisodes(series, dupeFreeRemoteEpisodes))
@ -154,13 +154,14 @@ private void AdjustDirectToDvdAirDate(Series series, IEnumerable<Episode> allEpi
} }
} }
private List<Episode> MapAbsoluteEpisodeNumbers(Series series, List<Episode> traktEpisodes) private List<Episode> MapAbsoluteEpisodeNumbers(List<Episode> remoteEpisodes)
{ {
//Return all episodes with no abs number, but distinct for those with abs number //Return all episodes with no abs number, but distinct for those with abs number
return traktEpisodes.Where(e => e.AbsoluteEpisodeNumber.HasValue) return remoteEpisodes.Where(e => e.AbsoluteEpisodeNumber.HasValue)
.DistinctBy(e => e.AbsoluteEpisodeNumber.Value) .OrderByDescending(e => e.SeasonNumber)
.Concat(traktEpisodes.Where(e => !e.AbsoluteEpisodeNumber.HasValue)) .DistinctBy(e => e.AbsoluteEpisodeNumber.Value)
.ToList(); .Concat(remoteEpisodes.Where(e => !e.AbsoluteEpisodeNumber.HasValue))
.ToList();
} }
private Episode GetEpisodeToUpdate(Series series, Episode episode, List<Episode> existingEpisodes) private Episode GetEpisodeToUpdate(Series series, Episode episode, List<Episode> existingEpisodes)