from django.contrib.auth.models import User
from pytask.taskapp.models import Profile, Task, Comment, Credit

""" A collection of helper methods. note that there is no validation done here.
we take care of validation and others checks in methods that invoke these methods.
"""

def updateProfile(user_profile, properties):
    """ updates the given properties in the profile for a user. 
    args:
        user_profile : a profile object
        properties : a dictionary with attributes to set as keys and corresponding values
    """
    
    for attr,value in properties.items():
        user_profile.__setattr__(attr,value)
    user_profile.save()

def createUser(username,email,password,dob,gender):
    """ create a user and create a profile and update its properties 
    args:
        username : a username that does not exist
        email : a valid email
        password : a password
        dob : a date object
        gender : u'M'/u'F' 
    """

    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 """
    
    su_user = createUser(username,email,password,dob,gender)
    su_user.is_staff = True
    su_user.is_superuser = True
    su_user.save()
    return su_user

def changeRole(role, user):
    """ change the status of user to role.
    """

    user_profile = user.get_profile()
    user_profile.rights = role
    user_profile.save()