diff --git a/tubesync/sync/tasks.py b/tubesync/sync/tasks.py index 4e9b41c1..a356ac95 100644 --- a/tubesync/sync/tasks.py +++ b/tubesync/sync/tasks.py @@ -25,10 +25,11 @@ from django.db.transaction import atomic from django.utils import timezone from django.utils.translation import gettext_lazy as _ from background_task import background -from django_huey import db_periodic_task, db_task, task as huey_task # noqa -from huey import CancelExecution from background_task.exceptions import InvalidTaskError from background_task.models import Task, CompletedTask +from django_huey import db_periodic_task, db_task, task as huey_task # noqa +from huey import CancelExecution +from common.huey import exponential_backoff from common.logger import log from common.errors import ( BgTaskWorkerError, DownloadFailedException, NoFormatException, NoMediaException, @@ -894,7 +895,7 @@ def save_all_media_for_source(source_id): update_task_status(task, None) -@db_task(priority=50, queue=Val(TaskQueue.LIMIT)) +@exponential_backoff(db_task, priority=50, retries=15, queue=Val(TaskQueue.LIMIT)) def refresh_formats(media_id): try: media = Media.objects.get(pk=media_id)