Cutlery/Kyoo
1
0
Fork 0
mirror of https://github.com/zoriya/Kyoo.git synced 2025-06-02 05:04:15 -04:00
Code Issues Packages Projects Releases Wiki Activity

Refining exception handling of the crawler

Browse Source
This commit is contained in:
Zoe Roux 2020-06-15 03:11:09 +02:00
parent fc65716906
commit 0728ef5e67
1 changed files with 56 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

104
Kyoo/Tasks/Crawler.cs
View File

@ -51,31 +51,24 @@ namespace Kyoo.Controllers
_transcoder = serviceProvider.GetService<ITranscoder>(); _transcoder = serviceProvider.GetService<ITranscoder>();
_config = serviceProvider.GetService<IConfiguration>(); _config = serviceProvider.GetService<IConfiguration>();
try using IServiceScope serviceScope = _serviceProvider.CreateScope();
{ await using ILibraryManager libraryManager = serviceScope.ServiceProvider.GetService<ILibraryManager>();
using IServiceScope serviceScope = _serviceProvider.CreateScope(); ICollection<Episode> episodes = await libraryManager.GetEpisodes();
await using ILibraryManager libraryManager = serviceScope.ServiceProvider.GetService<ILibraryManager>(); ICollection<Library> libraries = argument == null
ICollection<Episode> episodes = await libraryManager.GetEpisodes(); ? await libraryManager.GetLibraries()
ICollection<Library> libraries = argument == null : new [] { await libraryManager.GetLibrary(argument)};
? await libraryManager.GetLibraries()
: new [] { await libraryManager.GetLibrary(argument)};
foreach (Episode episode in episodes) foreach (Episode episode in episodes)
{
if (!File.Exists(episode.Path))
await libraryManager.DeleteEpisode(episode);
}
// TODO replace this grotesque way to load the providers.
foreach (Library library in libraries)
library.Providers = library.Providers;
await Task.WhenAll(libraries.Select(x => Scan(x, episodes, cancellationToken)).ToArray());
}
catch (Exception ex)
{ {
await Console.Error.WriteLineAsync($"Unknown exception thrown durring libraries scan.\nException: {ex.Message}"); if (!File.Exists(episode.Path))
await libraryManager.DeleteEpisode(episode);
} }
// TODO replace this grotesque way to load the providers.
foreach (Library library in libraries)
library.Providers = library.Providers;
await Task.WhenAll(libraries.Select(x => Scan(x, episodes, cancellationToken)).ToArray());
Console.WriteLine("Scan finished!"); Console.WriteLine("Scan finished!");
} }
@ -135,34 +128,49 @@ namespace Kyoo.Controllers
if (token.IsCancellationRequested) if (token.IsCancellationRequested)
return; return;
using IServiceScope serviceScope = _serviceProvider.CreateScope(); try
await using ILibraryManager libraryManager = serviceScope.ServiceProvider.GetService<ILibraryManager>();
string patern = _config.GetValue<string>("regex");
Regex regex = new Regex(patern, RegexOptions.IgnoreCase);
Match match = regex.Match(relativePath);
string showPath = Path.GetDirectoryName(path);
string collectionName = match.Groups["Collection"]?.Value;
string showName = match.Groups["ShowTitle"].Value;
int seasonNumber = int.TryParse(match.Groups["Season"].Value, out int tmp) ? tmp : -1;
int episodeNumber = int.TryParse(match.Groups["Episode"].Value, out tmp) ? tmp : -1;
int absoluteNumber = int.TryParse(match.Groups["Absolute"].Value, out tmp) ? tmp : -1;
Collection collection = await GetCollection(libraryManager, collectionName, library);
bool isMovie = seasonNumber == -1 && episodeNumber == -1 && absoluteNumber == -1;
Show show = await GetShow(libraryManager, showName, showPath, isMovie, library);
if (isMovie)
await libraryManager.RegisterEpisode(await GetMovie(show, path));
else
{ {
Season season = await GetSeason(libraryManager, show, seasonNumber, library); using IServiceScope serviceScope = _serviceProvider.CreateScope();
Episode episode = await GetEpisode(libraryManager, show, season, episodeNumber, absoluteNumber, path, library); await using ILibraryManager libraryManager = serviceScope.ServiceProvider.GetService<ILibraryManager>();
await libraryManager.RegisterEpisode(episode);
} string patern = _config.GetValue<string>("regex");
Regex regex = new Regex(patern, RegexOptions.IgnoreCase);
Match match = regex.Match(relativePath);
await libraryManager.AddShowLink(show, library, collection); string showPath = Path.GetDirectoryName(path);
Console.WriteLine($"Episode at {path} registered."); string collectionName = match.Groups["Collection"]?.Value;
string showName = match.Groups["ShowTitle"].Value;
int seasonNumber = int.TryParse(match.Groups["Season"].Value, out int tmp) ? tmp : -1;
int episodeNumber = int.TryParse(match.Groups["Episode"].Value, out tmp) ? tmp : -1;
int absoluteNumber = int.TryParse(match.Groups["Absolute"].Value, out tmp) ? tmp : -1;
Collection collection = await GetCollection(libraryManager, collectionName, library);
bool isMovie = seasonNumber == -1 && episodeNumber == -1 && absoluteNumber == -1;
Show show = await GetShow(libraryManager, showName, showPath, isMovie, library);
if (isMovie)
await libraryManager.RegisterEpisode(await GetMovie(show, path));
else
{
Season season = await GetSeason(libraryManager, show, seasonNumber, library);
Episode episode = await GetEpisode(libraryManager,
show,
season,
episodeNumber,
absoluteNumber,
path,
library);
await libraryManager.RegisterEpisode(episode);
}
await libraryManager.AddShowLink(show, library, collection);
Console.WriteLine($"Episode at {path} registered.");
}
catch (Exception ex)
{
await Console.Error.WriteLineAsync($"Unknown exception thrown while registering episode at {path}." +
$"\nException: {ex.Message}" +
$"\nAt {ex.StackTrace}");
}
} }
private async Task<Collection> GetCollection(ILibraryManager libraryManager, private async Task<Collection> GetCollection(ILibraryManager libraryManager,