Merge pull request #4 from omidcodes/004-add-celery-background-task

add celery task to update logs/task_activity.log on each Task object …

Author: omidcodes

.gitignore

@@ -175,4 +175,8 @@ cython_debug/
 
 
 # vscode
-.vscode
+.vscode
+
+# logs folder remains, but the contents will be ignored.
+logs/*
+!logs/.gitkeep

logs/.gitkeep

@@ -0,0 +1,3 @@
+from .celery import app as celery_app
+
+__all__ = ['celery_app']

requirements.txt

@@ -0,0 +1,8 @@
+import os
+from celery import Celery
+
+os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'taskflow_api.settings')
+
+app = Celery('taskflow_api')
+app.config_from_object('django.conf:settings', namespace='CELERY')
+app.autodiscover_tasks()

taskflow_api/__init__.py

@@ -1,4 +1,3 @@
-import os
 from pathlib import Path
 from decouple import config, Csv
 
@@ -126,4 +125,7 @@
     }
 }
 
-CELERY_BROKER_URL = config('CELERY_BROKER_URL', default='amqp://guest:guest@localhost:5672//')
+CELERY_BROKER_URL = config('CELERY_BROKER_URL', default='amqp://guest:guest@localhost:5672//')
+
+CELERY_ACCEPT_CONTENT = ['json']
+CELERY_TASK_SERIALIZER = 'json'

taskflow_api/celery.py

@@ -1,6 +1,8 @@
 from django.apps import AppConfig
 
-
 class TasksConfig(AppConfig):
     default_auto_field = 'django.db.models.BigAutoField'
     name = 'tasks'
+
+    def ready(self):
+        import tasks.signals

taskflow_api/settings.py

@@ -0,0 +1,11 @@
+from django.db.models.signals import post_save
+from django.dispatch import receiver
+from tasks.models import Task
+from tasks.tasks import log_task_action
+
+
+@receiver(post_save, sender=Task)
+def task_created_handler(sender, instance, created, **kwargs):
+
+    if created:
+        log_task_action.delay(instance.id, "created")

tasks/apps.py

@@ -0,0 +1,15 @@
+from celery import shared_task
+from django.utils.timezone import now
+from tasks.models import Task
+
+@shared_task
+def log_task_action(task_id, action):
+    """Log task ID and title to a file."""
+    try:
+        task = Task.objects.get(id=task_id)
+        log_line = f"[{now()}] Logging Celery Task : Task #{task.id} ('{task.title}') was {action} using celery.\n"
+    except Task.DoesNotExist:
+        log_line = f"[{now()}] ERROR: Task {task_id} not found for action '{action}'\n"
+
+    with open("logs/task_activity.log", "a") as f:
+        f.write(log_line)