From f550e32b5eca9459fb33a0a486beb9ee4e9cacd1 Mon Sep 17 00:00:00 2001
From: Someone <10882916+InterN0te@users.noreply.github.com>
Date: Sat, 9 Dec 2023 19:33:59 +0100
Subject: [PATCH] Fix secondary sorting on video key

---
 tubesync/sync/models.py | 24 +++++++-----------------
 1 file changed, 7 insertions(+), 17 deletions(-)

diff --git a/tubesync/sync/models.py b/tubesync/sync/models.py
index 7fd8c9e..46dca66 100644
--- a/tubesync/sync/models.py
+++ b/tubesync/sync/models.py
@@ -1410,35 +1410,25 @@ class Media(models.Model):
 
     def calculate_episode_number(self):
         filtered_media = Media.objects.filter(source=self.source)
+        position_counter = 1
         if self.source.source_type == Source.SOURCE_TYPE_YOUTUBE_PLAYLIST:
             # Calculate the episode number based on the position in the playlist
-            position_counter = 1
-
             for media in filtered_media:
                 if media == self:
                     return position_counter
                 position_counter += 1
         else:
             # Sort the filtered media by upload_date in ascending order and video ID
-            sorted_media = sorted(filtered_media, key=lambda x: (x.upload_date, x.id))
-
-            # Initialize an episode counter by year
-            episode_counter = {}
-
+            sorted_media = sorted(filtered_media, key=lambda x: (x.upload_date, x.key))
             self_year = self.upload_date.year if self.upload_date else self.created.year
-
             for media in sorted_media:
-                year = media.upload_date.year
-                if year not in episode_counter:
-                    # If it's the first video of the year, initialize the counter to 1
-                    episode_counter[year] = 1
-                else:
-                    # If it's not the first video of the year, use the current counter
-                    episode_counter[year] += 1
-
                 # Assign the calculated episode number to the media
                 if media == self:
-                    return episode_counter[self_year]
+                    return position_counter
+                year = media.upload_date.year if media.upload_date else media.created.year
+                if year == self_year:
+                    position_counter += 1
+
 
 
 class MediaServer(models.Model):