fix some media not appearing in timeline #625

Author: FuzzyGrim

src/app/statistics.py

@@ -323,7 +323,8 @@ def get_timeline(user_media):
         for media in queryset:
             local_start_date = timezone.localdate(media.start_date)
             local_end_date = timezone.localdate(media.end_date)
-            if local_start_date and local_end_date:
+
+            if media.start_date and media.end_date:
                 # add media to all months between start and end
                 current_date = local_start_date
                 while current_date <= local_end_date:
@@ -337,23 +338,22 @@ def get_timeline(user_media):
                     # Move to next month
                     current_date += relativedelta(months=1)
                     current_date = current_date.replace(day=1)
-            elif local_start_date:
+            elif media.start_date:
                 # If only start date, add to the start month
                 year = local_start_date.year
                 month = local_start_date.month
                 month_name = calendar.month_name[month]
                 month_year = f"{month_name} {year}"
 
                 timeline[month_year].append(media)
-            elif local_end_date:
+            elif media.end_date:
                 # If only end date, add to the end month
                 year = local_end_date.year
                 month = local_end_date.month
                 month_name = calendar.month_name[month]
                 month_year = f"{month_name} {year}"
 
                 timeline[month_year].append(media)
-
     # Convert to sorted dictionary with media sorted by start date
     # Create a list sorted by year and month in reverse order
     sorted_items = []
@@ -371,7 +371,6 @@ def get_timeline(user_media):
     for month_year, media_list, _, _ in sorted_items:
         # Sort the media list using our custom sort key
         result[month_year] = sorted(media_list, key=time_line_sort_key)
-
     return result