use bulk_create and transcation for importing user
This commit is contained in:
@@ -1,8 +1,11 @@
|
||||
import os
|
||||
import re
|
||||
import xlsxwriter
|
||||
|
||||
from django.db import transaction, IntegrityError
|
||||
from django.db.models import Q
|
||||
from django.http import HttpResponse
|
||||
from django.contrib.auth.hashers import make_password
|
||||
|
||||
from submission.models import Submission
|
||||
from utils.api import APIView, validate_serializer
|
||||
@@ -18,26 +21,27 @@ class UserAdminAPI(APIView):
|
||||
@validate_serializer(ImportUserSeralizer)
|
||||
@super_admin_required
|
||||
def post(self, request):
|
||||
"""
|
||||
Generate user
|
||||
"""
|
||||
data = request.data["users"]
|
||||
omitted_count = created_count = get_count = 0
|
||||
|
||||
user_list = []
|
||||
for user_data in data:
|
||||
if len(user_data) != 3 or len(user_data[0]) > 32:
|
||||
omitted_count += 1
|
||||
continue
|
||||
user, created = User.objects.get_or_create(username=user_data[0])
|
||||
user.set_password(user_data[1])
|
||||
user.email = user_data[2]
|
||||
user.save()
|
||||
if created:
|
||||
UserProfile.objects.create(user=user)
|
||||
created_count += 1
|
||||
else:
|
||||
get_count += 1
|
||||
return self.success({
|
||||
"omitted_count": omitted_count,
|
||||
"created_count": created_count,
|
||||
"get_count": get_count
|
||||
})
|
||||
return self.error(f"Error occurred while processing data '{user_data}'")
|
||||
user_list.append(User(username=user_data[0], password=make_password(user_data[1]), email=user_data[2]))
|
||||
|
||||
try:
|
||||
with transaction.atomic():
|
||||
ret = User.objects.bulk_create(user_list)
|
||||
UserProfile.objects.bulk_create([UserProfile(user=user) for user in ret])
|
||||
return self.success()
|
||||
except IntegrityError as e:
|
||||
# Extract detail from exception message
|
||||
# duplicate key value violates unique constraint "user_username_key"
|
||||
# DETAIL: Key (username)=(root11) already exists.
|
||||
return self.error(str(e).split("\n")[1])
|
||||
|
||||
@validate_serializer(EditUserSerializer)
|
||||
@super_admin_required
|
||||
|
||||
Reference in New Issue
Block a user