first commit

src-backend/custom_user/management/commands/promotetostaff.py

@@ -0,0 +1,27 @@
+from django.core.management import BaseCommand
+from django.contrib.auth import get_user_model
+
+class Command(BaseCommand):
+    """Add additional test data"""
+
+    # Show this when the user types help
+    help = "Give user staff status"
+
+    def add_arguments(self, parser):
+        parser.add_argument("email", nargs="?", type=str, help="Email of the user to promote to staff")
+
+    def handle(self, *args, **options):
+        """Actual Commandline executed function when manage.py command is called"""
+        User = get_user_model()
+        email = options.get("email")
+
+        if not email:
+            email = input("Enter the email of the user to promote to staff: ")
+
+        try:
+            user = User.objects.get(email=email)
+            user.is_staff = True
+            user.save()
+            self.stdout.write(self.style.SUCCESS(f"User {email} is now staff."))
+        except User.DoesNotExist:
+            self.stdout.write(self.style.ERROR(f"No user found with email {email}"))