def login(user_id, pin_number, user_role): if user_role == ROLES.get("admin"): administrator = Administrator.get_or_none(Administrator.user_id == user_id) if administrator and (administrator.pin_number == pin_number): session['user_id'] = administrator.id return True elif user_role == ROLES.get("moderator"): moderator = Moderator.get_or_none(Moderator.user_id == user_id) if moderator and (moderator.pin_number == pin_number): session['user_id'] = moderator.id return True elif user_role == ROLES.get("faculty"): faculty = Faculty.get_or_none(Faculty.user_id == user_id) if faculty and(faculty.pin_number == pin_number): session['user_id'] = faculty.id return True else: return False