shpaq/Prowlarr-Prowlarr
1
0
Fork 0
mirror of https://github.com/Prowlarr/Prowlarr.git synced 2025-09-17 17:14:18 +02:00
Code Issues Packages Projects Releases Wiki Activity

reset existing xem info during update

Browse Source
This commit is contained in:
kayone
2013-11-30 17:45:30 -08:00
parent b43c4e04f8
commit 575dc03e35
6 changed files with 29 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

74
src/NzbDrone.Core/DataAugmentation/Xem/XemService.cs
View File

@@ -1,17 +1,14 @@
using System;
using System.Collections.Generic;
using System.Linq;
using NLog;
using NzbDrone.Common.Cache;
using NzbDrone.Core.Lifecycle;
using NzbDrone.Core.Messaging.Commands;
using NzbDrone.Core.Messaging.Events;
using NzbDrone.Core.Tv;
using NzbDrone.Core.Tv.Events;
namespace NzbDrone.Core.DataAugmentation.Xem
{
public class XemService : IExecute<UpdateXemMappingsCommand>, IHandle<SeriesUpdatedEvent>, IHandleAsync<ApplicationStartedEvent>
public class XemService : IHandle<SeriesUpdatedEvent>
{
private readonly IEpisodeService _episodeService;
private readonly IXemProxy _xemProxy;
@@ -33,63 +30,28 @@ namespace NzbDrone.Core.DataAugmentation.Xem
}
public void Execute(UpdateXemMappingsCommand message)
{
UpdateMappings();
}
public void Handle(SeriesUpdatedEvent message)
{
UpdateMappings(message.Series);
}
public void HandleAsync(ApplicationStartedEvent message)
{
GetXemSeriesIds();
}
private void UpdateMappings()
{
_logger.Trace("Starting scene numbering update");
try
if (_cache.Count == 0)
{
var ids = GetXemSeriesIds();
var series = _seriesService.GetAllSeries();
var wantedSeries = series.Where(s => ids.Contains(s.TvdbId)).ToList();
foreach (var ser in wantedSeries)
{
PerformUpdate(ser);
}
_logger.Trace("Completed scene numbering update");
RefreshCache();
}
catch (Exception ex)
if (!_cache.Find(message.Series.TvdbId.ToString()))
{
_logger.WarnException("Error updating Scene Mappings", ex);
throw;
}
}
private void UpdateMappings(Series series)
{
if (!_cache.Find(series.TvdbId.ToString()))
{
_logger.Trace("Scene numbering is not available for {0} [{1}]", series.Title, series.TvdbId);
_logger.Trace("Scene numbering is not available for {0} [{1}]", message.Series.Title, message.Series.TvdbId);
return;
}
PerformUpdate(series);
PerformUpdate(message.Series);
}
private void PerformUpdate(Series series)
{
_logger.Trace("Updating scene numbering mapping for: {0}", series);
try
{
var episodesToUpdate = new List<Episode>();
var mappings = _xemProxy.GetSceneTvdbMappings(series.TvdbId);
if (!mappings.Any())
@@ -101,6 +63,13 @@ namespace NzbDrone.Core.DataAugmentation.Xem
var episodes = _episodeService.GetEpisodeBySeries(series.Id);
foreach (var episode in episodes)
{
episode.AbsoluteEpisodeNumber = 0;
episode.SceneSeasonNumber = 0;
episode.SceneEpisodeNumber = 0;
}
foreach (var mapping in mappings)
{
_logger.Trace("Setting scene numbering mappings for {0} S{1:00}E{2:00}", series, mapping.Tvdb.Season, mapping.Tvdb.Episode);
@@ -116,24 +85,21 @@ namespace NzbDrone.Core.DataAugmentation.Xem
episode.AbsoluteEpisodeNumber = mapping.Scene.Absolute;
episode.SceneSeasonNumber = mapping.Scene.Season;
episode.SceneEpisodeNumber = mapping.Scene.Episode;
episodesToUpdate.Add(episode);
}
_logger.Trace("Committing scene numbering mappings to database for: {0}", series);
_episodeService.UpdateEpisodes(episodesToUpdate);
_logger.Trace("Setting UseSceneMapping for {0}", series);
_episodeService.UpdateEpisodes(episodes);
series.UseSceneNumbering = true;
_seriesService.UpdateSeries(series);
}
_logger.Debug("XEM mapping updated for {0}", series);
}
catch (Exception ex)
{
_logger.ErrorException("Error updating scene numbering mappings for: " + series, ex);
}
}
private List<int> GetXemSeriesIds()
private void RefreshCache()
{
_cache.Clear();
@@ -141,10 +107,8 @@ namespace NzbDrone.Core.DataAugmentation.Xem
foreach (var id in ids)
{
_cache.Set(id.ToString(), true);
_cache.Set(id.ToString(), true, TimeSpan.FromHours(1));
}
return ids;
}
}
}