from celery import Task, shared_task

RETRY_QUEUE = "retry"


class DedicatedQueueRetryTask(Task):
    """
    Custom task sends all retried task to the dedicated retry queue.
    Is is needed to not to overload regular (high, medium, low) queues with retried tasks.
    """

    def retry(
        self, args=None, kwargs=None, exc=None, throw=True, eta=None, countdown=None, max_retries=None, **options
    ):
        # Just call retry with queue argument
        return super().retry(
            args=args,
            kwargs=kwargs,
            exc=exc,
            throw=throw,
            eta=eta,
            countdown=countdown,
            max_retries=max_retries,
            queue=RETRY_QUEUE,
            **options,
        )


def shared_dedicated_queue_retry_task(*args, **kwargs):
    return shared_task(*args, base=DedicatedQueueRetryTask, **kwargs)