From 53e8a4fbdbcd09395036dc686ee0bb8beccede58 Mon Sep 17 00:00:00 2001
From: Suryansh5545 <34577232+Suryansh5545@users.noreply.github.com>
Date: Fri, 11 Aug 2023 18:49:23 +0530
Subject: [PATCH] add check for debug mode in celery tasks

---
 apps/base/tasks.py | 42 +++++++++++++++++++++++-------------------
 settings/dev.py    |  8 --------
 2 files changed, 23 insertions(+), 27 deletions(-)

diff --git a/apps/base/tasks.py b/apps/base/tasks.py
index 5a0f34e..e0bdd86 100644
--- a/apps/base/tasks.py
+++ b/apps/base/tasks.py
@@ -6,29 +6,33 @@
 from storages.backends.sftpstorage import SFTPStorage
 from django.core.files import File
 from django.core.files.base import ContentFile
+from django.conf import settings
 
 @shared_task
 def export_all_data():
-    ticket_data = TicketResource().export()
-    transaction_data = TransactionResource().export()
-    check_in_data = CheckInResource().export()
-    storage = SFTPStorage()
+    if settings.DEBUG:
+        print("Exporting data to SFTP server task")
+    else:
+        ticket_data = TicketResource().export()
+        transaction_data = TransactionResource().export()
+        check_in_data = CheckInResource().export()
+        storage = SFTPStorage()
 
-    # Define the remote file path on the SFTP server
-    ticket_data_path = 'ticket_data.csv'
-    transaction_data_path = 'transaction_data.csv'
-    check_in_data_path = 'check_in_data.csv'
+        # Define the remote file path on the SFTP server
+        ticket_data_path = 'ticket_data.csv'
+        transaction_data_path = 'transaction_data.csv'
+        check_in_data_path = 'check_in_data.csv'
 
-    # Save the CSV data to SFTP storage
-    ticket_csv = ContentFile(ticket_data.csv)
-    transaction_csv = ContentFile(transaction_data.csv)
-    check_in_csv = ContentFile(check_in_data.csv)
-    if ((storage.exists(ticket_data_path)) or (storage.exists(transaction_data_path)) or (storage.exists(check_in_data_path))):
-        storage.delete(ticket_data_path)
-        storage.delete(transaction_data_path)
-        storage.delete(check_in_data_path)
+        # Save the CSV data to SFTP storage
+        ticket_csv = ContentFile(ticket_data.csv)
+        transaction_csv = ContentFile(transaction_data.csv)
+        check_in_csv = ContentFile(check_in_data.csv)
+        if ((storage.exists(ticket_data_path)) or (storage.exists(transaction_data_path)) or (storage.exists(check_in_data_path))):
+            storage.delete(ticket_data_path)
+            storage.delete(transaction_data_path)
+            storage.delete(check_in_data_path)
 
-    storage.save(ticket_data_path, ticket_csv)
-    storage.save(transaction_data_path, transaction_csv)
-    storage.save(check_in_data_path, check_in_csv)
+        storage.save(ticket_data_path, ticket_csv)
+        storage.save(transaction_data_path, transaction_csv)
+        storage.save(check_in_data_path, check_in_csv)
     
\ No newline at end of file
diff --git a/settings/dev.py b/settings/dev.py
index d50491b..aec3ff1 100644
--- a/settings/dev.py
+++ b/settings/dev.py
@@ -15,12 +15,4 @@
     }
 }
 
-SFTP_STORAGE_HOST = getenv("SFTP_STORAGE_HOST", "sftp_storage_host")
-SFTP_STORAGE_ROOT = getenv("SFTP_STORAGE_ROOT", "sftp_storage_root")
-SFTP_STORAGE_PARAMS = {
-    'port': getenv("SFTP_STORAGE_PORT", "sftp_storage_port"),
-    'username': getenv("SFTP_STORAGE_USERNAME", "sftp_storage_username"),
-    'password': getenv("SFTP_STORAGE_PASSWORD", "sftp_storage_password"),
-}
-
 EMAIL_BACKEND = "django.core.mail.backends.console.EmailBackend"