* Move the message to the configured export queue.

bgroff · Jul 10, 2018 · 16bf6d5 · 16bf6d5
1 parent bb47de4
commit 16bf6d5
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 2 deletions.
diff --git a/django_kala/django_kala/settings/celery.py b/django_kala/django_kala/settings/celery.py
@@ -1,4 +1,10 @@
 from .functions import get_env_variable
 
+EXPORT_QUEUE = get_env_variable('EXPORT_QUEUE', default='exports')
+
 CELERY_BROKER_URL = 'sqs://'
 CELERY_BROKER_TRANSPORT_OPTIONS = {'region': get_env_variable('AWS_REGION', default='us-west-2')}
+CELERY_TASK_ROUTES = {
+    'projects.tasks.export_project.ExportProjectTask': EXPORT_QUEUE,
+    'projects.tasks.export_document.ExportDocumentTask': EXPORT_QUEUE,
+}
diff --git a/django_kala/projects/views/documents/document.py b/django_kala/projects/views/documents/document.py
@@ -55,7 +55,7 @@ def dispatch(self, request, project_pk, document_pk, *args, **kwargs):
 
     def get(self, request, *args, **kwargs):
         task = ExportDocumentTask()
-        task.apply_async([self.document.pk, request.user.pk])
+        task.apply_async([self.document.pk, request.user.pk], queue=settings.EXPORT_QUEUE)
 
         return redirect(
             reverse(

diff --git a/django_kala/projects/views/projects/project.py b/django_kala/projects/views/projects/project.py
@@ -1,3 +1,4 @@
+from django.conf import settings
 from django.contrib.auth.mixins import LoginRequiredMixin
 from django.contrib.postgres.search import SearchVector
 from django.core.exceptions import PermissionDenied
@@ -129,7 +130,7 @@ def dispatch(self, request, pk, *args, **kwargs):
 
     def get(self, request, *args, **kwargs):
         task = ExportProjectTask()
-        task.apply_async([self.project.pk, request.user.pk])
+        task.apply_async([self.project.pk, request.user.pk], queue=settings.EXPORT_QUEUE)
 
         return redirect(
             reverse(