diff options
Diffstat (limited to 'taskapp/events/user.py')
-rw-r--r-- | taskapp/events/user.py | 19 |
1 files changed, 12 insertions, 7 deletions
diff --git a/taskapp/events/user.py b/taskapp/events/user.py index 4447643..e2ac30a 100644 --- a/taskapp/events/user.py +++ b/taskapp/events/user.py @@ -26,13 +26,17 @@ def createUser(username,email,password,dob,gender): gender : u'M'/u'F' """ - user = User(username=username, email=email) - user.set_password(password) - user.save() - properties = {'dob':dob, 'gender':gender} - user_profile = Profile(user=user) - updateProfile(user_profile, properties) - return user + try: + user = User.objects.get(username=username) + return user + except: + user = User(username=username, email=email) + user.set_password(password) + user.save() + properties = {'dob':dob, 'gender':gender} + user_profile = Profile(user=user) + updateProfile(user_profile, properties) + return user def createSuUser(username,email,password,dob,gender): """ create user using createUser method and set the is_superuser flag """ @@ -41,3 +45,4 @@ def createSuUser(username,email,password,dob,gender): su_user.is_staff = True su_user.is_superuser = True su_user.save() + return su_user |