Profiles

Indexes are created with the same uniqueness when copying a table New: Non-English episode support New: Renamed Quality Profiles to Profiles and made them more powerful New: Configurable wait time before grabbing a release to wait for a better quality
2026-04-25 22:37:27 -04:00 · 2014-06-08 01:22:55 -07:00
parent b72678a9ad
commit 74a38415cf
182 changed files with 2493 additions and 2433 deletions
@@ -0,0 +1,106 @@
+using System.Collections.Generic;
+using System.Linq;
+using NLog;
+using NzbDrone.Core.Lifecycle;
+using NzbDrone.Core.Messaging.Events;
+using NzbDrone.Core.Parser;
+using NzbDrone.Core.Qualities;
+using NzbDrone.Core.Tv;
+
+namespace NzbDrone.Core.Profiles
+{
+    public interface IProfileService
+    {
+        Profile Add(Profile profile);
+        void Update(Profile profile);
+        void Delete(int id);
+        List<Profile> All();
+        Profile Get(int id);
+    }
+
+    public class ProfileService : IProfileService, IHandle<ApplicationStartedEvent>
+    {
+        private readonly IProfileRepository _profileRepository;
+        private readonly ISeriesService _seriesService;
+        private readonly Logger _logger;
+
+        public ProfileService(IProfileRepository profileRepository, ISeriesService seriesService, Logger logger)
+        {
+            _profileRepository = profileRepository;
+            _seriesService = seriesService;
+            _logger = logger;
+        }
+
+        public Profile Add(Profile profile)
+        {
+            return _profileRepository.Insert(profile);
+        }
+
+        public void Update(Profile profile)
+        {
+            _profileRepository.Update(profile);
+        }
+
+        public void Delete(int id)
+        {
+            if (_seriesService.GetAllSeries().Any(c => c.ProfileId == id))
+            {
+                throw new ProfileInUseException(id);
+            }
+
+            _profileRepository.Delete(id);
+        }
+
+        public List<Profile> All()
+        {
+            return _profileRepository.All().ToList();
+        }
+
+        public Profile Get(int id)
+        {
+            return _profileRepository.Get(id);
+        }
+
+        private Profile AddDefaultProfile(string name, Quality cutoff, params Quality[] allowed)
+        {
+            var items = Quality.DefaultQualityDefinitions
+                            .OrderBy(v => v.Weight)
+                            .Select(v => new ProfileQualityItem { Quality = v.Quality, Allowed = allowed.Contains(v.Quality) })
+                            .ToList();
+
+            var profile = new Profile { Name = name, Cutoff = cutoff, Items = items, Language = Language.English };
+
+            return Add(profile);
+        }
+
+        public void Handle(ApplicationStartedEvent message)
+        {
+            if (All().Any()) return;
+
+            _logger.Info("Setting up default quality profiles");
+
+            AddDefaultProfile("SD", Quality.SDTV,
+                Quality.SDTV,
+                Quality.WEBDL480p,
+                Quality.DVD);
+
+            AddDefaultProfile("HD-720p", Quality.HDTV720p,
+                Quality.HDTV720p,
+                Quality.WEBDL720p,
+                Quality.Bluray720p);
+
+            AddDefaultProfile("HD-1080p", Quality.HDTV1080p,
+                Quality.HDTV1080p,
+                Quality.WEBDL1080p,
+                Quality.Bluray1080p);
+
+            AddDefaultProfile("HD - All", Quality.HDTV720p,
+                Quality.HDTV720p,
+                Quality.HDTV1080p,
+                Quality.WEBDL720p,
+                Quality.WEBDL1080p,
+                Quality.Bluray720p,
+                Quality.Bluray1080p);
+        }
+    }
+}