add generate user api

account/views/admin.py

@@ -1,12 +1,16 @@
+import os
+import re
+import xlsxwriter
 from django.db.models import Q
+from django.http import HttpResponse
 
 from submission.models import Submission
 from utils.api import APIView, validate_serializer
 from utils.shortcuts import rand_str
 
 from ..decorators import super_admin_required
-from ..models import AdminType, ProblemPermission, User
-from ..serializers import EditUserSerializer, UserSerializer
+from ..models import AdminType, ProblemPermission, User, UserProfile
+from ..serializers import EditUserSerializer, UserSerializer, GenerateUserSerializer
 
 
 class UserAdminAPI(APIView):
@@ -85,3 +89,82 @@ class UserAdminAPI(APIView):
                                Q(userprofile__real_name__icontains=keyword) |
                                Q(email__icontains=keyword))
         return self.success(self.paginate_data(request, user, UserSerializer))
+
+    def delete_one(self, user_id):
+        try:
+            user = User.objects.get(id=user_id)
+        except User.DoesNotExist:
+            return f"User {user_id} does not exist"
+        profile = user.userprofile
+        if profile.submission_number:
+            return f"Can't delete the user {user_id} as he/she has submissions"
+        user.delete()
+
+    @super_admin_required
+    def delete(self, request):
+        id = request.GET.get("id")
+        if not id:
+            return self.error("Invalid Parameter, user_id is required")
+        for user_id in id.split(","):
+            if user_id:
+                error = self.delete_one(user_id)
+                if error:
+                    return self.error(error)
+        return self.success()
+
+
+class GenerateUserAPI(APIView):
+    @super_admin_required
+    def get(self, request):
+        """
+        download users excel
+        """
+        file_id = request.GET.get("file_id")
+        if not file_id:
+            return self.error("Invalid Parameter, file_id is required")
+        if not re.match(r"[a-zA-Z0-9]+", file_id):
+            return self.error("Illegal file_id")
+        file_path = f"/tmp/{file_id}.xlsx"
+        if not os.path.isfile(file_path):
+            return self.error("File does not exist")
+        with open(file_path, "rb") as f:
+            raw_data = f.read()
+        os.remove(file_path)
+        response = HttpResponse(raw_data)
+        response["Content-Disposition"] = f"attachment; filename=users.xlsx"
+        response["Content-Type"] = "application/xlsx"
+        return response
+
+    @validate_serializer(GenerateUserSerializer)
+    @super_admin_required
+    def post(self, request):
+        data = request.data
+        number_max_length = max(len(str(data["number_from"])), len(str(data["number_to"])))
+        if number_max_length + len(data["prefix"]) + len(data["suffix"]) > 32:
+            return self.error("Username should not more than 32 characters")
+        if data["number_from"] >= data["number_to"]:
+            return self.error("Start number must be lower than end number")
+
+        password_length = data.get("password_length", 8)
+        default_email = data.get("default_email")
+
+        file_id = rand_str(8)
+        filename = f"/tmp/{file_id}.xlsx"
+        workbook = xlsxwriter.Workbook(filename)
+        worksheet = workbook.add_worksheet()
+        worksheet.set_column("A:B", 20)
+        worksheet.write("A1", "Username")
+        worksheet.write("B1", "Password")
+        i = 1
+        for number in range(data["number_from"], data["number_to"] + 1):
+            username = f"{data['prefix']}{number}{data['suffix']}"
+            password = rand_str(password_length)
+            user = User.objects.create(username=username, email=default_email)
+            user.set_password(password)
+            user.save()
+            UserProfile.objects.create(user=user)
+            worksheet.write_string(i, 0, username)
+            worksheet.write_string(i, 1, password)
+            i += 1
+        workbook.close()
+        return self.success(file_id)