#680 - Support new episode file sorting

This commit is contained in:
Luke Pulverenti 2014-01-19 11:33:36 -05:00
parent 6e2d4ad150
commit 28c88174ae
3 changed files with 80 additions and 47 deletions

View File

@ -331,6 +331,11 @@ namespace MediaBrowser.Model.Configuration
public bool DeleteEmptyFolders { get; set; } public bool DeleteEmptyFolders { get; set; }
/// <summary>
/// Will log results but will not actually make any changes
/// </summary>
public bool EnableTrialMode { get; set; }
public FileSortingOptions() public FileSortingOptions()
{ {
MinFileSizeMb = 50; MinFileSizeMb = 50;
@ -344,6 +349,8 @@ namespace MediaBrowser.Model.Configuration
SeasonFolderPattern = "Season %s"; SeasonFolderPattern = "Season %s";
SeasonZeroFolderName = "Season 0"; SeasonZeroFolderName = "Season 0";
EnableTrialMode = true;
} }
} }
} }

View File

@ -1,10 +1,10 @@
using MediaBrowser.Common.ScheduledTasks; using MediaBrowser.Common.IO;
using MediaBrowser.Common.ScheduledTasks;
using MediaBrowser.Controller.Configuration; using MediaBrowser.Controller.Configuration;
using MediaBrowser.Controller.Library; using MediaBrowser.Controller.Library;
using MediaBrowser.Model.Logging; using MediaBrowser.Model.Logging;
using System; using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.Linq;
using System.Threading; using System.Threading;
using System.Threading.Tasks; using System.Threading.Tasks;
@ -15,12 +15,14 @@ namespace MediaBrowser.Server.Implementations.FileSorting
private readonly IServerConfigurationManager _config; private readonly IServerConfigurationManager _config;
private readonly ILogger _logger; private readonly ILogger _logger;
private readonly ILibraryManager _libraryManager; private readonly ILibraryManager _libraryManager;
private readonly IFileSystem _fileSystem;
public SortingScheduledTask(IServerConfigurationManager config, ILogger logger, ILibraryManager libraryManager) public SortingScheduledTask(IServerConfigurationManager config, ILogger logger, ILibraryManager libraryManager, IFileSystem fileSystem)
{ {
_config = config; _config = config;
_logger = logger; _logger = logger;
_libraryManager = libraryManager; _libraryManager = libraryManager;
_fileSystem = fileSystem;
} }
public string Name public string Name
@ -45,34 +47,7 @@ namespace MediaBrowser.Server.Implementations.FileSorting
private void SortFiles(CancellationToken cancellationToken, IProgress<double> progress) private void SortFiles(CancellationToken cancellationToken, IProgress<double> progress)
{ {
var numComplete = 0; new TvFileSorter(_libraryManager, _logger, _fileSystem).Sort(_config.Configuration.FileSortingOptions, cancellationToken, progress);
var paths = _config.Configuration.FileSortingOptions.TvWatchLocations.ToList();
foreach (var path in paths)
{
cancellationToken.ThrowIfCancellationRequested();
try
{
SortFiles(path);
}
catch (Exception ex)
{
_logger.ErrorException("Error sorting files from {0}", ex, path);
}
numComplete++;
double percent = numComplete;
percent /= paths.Count;
progress.Report(100 * percent);
}
}
private void SortFiles(string path)
{
new TvFileSorter(_libraryManager, _logger).Sort(path, _config.Configuration.FileSortingOptions);
} }
public IEnumerable<ITaskTrigger> GetDefaultTriggers() public IEnumerable<ITaskTrigger> GetDefaultTriggers()

View File

