lkva/NDOJ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add name to import_users

Browse source
This commit is contained in:
cuom1999 2021-04-07 14:08:42 -05:00
parent dd725466e5
commit 06318a97e5
1 changed files with 19 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
judge/tasks/experiment.py
View file

@ -25,8 +25,8 @@ def generate_report(problem):
def import_users(csv_file): def import_users(csv_file):
# 1st row: username, password, organization # 1st row: username, password, name, organization
# ... row: a_username,passhere,organ # ... row: a_username, passhere, my_name, organ
try: try:
f = open(csv_file, 'r') f = open(csv_file, 'r')
except OSError: except OSError:
@ -37,10 +37,14 @@ def import_users(csv_file):
reader = csv.DictReader(f) reader = csv.DictReader(f)
for row in reader: for row in reader:
try:
username = row['username'] username = row['username']
pwd = row['password'] pwd = row['password']
except Exception:
print('username and/or password column missing')
print('Make sure your columns are: username, password, name, organization')
user, _ = User.objects.get_or_create(username=username, defaults={ user, created = User.objects.get_or_create(username=username, defaults={
'is_active': True, 'is_active': True,
}) })
@ -48,9 +52,17 @@ def import_users(csv_file):
'language': Language.get_python3(), 'language': Language.get_python3(),
'timezone': settings.DEFAULT_USER_TIME_ZONE, 'timezone': settings.DEFAULT_USER_TIME_ZONE,
}) })
if created:
print('Created user', username)
if pwd: if pwd:
user.set_password(pwd) user.set_password(pwd)
elif created:
user.set_password('lqdoj')
print('User', username, 'missing password, default=lqdoj')
if 'name' in row.keys() and row['name']:
user.first_name = row['name']
if 'organization' in row.keys() and row['organization']: if 'organization' in row.keys() and row['organization']:
org = Organization.objects.get(name=row['organization']) org = Organization.objects.get(name=row['organization'])