A celery abstract task class that is fired when a transaction is committed

celery.py

class TransactionAwareTask(Task):
    '''
    Task class which is aware of django db transactions and only executes tasks
    after transaction has been committed
    '''
    abstract = True

    def apply_async(self, *args, **kwargs):
        '''
        Unlike the default task in celery, this task does not return an async
        result
        '''
        transaction.on_commit(
            lambda: super(TransactionAwareTask, self).apply_async(
                *args, **kwargs))

jobs.py

@shared_task(base=TransactionAwareTask, bind=True, max_retries=settings.EVENT_MAX_RETRIES)
def push_event(self, event_id):
    event = Event.objects.get(id=event_id)
    # code to push webhook

old_jobs.py

@shared_task(bind=True, max_retries=settings.EVENT_MAX_RETRIES)
def push_event(self, event_id):
    event = Event.objects.get(id=event_id)
    # code to push webhook

views.py

with transaction.atomic():
    # update task
    # update driver
    event = Event.objects.create(...)
    push_event.delay(event.id)