tedkulp/Radarr
1
0
Fork 0
mirror of https://github.com/Radarr/Radarr.git synced 2026-04-23 22:25:14 -04:00
Code Issues Projects Releases Wiki Activity

New: Reject File if Sample Check Indeterminate

Browse Source
This commit is contained in:
Qstick
2020-07-19 01:45:23 -04:00
parent 1c85bd96f8
commit 2390ab98fa
6 changed files with 26 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/NzbDrone.Core/MediaFiles/DownloadedMovieImportService.cs
+1 -1
View File
@@ -131,7 +131,7 @@ namespace NzbDrone.Core.MediaFiles
return false;
}
if (_detectSample.IsSample(movie, videoFile, false) != DetectSampleResult.Sample)
if (_detectSample.IsSample(movie, videoFile) != DetectSampleResult.Sample)
{
_logger.Warn("Non-sample file detected: [{0}]", videoFile);
return false;
src/NzbDrone.Core/MediaFiles/MovieImport/DetectSample.cs
+4 -10
View File
@@ -9,7 +9,7 @@ namespace NzbDrone.Core.MediaFiles.MovieImport
{
public interface IDetectSample
{
DetectSampleResult IsSample(Movie movie, string path, bool isSpecial);
DetectSampleResult IsSample(Movie movie, string path);
}
public class DetectSample : IDetectSample
@@ -23,14 +23,8 @@ namespace NzbDrone.Core.MediaFiles.MovieImport
_logger = logger;
}
public DetectSampleResult IsSample(Movie movie, string path, bool isSpecial)
public DetectSampleResult IsSample(Movie movie, string path)
{
if (isSpecial)
{
_logger.Debug("Special, skipping sample check");
return DetectSampleResult.NotSample;
}
var extension = Path.GetExtension(path);
if (extension != null)
@@ -73,11 +67,11 @@ namespace NzbDrone.Core.MediaFiles.MovieImport
if (runTime.Value.TotalSeconds < minimumRuntime)
{
_logger.Debug("[{0}] appears to be a sample. Runtime: {1} seconds. Expected at least: {2} seconds", path, runTime, minimumRuntime);
_logger.Debug("[{0}] appears to be a sample. Runtime: {1} seconds. Expected at least: {2} seconds", path, runTime.Value.TotalSeconds, minimumRuntime);
return DetectSampleResult.Sample;
}
_logger.Debug("Runtime is over 90 seconds");
_logger.Debug("Runtime of {0} is more than {1} seconds, Not Sample", runTime.Value.TotalSeconds, minimumRuntime);
return DetectSampleResult.NotSample;
}
src/NzbDrone.Core/MediaFiles/MovieImport/ImportDecisionMaker.cs
+3 -3
View File
@@ -71,7 +71,7 @@ namespace NzbDrone.Core.MediaFiles.MovieImport
downloadClientItemInfo = _parsingService.EnhanceMovieInfo(downloadClientItemInfo);
}
var nonSampleVideoFileCount = GetNonSampleVideoFileCount(newFiles, movie, downloadClientItemInfo, folderInfo);
var nonSampleVideoFileCount = GetNonSampleVideoFileCount(newFiles, movie);
var decisions = new List<ImportDecision>();
@@ -180,11 +180,11 @@ namespace NzbDrone.Core.MediaFiles.MovieImport
return null;
}
private int GetNonSampleVideoFileCount(List<string> videoFiles, Movie movie, ParsedMovieInfo downloadClientItemInfo, ParsedMovieInfo folderInfo)
private int GetNonSampleVideoFileCount(List<string> videoFiles, Movie movie)
{
return videoFiles.Count(file =>
{
var sample = _detectSample.IsSample(movie, file, false);
var sample = _detectSample.IsSample(movie, file);
if (sample == DetectSampleResult.Sample)
{
src/NzbDrone.Core/MediaFiles/MovieImport/Specifications/NotSampleSpecification.cs
+11 -3
View File
@@ -19,14 +19,22 @@ namespace NzbDrone.Core.MediaFiles.MovieImport.Specifications
public Decision IsSatisfiedBy(LocalMovie localMovie, DownloadClientItem downloadClientItem)
{
var sample = _detectSample.IsSample(localMovie.Movie,
localMovie.Path,
false);
if (localMovie.ExistingFile)
{
_logger.Debug("Existing file, skipping sample check");
return Decision.Accept();
}
var sample = _detectSample.IsSample(localMovie.Movie, localMovie.Path);
if (sample == DetectSampleResult.Sample)
{
return Decision.Reject("Sample");
}
else if (sample == DetectSampleResult.Indeterminate)
{
return Decision.Reject("Unable to determine if file is a sample");
}
return Decision.Accept();
}