From 85fd8f2c65e407cba689e42177de8b929c445711 Mon Sep 17 00:00:00 2001
From: Vlad Ilies <ammi.vlad@gmail.com>
Date: Thu, 19 Jan 2017 20:08:15 +0200
Subject: [PATCH] bug fix for 15 movie wanted tab (#348)

---
 src/NzbDrone.Core/Tv/MovieRepository.cs | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/src/NzbDrone.Core/Tv/MovieRepository.cs b/src/NzbDrone.Core/Tv/MovieRepository.cs
index d9a8cc32e..b51a417d8 100644
--- a/src/NzbDrone.Core/Tv/MovieRepository.cs
+++ b/src/NzbDrone.Core/Tv/MovieRepository.cs
@@ -4,6 +4,7 @@ using System.Collections.Generic;
 using NzbDrone.Core.Datastore;
 using NzbDrone.Core.Messaging.Events;
 using NzbDrone.Core.Datastore.Extensions;
+using Marr.Data.QGen;
 
 namespace NzbDrone.Core.Tv
 {
@@ -137,17 +138,20 @@ namespace NzbDrone.Core.Tv
         public PagingSpec<Movie> MoviesWithoutFiles(PagingSpec<Movie> pagingSpec)
         {
 
-            var query = Query.Where(pagingSpec.FilterExpression)
+            pagingSpec.TotalRecords = GetMoviesWithoutFilesQuery(pagingSpec).GetRowCount();
+            pagingSpec.Records = GetMoviesWithoutFilesQuery(pagingSpec).ToList();
+
+            return pagingSpec;
+        }
+
+        public SortBuilder<Movie> GetMoviesWithoutFilesQuery(PagingSpec<Movie> pagingSpec)
+        {
+            return Query.Where(pagingSpec.FilterExpression)
                              .AndWhere(m => m.MovieFileId == 0)
                              .AndWhere(m => m.Status == MovieStatusType.Released)
                              .OrderBy(pagingSpec.OrderByClause(), pagingSpec.ToSortDirection())
                              .Skip(pagingSpec.PagingOffset())
                              .Take(pagingSpec.PageSize);
-
-            pagingSpec.Records = query.ToList();
-            pagingSpec.TotalRecords = pagingSpec.Records.Count;
-
-            return pagingSpec;
         }
     }
 }
\ No newline at end of file