Episode import uses specs and moves before import now

2026-04-21 22:04:31 -04:00 · 2013-07-06 14:47:49 -07:00
parent 9ed5a06504
commit aeb8ee06f6
22 changed files with 942 additions and 499 deletions
@@ -0,0 +1,94 @@
+using System;
+using System.Collections.Generic;
+using System.IO;
+using System.Linq;
+using System.Text;
+using NLog;
+using NzbDrone.Common;
+using NzbDrone.Common.Messaging;
+using NzbDrone.Core.MediaFiles.Events;
+
+namespace NzbDrone.Core.MediaFiles.EpisodeImport
+{
+    public interface IImportApprovedEpisodes
+    {
+        List<ImportDecision> Import(List<ImportDecision> decisions, bool newDownloads = false);
+    }
+    
+    public class ImportApprovedEpisodes : IImportApprovedEpisodes
+    {
+        private readonly IMoveEpisodeFiles _episodeFileMover;
+        private readonly MediaFileService _mediaFileService;
+        private readonly DiskProvider _diskProvider;
+        private readonly IMessageAggregator _messageAggregator;
+        private readonly Logger _logger;
+
+        public ImportApprovedEpisodes(IMoveEpisodeFiles episodeFileMover,
+                                      MediaFileService mediaFileService,
+                                      DiskProvider diskProvider,
+                                      IMessageAggregator messageAggregator,
+                                      Logger logger)
+        {
+            _episodeFileMover = episodeFileMover;
+            _mediaFileService = mediaFileService;
+            _diskProvider = diskProvider;
+            _messageAggregator = messageAggregator;
+            _logger = logger;
+        }
+
+        public List<ImportDecision> Import(List<ImportDecision> decisions, bool newDownload = false)
+        {
+            var qualifiedReports = GetQualifiedReports(decisions);
+            var imported = new List<ImportDecision>();
+
+            foreach (var report in qualifiedReports)
+            {
+                var localEpisode = report.LocalEpisode;
+
+                try
+                {
+                    if (imported.SelectMany(r => r.LocalEpisode.Episodes)
+                                         .Select(e => e.Id)
+                                         .ToList()
+                                         .Intersect(localEpisode.Episodes.Select(e => e.Id))
+                                         .Any())
+                    {
+                        continue;
+                    }
+
+                    var episodeFile = new EpisodeFile();
+                    episodeFile.DateAdded = DateTime.UtcNow;
+                    episodeFile.SeriesId = localEpisode.Series.Id;
+                    episodeFile.Path = localEpisode.Path.CleanPath();
+                    episodeFile.Size = _diskProvider.GetFileSize(localEpisode.Path);
+                    episodeFile.Quality = localEpisode.Quality;
+                    episodeFile.SeasonNumber = localEpisode.SeasonNumber;
+                    episodeFile.SceneName = Path.GetFileNameWithoutExtension(localEpisode.Path.CleanPath());
+                    episodeFile.Episodes = localEpisode.Episodes;
+
+                    if (newDownload)
+                    {
+                        episodeFile = _episodeFileMover.MoveEpisodeFile(episodeFile, localEpisode);
+                    }
+                    
+                    _mediaFileService.Add(episodeFile);
+                    _messageAggregator.PublishEvent(new EpisodeImportedEvent(episodeFile));
+
+                }
+                catch (Exception e)
+                {
+                    _logger.WarnException("Couldn't add report to download queue. " + localEpisode, e);
+                }
+            }
+
+            return imported;
+        }
+
+        private List<ImportDecision> GetQualifiedReports(List<ImportDecision> decisions)
+        {
+            return decisions.Where(c => c.Approved)
+                            .OrderByDescending(c => c.LocalEpisode.Quality)
+                            .ToList();
+        }
+    }
+}