New: Rewrite of download decision engine.

2026-04-20 21:54:25 -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,37 @@
+using System.Linq;
+using NLog;
+using Ninject;
+using NzbDrone.Core.Model;
+
+namespace NzbDrone.Core.Providers.DecisionEngine
+{
+    public class UpgradeDiskSpecification
+    {
+        private readonly EpisodeProvider _episodeProvider;
+        private readonly QualityUpgradeSpecification _qualityUpgradeSpecification;
+        private static readonly Logger logger = LogManager.GetCurrentClassLogger();
+
+        [Inject]
+        public UpgradeDiskSpecification(EpisodeProvider episodeProvider, QualityUpgradeSpecification qualityUpgradeSpecification)
+        {
+            _episodeProvider = episodeProvider;
+            _qualityUpgradeSpecification = qualityUpgradeSpecification;
+        }
+
+        public UpgradeDiskSpecification()
+        {
+        }
+
+        public virtual bool IsSatisfiedBy(EpisodeParseResult subject)
+        {
+            foreach (var file in _episodeProvider.GetEpisodesByParseResult(subject).Select(c => c.EpisodeFile).Where(c => c != null))
+            {
+                logger.Trace("Comparing file quality with report. Existing file is {0} proper:{1}", file.Quality, file.Proper);
+                if (!_qualityUpgradeSpecification.IsSatisfiedBy(new Quality { QualityType = file.Quality, Proper = file.Proper }, subject.Quality, subject.Series.QualityProfile.Cutoff))
+                    return false;
+            }
+
+            return true;
+        }
+    }
+}