web/notify.moe
Code Issues Pull Requests Activity

Improved top studios calculation

Browse Source
This commit is contained in:
Eduard Urbach 2018-11-29 15:07:17 +09:00
parent f807b446e7
commit 13bb27a957
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pages/profile/profile.go
View File

@ -51,7 +51,7 @@ func Profile(ctx *aero.Context, viewUser *arn.User) string {
var topStudios []*arn.Company var topStudios []*arn.Company
for _, item := range animeList.Items { for _, item := range animeList.Items {
if item.Status == arn.AnimeListStatusPlanned { if item.Status != arn.AnimeListStatusCompleted {
continue continue
} }
@ -59,15 +59,23 @@ func Profile(ctx *aero.Context, viewUser *arn.User) string {
reWatch := item.RewatchCount * item.Anime().EpisodeCount * item.Anime().EpisodeLength reWatch := item.RewatchCount * item.Anime().EpisodeCount * item.Anime().EpisodeLength
duration := time.Duration(currentWatch + reWatch) duration := time.Duration(currentWatch + reWatch)
animeWatchingTime += duration * time.Minute animeWatchingTime += duration * time.Minute
rating := 0.0
if item.Rating.Overall != 0 {
rating = item.Rating.Overall - arn.AverageRating
} else {
// Add 0.1 to avoid all affinities being 0 when a user doesn't have any rated anime.
rating = 0.1
}
for _, studio := range item.Anime().Studios() { for _, studio := range item.Anime().Studios() {
count, exists := studios[studio.ID] affinity, exists := studios[studio.ID]
if !exists { if !exists {
topStudios = append(topStudios, studio) topStudios = append(topStudios, studio)
} }
studios[studio.ID] = count + 1 studios[studio.ID] = affinity + rating
} }
} }