server.py

from flask import Flask, request, render_template, session, redirect, url_for, escape, request
import data_handler
import data_handler_2

app = Flask(__name__)


@app.route('/')
def index():
    username = session['username'] if 'username' in session else 'Anonymus'
    if username != 'Anonymus':
        image = data_handler_2.get_users_image(username)['image']
        id = data_handler_2.get_id_by_username(username)['id']
        return render_template("main_page.html", username=username, image=image, id=id)
    else:
        return render_template("main_page.html", username=username)


@app.route('/table')
def table():
    username = session['username'] if 'username' in session else 'Anonymus'
    info = data_handler.get_all_data_for_lapozo('question', 0)
    how_many_question = len(data_handler.get_all_data('question'))
    pages = int(how_many_question / 5)
    return render_template("table.html", info=info, username=username, pages=pages)


@app.route('/table/<int:number>')
def table__for_sites(number):
    offset_number = ((number -1) * 10)
    how_many_question = len(data_handler.get_all_data('question'))
    pages = int(how_many_question / 10)
    if (how_many_question % 10) != 0:
        pages = pages + 1
    username = session['username'] if 'username' in session else 'Anonymus'
    info = data_handler.get_all_data_for_lapozo('question', offset_number)
    return render_template("table.html", info=info, username=username, first_pages=offset_number, pages=pages)


@app.route('/question/add_new_question', methods=['GET', 'POST'])
def add_new_question():
    if request.method == 'GET':
        tags = data_handler.get_all_tags()
        return render_template("new_question.html", tags=tags)
    else:
        title = request.form["title"]
        message = request.form["message"]
        image = request.form["image"]
        tag = request.form["tag"]
        username = session['username']
        data_handler.add_new_question(title, message, image, tag, username)
        return redirect("/")


@app.route('/question/<int:question_id>', methods=["GET", "POST"])
def question_details(question_id):
    info = data_handler.get_question_by_id(question_id)
    info_tag = data_handler.get_tag(question_id)
    profile_question = data_handler_2.get_profile_by_question_id(question_id)
    question_data = info[0]['title']
    detail_data = info[0]['message']
    picture_data = info[0]['image']
    view_number = info[0]['view_number']
    popular_number = info[0]['vote_number']
    tag = info_tag['name']
    username = session['username'] if 'username' in session else 'Anonymus'
    answers = data_handler.get_answers_by_question_id(question_id)
    if request.method == 'GET':
        view_number += 1
        info[0]['view_number'] = view_number
        data_handler.update_view_number(view_number, question_id)
    return render_template("question.html", question_id=question_id, question_data=question_data,
                           detail_data=detail_data, profile_question=profile_question,
                           picture_data=picture_data, view_number=view_number, popular_number=popular_number,
                           tag=tag, answers=answers, username=username)


@app.route('/like/<int:question_id>', methods=["GET"])
def popularity(question_id):
    question = data_handler.get_question_by_id(question_id)
    popular_number = question[0]['vote_number']
    view_number = question[0]['view_number']
    if request.method == 'GET':
        popular_number += 1
        view_number -= 1
        data_handler.update_question(question_id, view_number, popular_number)
    return redirect(url_for('question_details', question_id=question_id))


@app.route('/dislike/<int:question_id>', methods=["GET"])
def not_so_popular(question_id):
    question = data_handler.get_question_by_id(question_id)
    popular_number = question[0]['vote_number']
    view_number = question[0]['view_number']
    if request.method == 'GET':
        popular_number += -1
        view_number -= 1
        data_handler.update_question(question_id, view_number, popular_number)
    return redirect(url_for('question_details', question_id=question_id))


@app.route("/question/add_new_answer/<int:question_id>", methods=['GET', 'POST'])
def add_new_answer(question_id):
    answer = request.form["answer"]
    username = session['username']
    data_handler.add_new_answer(answer, question_id, username)
    return redirect(url_for('question_details', question_id=question_id))


@app.route('/answer_like/<int:question_id>/<int:answer_id>', methods=["GET", "POST"])
def answer_more_popularity(question_id, answer_id):
    answer = data_handler.get_answer_by_id(answer_id)
    answer_popular_number = answer['vote_number']
    question = data_handler.get_question_by_id(question_id)
    view_number = question[0]['view_number']
    popular_number = question[0]['vote_number']
    if request.method == 'GET':
        answer_popular_number += 1
        data_handler.update_answer_vote_number(answer_id, answer_popular_number)
        view_number -= 1
        data_handler.update_question(question_id, view_number, popular_number)
    return redirect(
        url_for('question_details', question_id=question_id))


@app.route('/answer_dislike/<int:question_id>/<int:answer_id>', methods=["GET", "POST"])
def answer_less_popularity(question_id, answer_id):
    answer = data_handler.get_answer_by_id(answer_id)
    answer_popular_number = answer['vote_number']
    question = data_handler.get_question_by_id(question_id)
    view_number = question[0]['view_number']
    popular_number = question[0]['vote_number']
    if request.method == 'GET':
        answer_popular_number -= 1
        data_handler.update_answer_vote_number(answer_id, answer_popular_number)
        view_number -= 1
        data_handler.update_question(question_id, view_number, popular_number)
    return redirect(
        url_for('question_details', question_id=question_id))


@app.route('/answer_edit/<int:question_id>/<int:answer_id>', methods=["GET", "POST"])
def answer_edit(question_id, answer_id):
    if request.method == 'GET':
        answer = data_handler.get_answer_by_id(answer_id)
        return render_template("answer_edit.html", answer=answer)

    message = request.form['message']
    data_handler.update_answer(answer_id, message)
    return redirect(url_for('question_details', question_id=question_id))


