1
0
mirror of https://github.com/Radarr/Radarr.git synced 2024-09-19 07:52:33 +02:00

Fixed: Improved title sorting for 'A to Z'

This commit is contained in:
Mark McDowall 2014-11-05 08:45:50 -08:00
parent 3411c4e5a7
commit 97c9217c0d
10 changed files with 60 additions and 7 deletions

View File

@ -85,7 +85,7 @@ private void ValidateSeries(Series series)
series.Should().NotBeNull(); series.Should().NotBeNull();
series.Title.Should().NotBeNullOrWhiteSpace(); series.Title.Should().NotBeNullOrWhiteSpace();
series.CleanTitle.Should().Be(Parser.Parser.CleanSeriesTitle(series.Title)); series.CleanTitle.Should().Be(Parser.Parser.CleanSeriesTitle(series.Title));
series.SortTitle.Should().Be(Parser.Parser.NormalizeEpisodeTitle(series.Title)); series.SortTitle.Should().Be(SeriesTitleNormalizer.Normalize(series.Title, series.TvdbId));
series.Overview.Should().NotBeNullOrWhiteSpace(); series.Overview.Should().NotBeNullOrWhiteSpace();
series.AirTime.Should().NotBeNullOrWhiteSpace(); series.AirTime.Should().NotBeNullOrWhiteSpace();
series.FirstAired.Should().HaveValue(); series.FirstAired.Should().HaveValue();

View File

@ -292,6 +292,7 @@
<Compile Include="TvTests\SeriesServiceTests\AddSeriesFixture.cs" /> <Compile Include="TvTests\SeriesServiceTests\AddSeriesFixture.cs" />
<Compile Include="TvTests\SeriesServiceTests\UpdateMultipleSeriesFixture.cs" /> <Compile Include="TvTests\SeriesServiceTests\UpdateMultipleSeriesFixture.cs" />
<Compile Include="TvTests\SeriesServiceTests\UpdateSeriesFixture.cs" /> <Compile Include="TvTests\SeriesServiceTests\UpdateSeriesFixture.cs" />
<Compile Include="TvTests\SeriesTitleNormalizerFixture.cs" />
<Compile Include="TvTests\ShouldRefreshSeriesFixture.cs" /> <Compile Include="TvTests\ShouldRefreshSeriesFixture.cs" />
<Compile Include="UpdateTests\UpdatePackageProviderFixture.cs" /> <Compile Include="UpdateTests\UpdatePackageProviderFixture.cs" />
<Compile Include="UpdateTests\UpdateServiceFixture.cs" /> <Compile Include="UpdateTests\UpdateServiceFixture.cs" />

View File

@ -0,0 +1,28 @@
using System;
using FluentAssertions;
using NUnit.Framework;
using NzbDrone.Core.Tv;
namespace NzbDrone.Core.Test.TvTests
{
[TestFixture]
public class SeriesTitleNormalizerFixture
{
[Test]
public void should_use_precomputed_title_for_a_to_z()
{
SeriesTitleNormalizer.Normalize("A to Z", 281588).Should().Be("a to z");
}
[TestCase("2 Broke Girls", "2 broke girls")]
[TestCase("Archer (2009)", "archer 2009")]
[TestCase("The Office (US)", "office us")]
[TestCase("The Mentalist", "mentalist")]
[TestCase("The Good Wife", "good wife")]
[TestCase("The Newsroom (2012)", "newsroom 2012")]
public void should_normalize_title(String title, String expected)
{
SeriesTitleNormalizer.Normalize(title, 0).Should().Be(expected);
}
}
}

View File

