Fix completion for video games

This commit is contained in:
2023-03-05 17:27:39 -05:00
parent c757e743ac
commit df62865eea
6 changed files with 53 additions and 28 deletions

View File

@ -112,6 +112,14 @@ def check_scrobble_for_finish(
def get_scrobbles_for_media(media_obj, user: User) -> models.QuerySet:
Scrobble = apps.get_model(app_label="scrobbles", model_name="Scrobble")
if media_obj.__class__.__name__ == "Book":
media_query = None
media_class = media_obj.__class__.__name__
if media_class == "Book":
media_query = models.Q(book=media_obj)
if media_class == "VideoGame":
media_query = models.Q(video_game=media_obj)
if not media_query:
logger.warn("Do not know about media {media_class} 🙍")
return []
return Scrobble.objects.filter(media_query, user=user)