diff --git a/src/task_queue/database_tasks/media_library_cleanup.py b/src/task_queue/database_tasks/media_library_cleanup.py index db99fa3bf..bb221fbc6 100644 --- a/src/task_queue/database_tasks/media_library_cleanup.py +++ b/src/task_queue/database_tasks/media_library_cleanup.py @@ -11,14 +11,14 @@ ) from api.store.media_library import MediaLibraryStore as media_store from api.utils.constants import MEDIA_LIBRARY_CLEANUP_TEMPLATE -from database.models import FeatureFlag +from database.models import Community, FeatureFlag from task_queue.models import Task REMOVE_DUPLICATE_IMAGE_FLAG_KEY = "remove-duplicate-images-feature-flag" -def remove_duplicate_images(task): +def remove_duplicate_images(task=None): """ This checks all media on the platform and removes all duplicates. Its based on the "Remove Duplicate Images" feature flag. For communities that are subscribed @@ -33,7 +33,10 @@ def remove_duplicate_images(task): print("Generating hashes") result = media_store.generate_hashes(None, None, None) - communities = flag.enabled_communities() + if flag and do_updates: + communities = flag.enabled_communities() + else: + communities = Community.objects.all() # task = Task.objects.filter(name="Media Library Cleanup Routine").first() ids = [c.id for c in communities] clean_and_notify(ids,None,task.creator, do_updates)