QingdaoU
/
OnlineJudge
mirror of https://github.com/QingdaoU/OnlineJudge.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Add] Import real_name when Import Users

This commit is contained in:
Harry-zklcdc 2020-08-02 14:59:40 +08:00 committed by virusdefender
parent 856c3e52b6
commit cfb709f128
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
account/views/admin.py
View File

@ -28,14 +28,14 @@ class UserAdminAPI(APIView):
user_list = []
for user_data in data:
if len(user_data) != 3 or len(user_data[0]) > 32:
if len(user_data) != 4 or len(user_data[0]) > 32:
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])
UserProfile.objects.bulk_create([UserProfile(user=ret[i], real_name=data[i][3]) for i in range(len(ret))])
return self.success()
except IntegrityError as e:
# Extract detail from exception message