From e04be4fbdd85839a1af67044c593ea24cd77901c Mon Sep 17 00:00:00 2001
From: Colin Powell <colin@unbl.ink>
Date: Fri, 24 Nov 2023 00:26:35 +0100
Subject: [PATCH] Fix Last.FM imports coming in as videos

---
 vrobbler/apps/music/aggregators.py | 4 +++-
 vrobbler/apps/music/lastfm.py      | 1 +
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/vrobbler/apps/music/aggregators.py b/vrobbler/apps/music/aggregators.py
index ba0ac5a..248b012 100644
--- a/vrobbler/apps/music/aggregators.py
+++ b/vrobbler/apps/music/aggregators.py
@@ -26,7 +26,9 @@ def scrobble_counts(user=None):
     starting_day_of_current_year = now.date().replace(month=1, day=1)
 
     finished_scrobbles_qs = Scrobble.objects.filter(
-        user_filter, played_to_completion=True
+        user_filter,
+        played_to_completion=True,
+        media_type=Scrobble.MediaType.TRACK,
     )
     data = {}
     data["today"] = finished_scrobbles_qs.filter(
diff --git a/vrobbler/apps/music/lastfm.py b/vrobbler/apps/music/lastfm.py
index ec20977..e4cdff8 100644
--- a/vrobbler/apps/music/lastfm.py
+++ b/vrobbler/apps/music/lastfm.py
@@ -68,6 +68,7 @@ class LastFM:
                 track=track,
                 played_to_completion=True,
                 in_progress=False,
+                media_type=Scrobble.MediaType.TRACK,
             )
             # Vrobbler scrobbles on finish, LastFM scrobbles on start
             seconds_eariler = timestamp - timedelta(seconds=20)