first

apps/reminder/services.py

@@ -0,0 +1,48 @@
+from apps.reminder.models import Reminder as ReminderModel
+from apps.lead.models import Lead as LeadModel
+from rest_framework import exceptions
+from utils.get_user_workspace import get_user_workspace
+from utils.get_cuser_role import get_cuser_role
+
+
+def create_or_edit_reminder_lead(request, lead_id: int, datetime: str, desc: str = None) -> ReminderModel:
+    try:
+        workspace = get_user_workspace(request)
+        cuser_role = get_cuser_role(request)
+
+        lead = LeadModel.objects.get(id=lead_id)
+        customer = lead.customer
+
+        try:
+            reminder = ReminderModel.objects.get(type='LE', lead=lead, customer=customer, sys_ws=workspace, cuser_role=cuser_role)
+        except Exception as exp:
+            reminder = ReminderModel(lead=lead)
+
+        reminder.type = 'LE'
+        reminder.customer = customer
+        reminder.datetime = datetime
+        reminder.sys_ws = workspace
+        reminder.cuser_role = cuser_role
+        if desc:
+            reminder.desc = desc
+        reminder.save()
+
+        return reminder
+
+    except Exception as exp:
+        message = f'Error Creating or Update reminder for lead: {exp}'
+        raise exceptions.ValidationError({'error': message})
+
+
+def delete_reminder_lead(request, lead_id: int) -> ReminderModel:
+    try:
+        workspace = get_user_workspace(request)
+        cuser_role = get_cuser_role(request)
+
+        reminder = ReminderModel.objects.get(type='LE', lead_id=lead_id, sys_ws=workspace, cuser_role=cuser_role).delete()
+
+        return reminder
+
+    except Exception as exp:
+        message = f'Error Delete reminder for lead: {exp}'
+        raise exceptions.ValidationError({'error': message})