From 86356b19989b94484dddd10dcdeb8a6bd4623d4a Mon Sep 17 00:00:00 2001
From: Nikhil <118773738+pablohashescobar@users.noreply.github.com>
Date: Mon, 2 Sep 2024 21:30:12 +0530
Subject: [PATCH] chore: change ownership of workspace and delete user scripts
 (#1012)

---
 .../management/commands/change_ownership.py   | 67 +++++++++++++++++++
 .../ee/management/commands/delete_user.py     | 65 ++++++++++++++++++
 2 files changed, 132 insertions(+)
 create mode 100644 apiserver/plane/ee/management/commands/change_ownership.py
 create mode 100644 apiserver/plane/ee/management/commands/delete_user.py

diff --git a/apiserver/plane/ee/management/commands/change_ownership.py b/apiserver/plane/ee/management/commands/change_ownership.py
new file mode 100644
index 0000000000..0e0a289c1c
--- /dev/null
+++ b/apiserver/plane/ee/management/commands/change_ownership.py
@@ -0,0 +1,67 @@
+# Django imports
+from django.core.management import BaseCommand, CommandError
+
+
+# Module imports
+from plane.db.models import User, Workspace, WorkspaceMember
+
+
+class Command(BaseCommand):
+    help = "Change the ownership of the workspace"
+
+    def add_arguments(self, parser):
+        # Positional argument
+        parser.add_argument(
+            "--email", type=str, help="user email", required=True
+        )
+        parser.add_argument(
+            "--workspace_slug", type=str, help="workspace slug", required=True
+        )
+
+    def handle(self, *args, **options):
+        # get the user email from console
+        email = options.get("email", False)
+        workspace_slug = options.get("workspace_slug", False)
+
+        # raise error if email is not present
+        if not email or not workspace_slug:
+            raise CommandError("Error: Email and workspace slug is required")
+
+        # Fetch the user
+        user = User.objects.filter(email=email).first()
+
+        # Raise error if the user is not present
+        if not user:
+            raise CommandError(f"Error: User with {email} does not exists")
+
+        # Fetch the workspace
+        workspace = Workspace.objects.filter(slug=workspace_slug).first()
+
+        # Raise error if the workspace is not present
+        if not workspace:
+            raise CommandError(
+                f"Error: Workspace with {workspace_slug} does not exists"
+            )
+
+        # Change the ownership of the workspace
+        workspace.owner = user
+        workspace.save(update_fields=["owner"])
+
+        # Add the user to the workspace
+        workspace_member = WorkspaceMember.objects.filter(
+            workspace=workspace, member=user
+        ).first()
+
+        # If the user is not present in the workspace, create the user
+        if not workspace_member:
+            workspace_member = WorkspaceMember.objects.create(
+                workspace=workspace, member=user, role=20
+            )
+        else:
+            workspace_member.role = 20
+            workspace_member.save(update_fields=["role"])
+
+        # Print the success message
+        self.stdout.write(
+            self.style.SUCCESS("Workspace ownership changed successfully")
+        )
diff --git a/apiserver/plane/ee/management/commands/delete_user.py b/apiserver/plane/ee/management/commands/delete_user.py
new file mode 100644
index 0000000000..b708563c98
--- /dev/null
+++ b/apiserver/plane/ee/management/commands/delete_user.py
@@ -0,0 +1,65 @@
+# Django imports
+from django.core.management import BaseCommand, CommandError
+from django.db import connection
+
+# Module imports
+from plane.db.models import User
+
+
+class Command(BaseCommand):
+    help = "Delete the user"
+
+    def add_arguments(self, parser):
+        # Positional argument
+        parser.add_argument(
+            "--email", type=str, help="user email", required=True
+        )
+
+    def handle_lingering_constraints(self, user):
+        """Delete the related objects from the blacklistedtoken and outstandingtoken tables"""
+        with connection.cursor() as cursor:
+            # First, delete entries from the blacklistedtoken table
+            cursor.execute(
+                """
+                DELETE FROM token_blacklist_blacklistedtoken
+                WHERE token_id IN (
+                    SELECT id FROM token_blacklist_outstandingtoken
+                    WHERE user_id = %s
+                )
+            """,
+                [user.id],
+            )
+
+            # Then, delete entries from the outstandingtoken table
+            cursor.execute(
+                """
+                DELETE FROM token_blacklist_outstandingtoken
+                WHERE user_id = %s
+            """,
+                [user.id],
+            )
+
+    def handle(self, *args, **options):
+        # get the user email from console
+        email = options.get("email", False)
+
+        # raise error if email is not present
+        if not email:
+            raise CommandError("Error: Email is required")
+
+        # Fetch the user
+        user = User.objects.filter(email=email).first()
+
+        # Raise error if the user is not present
+        if not user:
+            raise CommandError(f"Error: User with {email} does not exists")
+
+        # Delete the related objects
+        self.handle_lingering_constraints(user)
+
+        # Delete the user
+        user.delete()
+
+        # Print the success message
+        self.stdout.write(self.style.SUCCESS("User deleted successfully"))
+        return