From e5d124e807769755e2db5962cfe5256812a0ef14 Mon Sep 17 00:00:00 2001
From: tcely <tcely@users.noreply.github.com>
Date: Sat, 22 Feb 2025 08:04:44 -0500
Subject: [PATCH] Update `rename_all_media_for_source`

Include the recent optimizations.
---
 tubesync/sync/tasks.py | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/tubesync/sync/tasks.py b/tubesync/sync/tasks.py
index f1a40fb6..eefcab3b 100644
--- a/tubesync/sync/tasks.py
+++ b/tubesync/sync/tasks.py
@@ -633,7 +633,25 @@ def rename_all_media_for_source(source_id):
         log.error(f'Task rename_all_media_for_source(pk={source_id}) called but no '
                   f'source exists with ID: {source_id}')
         return
-    for media in Media.objects.filter(source=source):
+    # Check that the settings allow renaming
+    rename_sources_setting = settings.RENAME_SOURCES or list()
+    create_rename_tasks = (
+        (
+            source.directory and
+            source.directory in rename_sources_setting
+        ) or
+        settings.RENAME_ALL_SOURCES
+    )
+    if not create_rename_tasks:
+        return
+    mqs = Media.objects.all().defer(
+        'metadata',
+        'thumb',
+    ).filter(
+        source=source,
+        downloaded=True,
+    )
+    for media in mqs:
         media.rename_files()