first

apps/ticket/services.py

@@ -0,0 +1,85 @@
+from apps.ticket.models import Ticket, TicketComment, TicketCommentFile
+from apps.cuser.models import CuserRole
+from apps.workspace.models import Workspace
+from apps.cuser.models import CuserRole as CuserRoleModel
+from django.db import transaction
+from rest_framework import exceptions
+import base64
+from abc import ABC, abstractmethod
+
+
+def create_ticket(title: str, desc: str, cuser_role: CuserRole, workspace: Workspace, files=None) -> Ticket:
+
+    with transaction.atomic():
+
+        try:
+
+            ticket = Ticket()
+            ticket.title = title
+            ticket.cuser_role = cuser_role
+            ticket.status = 0
+            ticket.sys_ws = workspace
+            ticket.save()
+
+            ticket_comment = TicketComment()
+            ticket_comment.ticket = ticket
+            ticket_comment.desc = desc
+            ticket_comment.cuser_role = cuser_role
+            ticket_comment.sys_ws = workspace
+
+            ticket_comment.save()
+
+            if files:
+                for file in files:
+                    ticket_file = TicketCommentFile()
+                    ticket_file.ticket_comment = ticket_comment
+                    ticket_file.file = file
+                    ticket_file.save()
+
+            return ticket
+
+        except Exception as exp:
+            transaction.set_rollback(True)
+            message = f'Error Create Ticket: {exp}'
+            raise exceptions.ValidationError({'error': message})
+
+
+def create_comment_ticket(ticket_id: int, desc: str, cuser_role: CuserRole, workspace: Workspace, files=None) -> TicketComment:
+
+    with transaction.atomic():
+
+        try:
+
+            role_type_name = cuser_role.role_type.role_name_en
+            ticket_status = None
+            if role_type_name == 'workspace_operator':
+                ticket = Ticket.objects.get(id=ticket_id)
+                if ticket.status == 0:
+                    ticket.status = 2
+                    ticket.save()
+            else:
+                ticket = Ticket.objects.get(id=ticket_id, cuser_role=cuser_role, sys_ws=workspace)
+                if ticket.status == 0:
+                    ticket.status = 1
+                    ticket.save()
+
+            ticket_comment = TicketComment()
+            ticket_comment.ticket = ticket
+            ticket_comment.desc = desc
+            ticket_comment.cuser_role = cuser_role
+            ticket_comment.sys_ws = workspace
+            ticket_comment.save()
+
+            if files:
+                for file in files:
+                    ticket_file = TicketCommentFile()
+                    ticket_file.ticket_comment = ticket_comment
+                    ticket_file.file = file
+                    ticket_file.save()
+
+            return ticket_comment
+
+        except Exception as exp:
+            transaction.set_rollback(True)
+            message = f'Error Create TicketComment: {exp}'
+            raise exceptions.ValidationError({'error': message})