celery flower service added

Healthlane-Technologies · Aug 30, 2024 · 7bbf449 · 7bbf449
1 parent 8e08316
commit 7bbf449
Show file tree

Hide file tree

Showing 7 changed files with 58 additions and 1 deletion.
diff --git a/backend/requirements/base.txt b/backend/requirements/base.txt
@@ -67,3 +67,4 @@ python3-saml==1.16.0
 pytz==2024.1
 setuptools==70.1.1
 XlsxWriter==3.1.9
+flower==2.0.1
diff --git a/deploy/dev.dockerfile b/deploy/dev.dockerfile
@@ -19,5 +19,6 @@ USER zango_user
 
 
 COPY init.sh /zango/
+COPY start_flower.sh /zango/
 WORKDIR /zango/
 CMD ["/bin/sh", "init.sh"]
diff --git a/deploy/docker_compose.dev.yml b/deploy/docker_compose.dev.yml
@@ -60,6 +60,29 @@ services:
       redis:
         condition: service_healthy
 
+  celery-flower:
+    build:
+      context: .
+      dockerfile: dev.dockerfile
+      args:
+        - HOST_UID=${HOST_UID}
+        - HOST_GID=${HOST_GID}
+    restart: always
+    entrypoint:
+      - /bin/sh
+      - start_flower.sh
+    ports:
+      - "5555:5555"
+    env_file:
+      - .env
+    volumes:
+      - .:/zango/
+    depends_on:
+      - postgres
+      - redis
+      - celery
+      - celery_beat
+
   redis:
     image: redis
     ports:
@@ -83,6 +106,5 @@ services:
       redis:
         condition: service_healthy
 
-
 volumes:
   dev_db:
diff --git a/deploy/docker_compose.prod.yml b/deploy/docker_compose.prod.yml
@@ -81,6 +81,28 @@ services:
       timeout: 5s
       retries: 3
 
+  celery-flower:
+    build:
+      context: .
+      dockerfile: prod.dockerfile
+      args:
+        - HOST_UID=${HOST_UID}
+        - HOST_GID=${HOST_GID}
+    restart: always
+    entrypoint:
+      - /bin/sh
+      - start_flower.sh
+    ports:
+      - "5555:5555"
+    env_file:
+      - .env
+    volumes:
+      - .:/zango/
+    depends_on:
+      - postgres
+      - redis
+      - celery
+
   celery_beat:
     image: kczelthy/zango:latest
     command: /bin/sh -c "cd ${PROJECT_NAME} && celery -A ${PROJECT_NAME} beat -l info --scheduler django_celery_beat.schedulers:DatabaseScheduler"

diff --git a/deploy/prod.dockerfile b/deploy/prod.dockerfile
@@ -19,5 +19,6 @@ RUN apt update && \
 USER zango_user
 
 COPY init.sh /zango/
+COPY start_flower.sh /zango/
 WORKDIR /zango/
 CMD ["/bin/sh", "init.sh"]
diff --git a/deploy/start_flower.sh b/deploy/start_flower.sh
@@ -0,0 +1,9 @@
+#!/bin/sh
+
+cd ${PROJECT_NAME}
+until timeout 5s celery -A ${PROJECT_NAME} inspect ping; do
+    >&2 echo "Celery workers not available"
+done
+
+echo 'Starting flower'
+celery -A ${PROJECT_NAME} flower --port=5555 --broker=redis://redis:6379/0
diff --git a/setup_project.py b/setup_project.py
@@ -24,6 +24,7 @@ def load_necessary_files(project_dir, project_name, without_db):
     shutil.copy("deploy/dev.dockerfile", f"{project_dir}/dev.dockerfile")
     shutil.copy("deploy/prod.dockerfile", f"{project_dir}/prod.dockerfile")
     shutil.copy("deploy/init.sh", f"{project_dir}/init.sh")
+    shutil.copy("deploy/start_flower.sh", f"{project_dir}/start_flower.sh")
 
 
 def write_env_file(project_dir, args):