@ -1,5 +1,4 @@
using System.Globalization; using MediaBrowser.Common.IO;
using MediaBrowser.Common.IO;
using MediaBrowser.Controller.Entities.TV; using MediaBrowser.Controller.Entities.TV;
using MediaBrowser.Controller.Library; using MediaBrowser.Controller.Library;
using MediaBrowser.Controller.Providers; using MediaBrowser.Controller.Providers;
@ -9,8 +8,10 @@ using MediaBrowser.Model.Entities;
using MediaBrowser.Model.Logging; using MediaBrowser.Model.Logging;
using System; using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.Globalization;
using System.IO; using System.IO;
using System.Linq; using System.Linq;
using System.Threading;
namespace MediaBrowser.Server.Implementations.FileSorting namespace MediaBrowser.Server.Implementations.FileSorting
{ {
@ -20,23 +21,25 @@ namespace MediaBrowser.Server.Implementations.FileSorting
private readonly ILogger _logger; private readonly ILogger _logger;
private readonly IFileSystem _fileSystem; private readonly IFileSystem _fileSystem;
private static CultureInfo _usCulture = new CultureInfo("en-US"); private static readonly CultureInfo UsCulture = new CultureInfo("en-US");
public TvFileSorter(ILibraryManager libraryManager, ILogger logger) public TvFileSorter(ILibraryManager libraryManager, ILogger logger, IFileSystem fileSystem)
{ {
_libraryManager = libraryManager; _libraryManager = libraryManager;
_logger = logger; _logger = logger;
_fileSystem = fileSystem;
} }
public void Sort(string path, FileSortingOptions options) public void Sort(FileSortingOptions options, CancellationToken cancellationToken, IProgress<double> progress)
{ {
var minFileBytes = options.MinFileSizeMb * 1024 * 1024; var minFileBytes = options.MinFileSizeMb * 1024 * 1024;
var eligibleFiles = new DirectoryInfo(path) var eligibleFiles = options.TvWatchLocations.SelectMany(GetEligibleFiles)
.EnumerateFiles("*", SearchOption.AllDirectories)
.Where(i => EntityResolutionHelper.IsVideoFile(i.FullName) && i.Length >= minFileBytes) .Where(i => EntityResolutionHelper.IsVideoFile(i.FullName) && i.Length >= minFileBytes)
.ToList(); .ToList();
progress.Report(10);
if (eligibleFiles.Count > 0) if (eligibleFiles.Count > 0)
{ {
var allSeries = _libraryManager.RootFolder var allSeries = _libraryManager.RootFolder
@ -44,12 +47,27 @@ namespace MediaBrowser.Server.Implementations.FileSorting
.Where(i => i.LocationType == LocationType.FileSystem) .Where(i => i.LocationType == LocationType.FileSystem)
.ToList(); .ToList();
var numComplete = 0;
foreach (var file in eligibleFiles) foreach (var file in eligibleFiles)
{ {
SortFile(file.FullName, options, allSeries); SortFile(file.FullName, options, allSeries);
numComplete++;
double percent = numComplete;
percent /= eligibleFiles.Count;
progress.Report(100 * percent);
} }
} }
cancellationToken.ThrowIfCancellationRequested();
progress.Report(99);
if (!options.EnableTrialMode)
{
foreach (var path in options.TvWatchLocations)
{
if (options.LeftOverFileExtensionsToDelete.Length > 0) if (options.LeftOverFileExtensionsToDelete.Length > 0)
{ {
DeleteLeftOverFiles(path, options.LeftOverFileExtensionsToDelete); DeleteLeftOverFiles(path, options.LeftOverFileExtensionsToDelete);
@ -60,6 +78,26 @@ namespace MediaBrowser.Server.Implementations.FileSorting
DeleteEmptyFolders(path); DeleteEmptyFolders(path);
} }
} }
}
progress.Report(100);
}
private IEnumerable<FileInfo> GetEligibleFiles(string path)
{
try
{
return new DirectoryInfo(path)
.EnumerateFiles("*", SearchOption.AllDirectories)
.ToList();
}
catch (IOException ex)
{
_logger.ErrorException("Error getting files from {0}", ex, path);
return new List<FileInfo>();
}
}
private void SortFile(string path, FileSortingOptions options, IEnumerable<Series> allSeries) private void SortFile(string path, FileSortingOptions options, IEnumerable<Series> allSeries)
{ {
@ -147,8 +185,8 @@ namespace MediaBrowser.Server.Implementations.FileSorting
var episodeFileName = string.Format("{0} - {1}x{2} - {3}", var episodeFileName = string.Format("{0} - {1}x{2} - {3}",
_fileSystem.GetValidFilename(series.Name), _fileSystem.GetValidFilename(series.Name),
seasonNumber.ToString(_usCulture), seasonNumber.ToString(UsCulture),
episodeNumber.ToString("00", _usCulture), episodeNumber.ToString("00", UsCulture),
_fileSystem.GetValidFilename(episode.Name) _fileSystem.GetValidFilename(episode.Name)
); );
@ -158,6 +196,13 @@ namespace MediaBrowser.Server.Implementations.FileSorting
return newPath; return newPath;
} }
/// <summary>
/// Gets the season folder path.
/// </summary>
/// <param name="series">The series.</param>
/// <param name="seasonNumber">The season number.</param>
/// <param name="options">The options.</param>
/// <returns>System.String.</returns>
private string GetSeasonFolderPath(Series series, int seasonNumber, FileSortingOptions options) private string GetSeasonFolderPath(Series series, int seasonNumber, FileSortingOptions options)
{ {
// If there's already a season folder, use that // If there's already a season folder, use that
@ -184,13 +229,19 @@ namespace MediaBrowser.Server.Implementations.FileSorting
} }
var seasonFolderName = options.SeasonFolderPattern var seasonFolderName = options.SeasonFolderPattern
.Replace("%s", seasonNumber.ToString(_usCulture)) .Replace("%s", seasonNumber.ToString(UsCulture))
.Replace("%0s", seasonNumber.ToString("00", _usCulture)) .Replace("%0s", seasonNumber.ToString("00", UsCulture))
.Replace("%00s", seasonNumber.ToString("000", _usCulture)); .Replace("%00s", seasonNumber.ToString("000", UsCulture));
return Path.Combine(path, _fileSystem.GetValidFilename(seasonFolderName)); return Path.Combine(path, _fileSystem.GetValidFilename(seasonFolderName));
} }
/// <summary>
/// Gets the matching series.
/// </summary>
/// <param name="seriesName">Name of the series.</param>
/// <param name="allSeries">All series.</param>
/// <returns>Series.</returns>
private Series GetMatchingSeries(string seriesName, IEnumerable<Series> allSeries) private Series GetMatchingSeries(string seriesName, IEnumerable<Series> allSeries)
{ {
int? yearInName; int? yearInName;