def edit_user(userid):
template = os.path.join(templates_folder, "edit.html.mako")
if request.method.lower() == "get":
user = User.query.filter_by(id=userid).first()
user = user.__dict__
user = remove_unwanted_keyvalue(user)
return render(result=user, template=template)
else:
return method_not_supported_message(request.method.lower())
def view_single_task(taskid):
log.info("Getting data for task: %s" % taskid)
template = os.path.join(templates_folder, "show.html.mako")
if request.method.lower() == "get":
taskid = int(taskid)
task = Task.query.filter_by(id=taskid).first()
task = task.__dict__
task = remove_unwanted_keyvalue(task)
return render(result=task, template=template, error_msg=None)
def view_single_task(taskid):
log.info("Getting data for task: %s"%taskid)
template = os.path.join(templates_folder, "show.html.mako")
if request.method.lower() == "get":
taskid = int(taskid)
task = Task.query.filter_by(id=taskid).first()
task = task.__dict__
task = remove_unwanted_keyvalue(task)
return render(result=task, template=template, error_msg=None)
def edit_user(userid):
template = os.path.join(templates_folder, "edit.html.mako")
if request.method.lower() == "get":
user = User.query.filter_by(id=userid).first()
user = user.__dict__
user = remove_unwanted_keyvalue(user)
return render(result=user, template=template)
else:
return method_not_supported_message(request.method.lower())
def view():
template = os.path.join(templates_folder, "index.html.mako")
if request.method.lower() == "get":
users = User.query.all()
all_users = []
for user in users:
user_dict_repr = user.__dict__
user_dict_repr = remove_unwanted_keyvalue(user_dict_repr)
all_users.append(user_dict_repr)
return render(result=all_users, template=template)
elif request.method.lower() == "post":
log.info("Received new user data: %s" % request.form)
# Todo:
# 2. Need to salt password
# 3. Have not implemented photo upload. (Might not be implementing)
user_with_email = User.query.filter_by(
email=request.form['email']).first()
if user_with_email is None:
new_user = User(request.form['first_name'].strip(),
request.form['last_name'].strip(),
request.form['email'].strip(),
request.form['password'].strip())
engine.session.add(new_user)
engine.session.commit()
log.info("New user created.")
return redirect(
url_for('users.view_single_user', userid=new_user.id))
else:
# email is already registered. throw an error
return redirect(
url_for('users.view_single_user', userid="email_registered"))
else:
return method_not_supported_message(request.method.lower())
def view(owner_id):
template = os.path.join(templates_folder, "index.html.mako")
if owner_id:
log.info("Getting all tasks for user %s."%owner_id)
tasks = Task.query.filter_by(owner_id=owner_id).all()
all_tasks = []
for task in tasks:
task_dict_repr = task.__dict__
task_dict_repr = remove_unwanted_keyvalue(task_dict_repr)
all_tasks.append(task_dict_repr)
return render(result=dict(all_tasks=all_tasks), template=template)
else:
log.info("No owner id specified. Returning empty.")
return render(result=[], template=template)
def view(owner_id):
template = os.path.join(templates_folder, "index.html.mako")
if owner_id:
log.info("Getting all tasks for user %s." % owner_id)
tasks = Task.query.filter_by(owner_id=owner_id).all()
all_tasks = []
for task in tasks:
task_dict_repr = task.__dict__
task_dict_repr = remove_unwanted_keyvalue(task_dict_repr)
all_tasks.append(task_dict_repr)
return render(result=dict(all_tasks=all_tasks), template=template)
else:
log.info("No owner id specified. Returning empty.")
return render(result=[], template=template)
def view_single_user(userid):
template = os.path.join(templates_folder, "show.html.mako")
if request.method.lower() == "get":
try:
userid = int(userid)
user = User.query.filter_by(id=userid).first()
user = user.__dict__
user = remove_unwanted_keyvalue(user)
return render(result=user, template=template, error_msg=None)
except ValueError:
if userid == "email_registered":
error_msg = dict(error="Email already registered!")
elif userid == "email_pw_wrong":
error_msg = dict(error="Email/password combination is wrong.")
return render(template=template, error_msg=error_msg)
elif request.method.lower() == "post":
log.info("Received new user data to edit: %s" % request.form)
user = User.query.filter_by(id=userid).first()
for attr in user.__dict__.keys():
if attr != "id" and attr in request.form:
setattr(user, attr, request.form[attr])
engine.session.commit()
log.info("User edited")
return redirect(url_for('users.view_single_user', userid=userid))
else:
return method_not_supported_message(request.method.lower())
def view_single_user(userid):
template = os.path.join(templates_folder, "show.html.mako")
if request.method.lower() == "get":
try:
userid = int(userid)
user = User.query.filter_by(id=userid).first()
user = user.__dict__
user = remove_unwanted_keyvalue(user)
return render(result=user, template=template, error_msg=None)
except ValueError:
if userid == "email_registered":
error_msg = dict(error="Email already registered!")
elif userid == "email_pw_wrong":
error_msg = dict(error="Email/password combination is wrong.")
return render(template=template, error_msg=error_msg)
elif request.method.lower() == "post":
log.info("Received new user data to edit: %s"%request.form)
user = User.query.filter_by(id=userid).first()
for attr in user.__dict__.keys():
if attr != "id" and attr in request.form:
setattr(user, attr, request.form[attr])
engine.session.commit()
log.info("User edited")
return redirect(url_for('users.view_single_user', userid=userid))
else:
return method_not_supported_message(request.method.lower())
def view():
template = os.path.join(templates_folder, "index.html.mako")
if request.method.lower() == "get":
users = User.query.all()
all_users = []
for user in users:
user_dict_repr = user.__dict__
user_dict_repr = remove_unwanted_keyvalue(user_dict_repr)
all_users.append(user_dict_repr)
return render(result=all_users, template=template)
elif request.method.lower() == "post":
log.info("Received new user data: %s"%request.form)
# Todo:
# 2. Need to salt password
# 3. Have not implemented photo upload. (Might not be implementing)
user_with_email = User.query.filter_by(email=request.form['email']).first()
if user_with_email is None:
new_user = User(request.form['first_name'].strip(), request.form['last_name'].strip(),
request.form['email'].strip(), request.form['password'].strip())
engine.session.add(new_user)
engine.session.commit()
log.info("New user created.")
return redirect(url_for('users.view_single_user', userid=new_user.id))
else:
# email is already registered. throw an error
return redirect(url_for('users.view_single_user', userid="email_registered"))
else:
return method_not_supported_message(request.method.lower())
