def login():
username = request.form['username'].lower()
password = request.form['password']
if username in user_data_handler.get_usernames() and hash.verify_pass(
password,
user_data_handler.get_hashed_pass(username)['password']):
session['username'] = username
user_id = user_data_handler.get_userid_by_username(username)
user_id = user_id['id']
session['user_id'] = user_id
return redirect('/')
return redirect('/')
def show_answer_comments(question_id,
answer_id,
comment_id=None,
comment_message=''):
user_id = 0
if 'username' in session:
session_name = f"You are logged in as {escape(session['username'])}"
user_name = escape(session['username'])
user_id = int(escape(session['user_id']))
user_role = user_data_handler.get_user_role(user_id)['role']
else:
session_name = 'You are not logged in'
user_name = None
user_role = None
if request.method == "POST":
user_id = user_data_handler.get_userid_by_username(user_name)
user_id = user_id['id']
add_dict = {
'answer_id': answer_id,
'user_id': user_id,
'comment': request.form['comment']
}
sql_handler.add_answer_comment(add_dict)
if comment_id is not None:
where_url = url_for('edit_answer_comment',
question_id=question_id,
answer_id=answer_id,
comment_id=comment_id)
else:
where_url = url_for('show_answer_comments',
question_id=question_id,
answer_id=answer_id)
comments = sql_handler.get_answer_comments(answer_id)
answer = sql_handler.get_answer_by_answer_id(answer_id)
return render_template('comments.html',
comments=comments,
answer=answer['message'],
question_id=question_id,
back_url=url_for("show_question_details",
id_=question_id),
comment_id=comment_id,
comment_to_edit=comment_message,
where_url=where_url,
logged_in_as=session_name,
user_name=user_name,
user_id=user_id,
user_role=user_role)
def route_add():
if 'username' in session:
session_name = f"You are logged in as {escape(session['username'])}"
user_name = escape(session['username'])
user_id = int(escape(session['user_id']))
user_role = user_data_handler.get_user_role(user_id)['role']
else:
return redirect('/')
# When finished adding
if request.method == "POST":
if 'username' in session:
username = escape(session['username'])
user_id = user_data_handler.get_userid_by_username(username)
user_id = user_id['id']
row = {
'user_id': str(user_id),
'title': str(request.form["title"]),
'message': str(request.form["message"])
}
sql_handler.add_new_question(row)
return redirect("/")
else:
user_name = None
user_id = int(escape(session['user_id']))
session_name = 'You are not logged in'
dir_ = request.args.get("dir_")
id_title = sql_handler.sort_questions(dir_)
error_message = 'You have to log in first!'
return render_template("list.html",
id_title=id_title,
logged_in_as=session_name,
user_name=user_name,
error_message=error_message,
user_id=user_id,
user_role=user_role)
# When someone clicks on add new question button
return render_template("add_edit.html",
id_=None,
title='Add new question',
logged_in_as=session_name,
user_name=user_name,
user_role=user_role)
def show_question_comments(id_, comment_id=None, comment_message=''):
user_id = 0
if 'username' in session:
session_name = f"You are logged in as {escape(session['username'])}"
user_name = escape(session['username'])
user_id = int(escape(session['user_id']))
user_role = user_data_handler.get_user_role(user_id)['role']
else:
session_name = 'You are not logged in'
user_name = None
user_role = None
if request.method == 'POST':
user_id = user_data_handler.get_userid_by_username(user_name)
user_id = user_id['id']
add_dict = {
'question_id': id_,
'user_id': user_id,
'comment': request.form['comment']
}
sql_handler.add_question_comment(add_dict)
if comment_id is not None:
where_url = url_for('edit_question_comment',
comment_id=comment_id,
id_=id_)
else:
where_url = url_for('show_question_comments', id_=id_)
comments = sql_handler.get_question_comments(id_)
question = sql_handler.get_question_details_by_id(id_)
return render_template('comments.html',
comments=comments,
where_url=where_url,
question=question['message'],
question_title=question['title'],
back_url=url_for("show_question_details", id_=id_),
id_=id_,
comment_id=comment_id,
comment_to_edit=comment_message,
logged_in_as=session_name,
user_name=user_name,
user_role=user_role,
user_id=user_id)
def view_profile(user_name):
if 'username' in session:
session_name = f"You are logged in as {escape(session['username'])}"
user_name = escape(session['username'])
user_id = int(escape(session['user_id']))
user_role = user_data_handler.get_user_role(user_id)['role']
else:
session_name = 'You are not logged in'
user_role = None
user = user_name
user_id = user_data_handler.get_userid_by_username(user)['id']
user_questions = user_data_handler.list_questions_by_user_id(user_id)
user_answers_count = user_data_handler.count_answers_by_user_id(user_id)
user_answers = user_data_handler.list_answers_by_user_id(user_id)
user_comments_count_question = user_data_handler.count_comments_on_question_by_user_id(
user_id)
user_comments_on_questions = user_data_handler.list_comments_on_questions_by_user_id(
user_id)
user_comments_count_answer = user_data_handler.count_comments_on_answer_by_user_id(
user_id)
user_comments_on_answers = user_data_handler.list_comments_on_answers_by_user_id(
user_id)
return render_template(
'user_profile.html',
user_name=user_name,
user=user,
user_questions=user_questions,
user_answers=user_answers,
logged_in_as=session_name,
user_answers_count=user_answers_count,
user_id=user_id,
user_role=user_role,
user_comments_count_question=user_comments_count_question,
user_comments_on_questions=user_comments_on_questions,
user_comments_count_answer=user_comments_count_answer,
user_comments_on_answers=user_comments_on_answers)
def show_question_details(id_=None, answer_id=None, answer_message=''):
user_id = 0
if 'username' in session:
session_name = f"You are logged in as {escape(session['username'])}"
user_name = escape(session['username'])
user_id = int(escape(session['user_id']))
user_role = user_data_handler.get_user_role(user_id)['role']
else:
session_name = 'You are not logged in'
user_name = None
user_role = None
question = sql_handler.get_question_details_by_id(id_)
answers = sql_handler.list_answers_by_question_id(id_)
comment_number = sql_handler.count_comments_for_question(id_)
list_of_comment_numbers_on_answers = []
question_username = sql_handler.show_username_at_question(id_)
authors_of_answers = user_data_handler.get_author_of_answer_by_questionid(
id_)
authors_of_answers = [
element['username'] for element in authors_of_answers
]
if answer_id is not None:
where_url = url_for("edit_answer", answer_id=answer_id)
else:
where_url = url_for("show_question_details", id_=id_)
for item in answers:
answer_id = item['id']
num_of_comments = sql_handler.count_comments_for_answer(answer_id)
list_of_comment_numbers_on_answers.append(num_of_comments)
if request.method == "POST": # When you submit an answer
user_id = user_data_handler.get_userid_by_username(user_name)
user_id = user_id['id']
answer = {
'id': id_,
'user_id': user_id,
'message': str(request.form['answer'])
}
sql_handler.add_new_answer(answer)
return redirect(url_for('show_question_details', id_=id_))
return render_template(
"question_details.html",
row_user_id=question['user_id'],
row_title=question['title'],
row_question=question['message'],
answer_list=answers,
where_url=where_url,
comment_number=comment_number,
comment_number_answer=list_of_comment_numbers_on_answers,
id_=id_,
answer_message=answer_message,
logged_in_as=session_name,
user_name=user_name,
user_role=user_role,
user_id=user_id,
question_username=question_username['username'],
authors_of_answers=authors_of_answers)