Add import users

judge/tasks/experiment.py

@@ -1,10 +1,6 @@
-from django.contrib.auth.models import User
-from django.conf import settings
-
-from judge.models import SubmissionTestCase, Problem, Profile, Language, Organization
+from judge.models import SubmissionTestCase, Problem
 
 from collections import defaultdict
-import csv
 
 def generate_report(problem):
     testcases = SubmissionTestCase.objects.filter(submission__problem=problem).all()
@@ -21,52 +17,4 @@ def generate_report(problem):
         rate[i] = score[i] / total[i]
 
     for i, _ in sorted(rate.items(), key=lambda x: x[1], reverse=True):
-        print(i, score[i], total[i], rate[i])
-
-
-def import_users(csv_file):
-    # 1st row: username, password, name, organization
-    # ... row: a_username, passhere, my_name, organ
-    try:
-        f = open(csv_file, 'r')
-    except OSError:
-        print("Could not open csv file", csv_file)
-        return
-
-    with f:
-        reader = csv.DictReader(f)
-
-        for row in reader:
-            try:
-                username = row['username']
-                pwd = row['password']
-            except Exception:
-                print('username and/or password column missing')
-                print('Make sure your columns are: username, password, name, organization')
-            
-            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:
-                print('Created user', username)
-
-            if 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']:
-                org = Organization.objects.get(name=row['organization'])
-                profile.organizations.add(org)
-            user.email = row['email']
-            user.save()
-            profile.save()
+        print(i, score[i], total[i], rate[i])