@ -27,7 +27,7 @@ private void SetSortTitles(IDbConnection conn, IDbTransaction tran)
var id = seriesReader.GetInt32(0); var id = seriesReader.GetInt32(0);
var title = seriesReader.GetString(1); var title = seriesReader.GetString(1);
var sortTitle = Parser.Parser.NormalizeEpisodeTitle(title).ToLower(); var sortTitle = Parser.Parser.NormalizeTitle(title).ToLower();
using (IDbCommand updateCmd = conn.CreateCommand()) using (IDbCommand updateCmd = conn.CreateCommand())
{ {

View File

@ -134,7 +134,7 @@ private static Series MapSeries(Show show)
series.ImdbId = show.imdb_id; series.ImdbId = show.imdb_id;
series.Title = show.title; series.Title = show.title;
series.CleanTitle = Parser.Parser.CleanSeriesTitle(show.title); series.CleanTitle = Parser.Parser.CleanSeriesTitle(show.title);
series.SortTitle = Parser.Parser.NormalizeEpisodeTitle(show.title).ToLower(); series.SortTitle = SeriesTitleNormalizer.Normalize(show.title, show.tvdb_id);
series.Year = GetYear(show.year, show.first_aired); series.Year = GetYear(show.year, show.first_aired);
series.FirstAired = FromIso(show.first_aired_iso); series.FirstAired = FromIso(show.first_aired_iso);
series.Overview = show.overview; series.Overview = show.overview;

View File

@ -765,6 +765,7 @@
<SubType>Code</SubType> <SubType>Code</SubType>
</Compile> </Compile>
<Compile Include="Tv\SeriesStatusType.cs" /> <Compile Include="Tv\SeriesStatusType.cs" />
<Compile Include="Tv\SeriesTitleNormalizer.cs" />
<Compile Include="Tv\SeriesTypes.cs" /> <Compile Include="Tv\SeriesTypes.cs" />
<Compile Include="Tv\ShouldRefreshSeries.cs" /> <Compile Include="Tv\ShouldRefreshSeries.cs" />
<Compile Include="Update\Commands\ApplicationUpdateCommand.cs" /> <Compile Include="Update\Commands\ApplicationUpdateCommand.cs" />

View File

@ -323,7 +323,7 @@ public static string CleanupEpisodeTitle(string title)
return MultiPartCleanupRegex.Replace(title, string.Empty).Trim(); return MultiPartCleanupRegex.Replace(title, string.Empty).Trim();
} }
public static string NormalizeEpisodeTitle(string title) public static string NormalizeTitle(string title)
{ {
string singleSpaces = WordDelimiterRegex.Replace(title, " "); string singleSpaces = WordDelimiterRegex.Replace(title, " ");
string noPunctuation = PunctuationRegex.Replace(singleSpaces, String.Empty); string noPunctuation = PunctuationRegex.Replace(singleSpaces, String.Empty);

View File

@ -106,12 +106,12 @@ public List<Episode> GetEpisodesBySeason(int seriesId, int seasonNumber)
public Episode FindEpisodeByName(int seriesId, int seasonNumber, string episodeTitle) public Episode FindEpisodeByName(int seriesId, int seasonNumber, string episodeTitle)
{ {
// TODO: can replace this search mechanism with something smarter/faster/better // TODO: can replace this search mechanism with something smarter/faster/better
var search = Parser.Parser.NormalizeEpisodeTitle(episodeTitle); var search = Parser.Parser.NormalizeTitle(episodeTitle);
return _episodeRepository.GetEpisodes(seriesId, seasonNumber) return _episodeRepository.GetEpisodes(seriesId, seasonNumber)
.FirstOrDefault(e => .FirstOrDefault(e =>
{ {
// normalize episode title // normalize episode title
string title = Parser.Parser.NormalizeEpisodeTitle(e.Title); string title = Parser.Parser.NormalizeTitle(e.Title);
// find episode title within search string // find episode title within search string
return (title.Length > 0) && search.Contains(title); return (title.Length > 0) && search.Contains(title);
}); });

View File

@ -77,7 +77,7 @@ public Series AddSeries(Series newSeries)
newSeries.Monitored = true; newSeries.Monitored = true;
newSeries.CleanTitle = newSeries.Title.CleanSeriesTitle(); newSeries.CleanTitle = newSeries.Title.CleanSeriesTitle();
newSeries.SortTitle = Parser.Parser.NormalizeEpisodeTitle(newSeries.Title).ToLower(); newSeries.SortTitle = SeriesTitleNormalizer.Normalize(newSeries.Title, newSeries.TvdbId);
_seriesRepository.Insert(newSeries); _seriesRepository.Insert(newSeries);
_eventAggregator.PublishEvent(new SeriesAddedEvent(GetSeries(newSeries.Id))); _eventAggregator.PublishEvent(new SeriesAddedEvent(GetSeries(newSeries.Id)));

View File

@ -0,0 +1,23 @@
using System;
using System.Collections.Generic;
namespace NzbDrone.Core.Tv
{
public static class SeriesTitleNormalizer
{
private readonly static Dictionary<Int32, String> PreComputedTitles = new Dictionary<Int32, String>
{
{ 281588, "a to z" }
};
public static String Normalize(String title, Int32 tvdbId)
{
if (PreComputedTitles.ContainsKey(tvdbId))
{
return PreComputedTitles[tvdbId];
}
return Parser.Parser.NormalizeTitle(title).ToLower();
}
}
}