mirror
/
Radarr
mirror of https://github.com/Radarr/Radarr
1
0
Fork 0
Code Issues Releases Wiki Activity

Movie stats for Calendar

This commit is contained in:
Bogdan 2024-01-18 01:18:52 +02:00
parent 84c2d7f69d
commit cc7104a814
1 changed files with 20 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/Radarr.Api.V3/Calendar/CalendarController.cs
View File

@ -8,6 +8,7 @@ using NzbDrone.Core.DecisionEngine.Specifications;
using NzbDrone.Core.Languages;
using NzbDrone.Core.Movies;
using NzbDrone.Core.Movies.Translations;
using NzbDrone.Core.MovieStats;
using NzbDrone.Core.Tags;
using NzbDrone.SignalR;
using Radarr.Api.V3.Movies;
@ -21,6 +22,7 @@ namespace Radarr.Api.V3.Calendar
{
private readonly IMovieService _moviesService;
private readonly IMovieTranslationService _movieTranslationService;
private readonly IMovieStatisticsService _movieStatisticsService;
private readonly IUpgradableSpecification _qualityUpgradableSpecification;
private readonly ITagService _tagService;
private readonly IConfigService _configService;
@ -28,6 +30,7 @@ namespace Radarr.Api.V3.Calendar
public CalendarController(IBroadcastSignalRMessage signalR,
IMovieService moviesService,
IMovieTranslationService movieTranslationService,
IMovieStatisticsService movieStatisticsService,
IUpgradableSpecification qualityUpgradableSpecification,
ITagService tagService,
IConfigService configService)
@ -35,6 +38,7 @@ namespace Radarr.Api.V3.Calendar
{
_moviesService = moviesService;
_movieTranslationService = movieTranslationService;
_movieStatisticsService = movieStatisticsService;
_qualityUpgradableSpecification = qualityUpgradableSpecification;
_tagService = tagService;
_configService = configService;
@ -97,7 +101,10 @@ namespace Radarr.Api.V3.Calendar
var translations = _movieTranslationService.GetAllTranslationsForMovieMetadata(movie.MovieMetadataId);
var translation = GetMovieTranslation(translations, movie.MovieMetadata, language);
resources.Add(movie.ToResource(availDelay, translation, _qualityUpgradableSpecification));
var resource = movie.ToResource(availDelay, translation, _qualityUpgradableSpecification);
FetchAndLinkMovieStatistics(resource);
resources.Add(resource);
}
return resources;
@ -116,5 +123,17 @@ namespace Radarr.Api.V3.Calendar
return translations.FirstOrDefault(t => t.Language == language && t.MovieMetadataId == movie.Id);
}
private void FetchAndLinkMovieStatistics(MovieResource resource)
{
LinkMovieStatistics(resource, _movieStatisticsService.MovieStatistics(resource.Id));
}
private void LinkMovieStatistics(MovieResource resource, MovieStatistics movieStatistics)
{
resource.Statistics = movieStatistics.ToResource();
resource.HasFile = movieStatistics.MovieFileCount > 0;
resource.SizeOnDisk = movieStatistics.SizeOnDisk;
}
}
}