@app.route('/comment/<int:question_id>', methods=["GET", "POST"])
def question_comment(question_id):
    if request.method == 'GET':
        comments = data_handler.get_all_comments(question_id)
        return render_template("comment.html", question_id=question_id, comments=comments)
    else:
        comment = request.form["comment"]
        username = session['username']
        data_handler.add_new_comment(comment, question_id, username)
        comments = data_handler.get_all_comments(question_id)
        return render_template("comment.html", question_id=question_id, comments=comments)


@app.route('/comment_answer/<int:answer_id>', methods=["GET", "POST"])
def answer_comment(answer_id):
    if request.method == 'GET':
        comments = data_handler.get_all_comments_answer(answer_id)
        question_id = data_handler.get_question_id_by_answer_id(answer_id)
        return render_template("comment_answer.html", answer_id=answer_id, comments=comments, question_id=question_id)
    else:
        comment_answer = request.form["comment"]
        username = session['username']
        data_handler.add_new_comment_answer(comment_answer, answer_id, username)
        comments = data_handler.get_all_comments_answer(answer_id)
        question_id = data_handler.get_question_id_by_answer_id(answer_id)
        return render_template("comment_answer.html", answer_id=answer_id, comments=comments, question_id=question_id)


@app.route('/search_results', methods=['GET', 'POST'])
def find_search_results():
    form_data = request.form.to_dict()
    search_phrase = form_data['search_phrase']
    if search_phrase == '':
        search_phrase = ' '

    return redirect(url_for('show_search_results', search_phrase=search_phrase))


@app.route('/search?q=<search_phrase>')
def show_search_results(search_phrase):
    results = data_handler.get_search_results(search_phrase)
    return render_template('search_results.html', results=results)


@app.route('/delete-tag/<int:question_id>')
def delete_tag_from_question(question_id):
    data_handler.delete_tag_from_question(question_id)
    return redirect(url_for('question_details', question_id=question_id))


@app.route('/delete_comment/<int:question_id>/<int:comment_id>')
def delete_question_comment(question_id, comment_id):
    data_handler.delete_line(comment_id)
    comments = data_handler.get_all_comments(question_id)

    return render_template('comment.html', question_id=question_id, comments=comments)


@app.route('/delete_comment_answer/<int:answer_id>/<int:comment_id>')
def delete_answer_comment(answer_id, comment_id):
    data_handler.delete_line(comment_id)
    comments = data_handler.get_all_comments_answer(answer_id)
    question_id = data_handler.get_question_id_by_answer_id(answer_id)
    return render_template('comment_answer.html', answer_id=answer_id, question_id=question_id, comments=comments)


@app.route('/edit_comment_answer/<int:answer_id>/<int:comment_id>', methods=["GET", "POST"])
def edit_answer_comment(answer_id, comment_id):
    comment = data_handler.get_comment_by_id(comment_id)
    if request.method == 'GET':
        return render_template("comment_edit.html", comment=comment, answer_id=answer_id)
    else:
        message = request.form['message']
        comment_id = comment['id']
        data_handler.update_comment(comment_id, message)
        return redirect(url_for('answer_comment', answer_id=answer_id))


@app.route('/tags')
def number_of_tags():
    tags = data_handler.count_tags()
    return render_template("tags.html", tags=tags)


@app.route('/registration', methods=["GET", "POST"])
def registration():
    if request.method == 'GET':
        return render_template("user_registration.html")
    else:
        username = request.form['username']
        every_username = data_handler_2.check_name_in_database(username)

        if every_username is None:
            image = request.form['image']
            password = request.form['password']
            hashed_password = data_handler_2.hash_password(password)
            if len(image) < 5:
                data_handler_2.save_registration_without_image(username, hashed_password)
            else:
                data_handler_2.save_registration(username, hashed_password, image)
            return render_template("log_in.html")
        else:
            return render_template('message.html', message='Username already taken', url=url_for('registration'))


@app.route('/log_in', methods=['GET', 'POST'])
def log_in():
    if request.method == 'GET':
        return render_template("log_in.html")
    else:
        username_check = request.form['username']
        password_check_input = request.form['password']
        username = data_handler_2.check_name_in_database(username_check)

        if username is not None:
            password_check_database = data_handler_2.get_users_password(username_check)['password']
            verify = data_handler_2.verify_password(password_check_input, password_check_database)
            if verify is True:
                username = session['username'] = request.form['username']
                return render_template('message.html', message='Successful log in as {0}'.format(username), url=url_for('index'))
            else:
                return render_template('message.html', message='You wrote wrong username or password',
                                       url=url_for('log_in'))
        else:
            return render_template('message.html', message='You wrote wrong username or password', url=url_for('log_in'))


@app.route('/logout')
def logout():
    if 'username' not in session:
        return render_template('message.html', message='You are not logged in, silly', url=url_for('log_in'))
    else:
        session.pop('username', None)
        return render_template('message.html', message='Check out is successful', url=url_for('index'))

@app.route('/user-page/<int:user_id>')
def user_page(user_id):
    questions = data_handler_2.get_questions_by_user(user_id)
    answers = data_handler_2.get_answers_by_user(user_id)
    question_comments = data_handler_2.get_question_comments_by_user(user_id)
    answer_comments = data_handler_2.get_answer_comments_by_user(user_id)
    return render_template("user_page.html", questions=questions, answers=answers, question_comments=question_comments,
                           answer_comments=answer_comments)


@app.route('/user-list')
def user_list():
    users = data_handler_2.get_all_user_data()
    return render_template('user_list.html', users=users)


if __name__ == '__main__':
    app.secret_key = 'superheroes'
    app.run(debug=True)