New: Update AutoTags on series update

Closes #6783
This commit is contained in:
Mark McDowall 2024-07-06 18:54:51 -07:00
parent 67943edfbc
commit 4cb38d4896
2 changed files with 46 additions and 25 deletions

View File

@ -198,34 +198,11 @@ namespace NzbDrone.Core.Tv
private void UpdateTags(Series series) private void UpdateTags(Series series)
{ {
_logger.Trace("Updating tags for {0}", series); var tagsUpdated = _seriesService.UpdateTags(series);
var tagsAdded = new HashSet<int>(); if (tagsUpdated)
var tagsRemoved = new HashSet<int>();
var changes = _autoTaggingService.GetTagChanges(series);
foreach (var tag in changes.TagsToRemove)
{
if (series.Tags.Contains(tag))
{
series.Tags.Remove(tag);
tagsRemoved.Add(tag);
}
}
foreach (var tag in changes.TagsToAdd)
{
if (!series.Tags.Contains(tag))
{
series.Tags.Add(tag);
tagsAdded.Add(tag);
}
}
if (tagsAdded.Any() || tagsRemoved.Any())
{ {
_seriesService.UpdateSeries(series); _seriesService.UpdateSeries(series);
_logger.Debug("Updated tags for '{0}'. Added: {1}, Removed: {2}", series.Title, tagsAdded.Count, tagsRemoved.Count);
} }
} }

View File

@ -2,6 +2,7 @@ using System.Collections.Generic;
using System.Linq; using System.Linq;
using NLog; using NLog;
using NzbDrone.Common.Extensions; using NzbDrone.Common.Extensions;
using NzbDrone.Core.AutoTagging;
using NzbDrone.Core.Messaging.Events; using NzbDrone.Core.Messaging.Events;
using NzbDrone.Core.Parser; using NzbDrone.Core.Parser;
using NzbDrone.Core.Tv.Events; using NzbDrone.Core.Tv.Events;
@ -30,6 +31,7 @@ namespace NzbDrone.Core.Tv
List<Series> UpdateSeries(List<Series> series, bool useExistingRelativeFolder); List<Series> UpdateSeries(List<Series> series, bool useExistingRelativeFolder);
bool SeriesPathExists(string folder); bool SeriesPathExists(string folder);
void RemoveAddOptions(Series series); void RemoveAddOptions(Series series);
bool UpdateTags(Series series);
} }
public class SeriesService : ISeriesService public class SeriesService : ISeriesService
@ -38,18 +40,21 @@ namespace NzbDrone.Core.Tv
private readonly IEventAggregator _eventAggregator; private readonly IEventAggregator _eventAggregator;
private readonly IEpisodeService _episodeService; private readonly IEpisodeService _episodeService;
private readonly IBuildSeriesPaths _seriesPathBuilder; private readonly IBuildSeriesPaths _seriesPathBuilder;
private readonly IAutoTaggingService _autoTaggingService;
private readonly Logger _logger; private readonly Logger _logger;
public SeriesService(ISeriesRepository seriesRepository, public SeriesService(ISeriesRepository seriesRepository,
IEventAggregator eventAggregator, IEventAggregator eventAggregator,
IEpisodeService episodeService, IEpisodeService episodeService,
IBuildSeriesPaths seriesPathBuilder, IBuildSeriesPaths seriesPathBuilder,
IAutoTaggingService autoTaggingService,
Logger logger) Logger logger)
{ {
_seriesRepository = seriesRepository; _seriesRepository = seriesRepository;
_eventAggregator = eventAggregator; _eventAggregator = eventAggregator;
_episodeService = episodeService; _episodeService = episodeService;
_seriesPathBuilder = seriesPathBuilder; _seriesPathBuilder = seriesPathBuilder;
_autoTaggingService = autoTaggingService;
_logger = logger; _logger = logger;
} }
@ -205,6 +210,7 @@ namespace NzbDrone.Core.Tv
// Never update AddOptions when updating a series, keep it the same as the existing stored series. // Never update AddOptions when updating a series, keep it the same as the existing stored series.
series.AddOptions = storedSeries.AddOptions; series.AddOptions = storedSeries.AddOptions;
UpdateTags(series);
var updatedSeries = _seriesRepository.Update(series); var updatedSeries = _seriesRepository.Update(series);
if (publishUpdatedEvent) if (publishUpdatedEvent)
@ -233,6 +239,8 @@ namespace NzbDrone.Core.Tv
{ {
_logger.Trace("Not changing path for: {0}", s.Title); _logger.Trace("Not changing path for: {0}", s.Title);
} }
UpdateTags(s);
} }
_seriesRepository.UpdateMany(series); _seriesRepository.UpdateMany(series);
@ -250,5 +258,41 @@ namespace NzbDrone.Core.Tv
{ {
_seriesRepository.SetFields(series, s => s.AddOptions); _seriesRepository.SetFields(series, s => s.AddOptions);
} }
public bool UpdateTags(Series series)
{
_logger.Trace("Updating tags for {0}", series);
var tagsAdded = new HashSet<int>();
var tagsRemoved = new HashSet<int>();
var changes = _autoTaggingService.GetTagChanges(series);
foreach (var tag in changes.TagsToRemove)
{
if (series.Tags.Contains(tag))
{
series.Tags.Remove(tag);
tagsRemoved.Add(tag);
}
}
foreach (var tag in changes.TagsToAdd)
{
if (!series.Tags.Contains(tag))
{
series.Tags.Add(tag);
tagsAdded.Add(tag);
}
}
if (tagsAdded.Any() || tagsRemoved.Any())
{
_logger.Debug("Updated tags for '{0}'. Added: {1}, Removed: {2}", series.Title, tagsAdded.Count, tagsRemoved.Count);
return true;
}
return false;
}
} }
} }