feat: add scheduled task to clean users documents folder at 00:00

makedoc/tasks.py

@@ -0,0 +1,22 @@
+import os
+import shutil
+
+from celery import shared_task
+from django.conf import settings
+
+
+@shared_task
+def delete_files():
+    directory = settings.BASE_DIR / "makedoc" / "tempdoc"
+
+    if not os.path.exists(directory):
+        return "Directory does not exist"
+    for item in os.listdir(directory):
+        item_path = os.path.join(directory, item)
+        try:
+            if os.path.isdir(item_path):
+                shutil.rmtree(item_path)
+        except Exception:
+            return f"Failed to delete {item_path}. Reason {e}"
+
+    return "Successfully deleted all user folders"

melnichanka/settings.py

@@ -14,6 +14,7 @@
 from datetime import timedelta
 from pathlib import Path
 
+from celery.schedules import crontab
 from dotenv import load_dotenv
 
 load_dotenv()
@@ -215,3 +216,12 @@
     "VERSION": "1.0.0",
     "SERVE_INCLUDE_SCHEMA": False,
 }
+
+# Celery
+
+CELERY_BEAT_SCHEDULE = {
+    "delete_files_daily": {
+        "task": "makedoc.tasks.delete_files",
+        "schedule": crontab(hour=23, minute=59),
+    },
+}