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

Fix netimport search and add NetImportSyncCommand

Browse Source
This commit is contained in:
Leonardo Galli
2017-01-23 14:04:01 +01:00
parent 2b7afd3272
commit 93d6505f85
6 changed files with 114 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
src/NzbDrone.Core/NetImport/NetImportSearchService.cs
View File

@@ -3,6 +3,8 @@ using System.Collections.Generic;
using System.Linq;
using System.Text;
using NLog;
using NzbDrone.Core.Messaging.Commands;
using NzbDrone.Core.Messaging.Events;
using NzbDrone.Core.Tv;
namespace NzbDrone.Core.NetImport
@@ -13,7 +15,7 @@ namespace NzbDrone.Core.NetImport
List<Movie> FetchAndFilter(int listId, bool onlyEnableAuto);
}
public class NetImportSearchService : IFetchNetImport
public class NetImportSearchService : IFetchNetImport, IExecute<NetImportSyncCommand>
{
private readonly Logger _logger;
private readonly INetImportFactory _netImportFactory;
@@ -26,39 +28,29 @@ namespace NzbDrone.Core.NetImport
_logger = logger;
}
public List<Movie> Fetch(int listId, bool onlyEnableAuto)
public List<Movie> Fetch(int listId, bool onlyEnableAuto = false)
{
return MovieListSearch(listId, onlyEnableAuto);
}
public List<Movie> FetchAndFilter(int listId, bool onlyEnableAuto)
{
var existingMovies = _movieService.GetAllMovies();
var movies = MovieListSearch(listId, onlyEnableAuto);
// remove from movies list where existMovies (choose one)
// movies.RemoveAll(x => existingMovies.Contains(x));
// return movies;
//// or
// movies.RemoveAll(a => existingMovies.Exists(w => w.TmdbId == a.TmdbId));
// return movies;
//// or
return movies.Where(x => !existingMovies.Contains(x)).ToList();
return movies.Where(x => !_movieService.MovieExists(x)).ToList();
}
public List<Movie> MovieListSearch(int listId, bool onlyEnableAuto)
public List<Movie> MovieListSearch(int listId, bool onlyEnableAuto = false)
{
var movies = new List<Movie>();
var importLists = _netImportFactory.GetAvailableProviders();
var lists = listId == 0 ? importLists.Where(n => ((NetImportDefinition)n.Definition).Enabled == true) : importLists.Where(n => ((NetImportDefinition)n.Definition).Id == listId);
var lists = listId == 0 ? importLists : importLists.Where(n => ((NetImportDefinition)n.Definition).Id == listId);
if (onlyEnableAuto)
{
lists = importLists.Where(a => a.EnableAuto == true);
lists = importLists.Where(a => ((NetImportDefinition)a.Definition).EnableAuto);
}
foreach (var list in lists)
@@ -66,7 +58,16 @@ namespace NzbDrone.Core.NetImport
movies.AddRange(list.Fetch());
}
_logger.Debug("Found {0} movies from list(s) {1}", movies.Count, string.Join(", ", lists.Select(l => l.Definition.Name)));
return movies;
}
public void Execute(NetImportSyncCommand message)
{
var movies = FetchAndFilter(2, false);
_logger.Debug("Found {0} movies on your lists not in your library", movies.Count);
}
}
}