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.Languages;
using NzbDrone.Core.Movies; using NzbDrone.Core.Movies;
using NzbDrone.Core.Movies.Translations; using NzbDrone.Core.Movies.Translations;
using NzbDrone.Core.MovieStats;
using NzbDrone.Core.Tags; using NzbDrone.Core.Tags;
using NzbDrone.SignalR; using NzbDrone.SignalR;
using Radarr.Api.V3.Movies; using Radarr.Api.V3.Movies;
@ -21,6 +22,7 @@ namespace Radarr.Api.V3.Calendar
{ {
private readonly IMovieService _moviesService; private readonly IMovieService _moviesService;
private readonly IMovieTranslationService _movieTranslationService; private readonly IMovieTranslationService _movieTranslationService;
private readonly IMovieStatisticsService _movieStatisticsService;
private readonly IUpgradableSpecification _qualityUpgradableSpecification; private readonly IUpgradableSpecification _qualityUpgradableSpecification;
private readonly ITagService _tagService; private readonly ITagService _tagService;
private readonly IConfigService _configService; private readonly IConfigService _configService;
@ -28,6 +30,7 @@ namespace Radarr.Api.V3.Calendar
public CalendarController(IBroadcastSignalRMessage signalR, public CalendarController(IBroadcastSignalRMessage signalR,
IMovieService moviesService, IMovieService moviesService,
IMovieTranslationService movieTranslationService, IMovieTranslationService movieTranslationService,
IMovieStatisticsService movieStatisticsService,
IUpgradableSpecification qualityUpgradableSpecification, IUpgradableSpecification qualityUpgradableSpecification,
ITagService tagService, ITagService tagService,
IConfigService configService) IConfigService configService)
@ -35,6 +38,7 @@ namespace Radarr.Api.V3.Calendar
{ {
_moviesService = moviesService; _moviesService = moviesService;
_movieTranslationService = movieTranslationService; _movieTranslationService = movieTranslationService;
_movieStatisticsService = movieStatisticsService;
_qualityUpgradableSpecification = qualityUpgradableSpecification; _qualityUpgradableSpecification = qualityUpgradableSpecification;
_tagService = tagService; _tagService = tagService;
_configService = configService; _configService = configService;
@ -97,7 +101,10 @@ namespace Radarr.Api.V3.Calendar
var translations = _movieTranslationService.GetAllTranslationsForMovieMetadata(movie.MovieMetadataId); var translations = _movieTranslationService.GetAllTranslationsForMovieMetadata(movie.MovieMetadataId);
var translation = GetMovieTranslation(translations, movie.MovieMetadata, language); 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; return resources;
@ -116,5 +123,17 @@ namespace Radarr.Api.V3.Calendar
return translations.FirstOrDefault(t => t.Language == language && t.MovieMetadataId == movie.Id); 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;
}
} }
} }