115 lines
3 KiB
Python
115 lines
3 KiB
Python
import csv
|
|
import re
|
|
|
|
from django.conf import settings
|
|
from django.contrib.auth.models import User
|
|
|
|
from judge.models import Profile, Language, Organization
|
|
|
|
|
|
fields = ["username", "password", "name", "school", "email", "organizations"]
|
|
descriptions = [
|
|
"my_username(edit old one if exist)",
|
|
"123456 (must have)",
|
|
"Le Van A (can be empty)",
|
|
"Le Quy Don (can be empty)",
|
|
"email@email.com (can be empty)",
|
|
"org1&org2&org3&... (can be empty - org slug in URL)",
|
|
]
|
|
|
|
|
|
def csv_to_dict(csv_file):
|
|
rows = csv.reader(csv_file.read().decode().split("\n"))
|
|
header = next(rows)
|
|
header = [i.lower() for i in header]
|
|
|
|
if "username" not in header:
|
|
return []
|
|
|
|
res = []
|
|
|
|
for row in rows:
|
|
if len(row) != len(header):
|
|
continue
|
|
cur_dict = {i: "" for i in fields}
|
|
for i in range(len(header)):
|
|
if header[i] not in fields:
|
|
continue
|
|
cur_dict[header[i]] = row[i]
|
|
if cur_dict["username"]:
|
|
res.append(cur_dict)
|
|
return res
|
|
|
|
|
|
def is_valid_username(username):
|
|
match = re.match(r"\w+", username)
|
|
return match is not None and match.group() == username
|
|
|
|
|
|
# return result log
|
|
def import_users(users):
|
|
log = ""
|
|
for i, row in enumerate(users):
|
|
cur_log = str(i + 1) + ". "
|
|
|
|
username = row["username"]
|
|
if not is_valid_username(username):
|
|
log += username + ": Invalid username\n"
|
|
continue
|
|
|
|
cur_log += username + ": "
|
|
pwd = row["password"]
|
|
user, created = User.objects.get_or_create(
|
|
username=username,
|
|
defaults={
|
|
"is_active": True,
|
|
},
|
|
)
|
|
profile, _ = Profile.objects.get_or_create(
|
|
user=user,
|
|
defaults={
|
|
"language": Language.get_python3(),
|
|
"timezone": settings.DEFAULT_USER_TIME_ZONE,
|
|
},
|
|
)
|
|
|
|
if created:
|
|
cur_log += "Create new - "
|
|
else:
|
|
cur_log += "Edit - "
|
|
|
|
if pwd:
|
|
user.set_password(pwd)
|
|
elif created:
|
|
user.set_password("lqdoj")
|
|
cur_log += "Missing password, set password = lqdoj - "
|
|
|
|
if "name" in row.keys() and row["name"]:
|
|
user.first_name = row["name"]
|
|
|
|
if "school" in row.keys() and row["school"]:
|
|
user.last_name = row["school"]
|
|
|
|
if row["organizations"]:
|
|
orgs = row["organizations"].split("&")
|
|
added_orgs = []
|
|
for o in orgs:
|
|
try:
|
|
org = Organization.objects.get(slug=o)
|
|
profile.organizations.add(org)
|
|
added_orgs.append(org.name)
|
|
except Organization.DoesNotExist:
|
|
continue
|
|
if added_orgs:
|
|
cur_log += "Added to " + ", ".join(added_orgs) + " - "
|
|
|
|
if row["email"]:
|
|
user.email = row["email"]
|
|
|
|
user.save()
|
|
profile.save()
|
|
cur_log += "Saved\n"
|
|
log += cur_log
|
|
log += "FINISH"
|
|
|
|
return log
|