New: Rewrite of download decision engine.

2026-04-20 22:14:34 -04:00 · 2012-02-06 21:08:07 -08:00
parent a168bdfa00
commit 5717b7f596
60 changed files with 2013 additions and 1745 deletions
@@ -0,0 +1,44 @@
+using System.Linq;
+using NLog;
+using Ninject;
+using NzbDrone.Core.Model;
+
+namespace NzbDrone.Core.Providers.DecisionEngine
+{
+    public class UpgradeHistorySpecification
+    {
+        private readonly EpisodeProvider _episodeProvider;
+        private readonly HistoryProvider _historyProvider;
+        private readonly QualityUpgradeSpecification _qualityUpgradeSpecification;
+        private static readonly Logger logger = LogManager.GetCurrentClassLogger();
+
+        [Inject]
+        public UpgradeHistorySpecification(EpisodeProvider episodeProvider, HistoryProvider historyProvider, QualityUpgradeSpecification qualityUpgradeSpecification)
+        {
+            _episodeProvider = episodeProvider;
+            _historyProvider = historyProvider;
+            _qualityUpgradeSpecification = qualityUpgradeSpecification;
+        }
+
+        public UpgradeHistorySpecification()
+        {
+
+        }
+
+        public virtual bool IsSatisfiedBy(EpisodeParseResult subject)
+        {
+            foreach (var episode in _episodeProvider.GetEpisodesByParseResult(subject))
+            {
+                var bestQualityInHistory = _historyProvider.GetBestQualityInHistory(subject.Series.SeriesId, episode.SeasonNumber, episode.EpisodeNumber);
+                if (bestQualityInHistory != null)
+                {
+                    logger.Trace("Comparing history quality with report. History is {0}", bestQualityInHistory);
+                    if (!_qualityUpgradeSpecification.IsSatisfiedBy(bestQualityInHistory, subject.Quality, subject.Series.QualityProfile.Cutoff))
+                        return false;
+                }
+            }
+
+            return true;
+        }
+    }
+}