server.py

import os
from flask_mail import Mail, Message
from flask import Flask, render_template, request, flash, redirect, url_for, session

import connection

# TODO: change path to absolute for all users

template_dir = os.path.abspath('Static')
STATIC_DIR = os.path.abspath('Static')
app = Flask(__name__, template_folder=template_dir, static_folder=STATIC_DIR)
app.secret_key = b'_5#y2L"F4Q8z\n\xec]/'

mail = Mail(app)  # instantiate the mail class

# configuration of mail
app.config['MAIL_SERVER'] = 'smtp.gmail.com'
app.config['MAIL_PORT'] = 465
app.config['MAIL_USERNAME'] = 'sagarmathacontactform@gmail.com'
app.config['MAIL_PASSWORD'] = 'Ajhb$67dn'
app.config['MAIL_USE_TLS'] = False
app.config['MAIL_USE_SSL'] = True
mail = Mail(app)


@app.route('/')
@app.route('/index.html')
def route_list():
    sections_to_show = []
    all_settings = connection.open_json_settings()
    if all_settings["indexIntroSection"] == 'show':
        sections_to_show.append('indexIntroSection')
    if all_settings["indexTodaySpecialMenu"] == 'show':
        sections_to_show.append('indexTodaySpecialMenu')

    return render_template("index.html", sections_to_show=sections_to_show)


@app.route('/login', methods=["GET", "POST"])
def login():
    if 'username' in session:
        return redirect(url_for('inner'))
    else:
        if request.method == "POST":
            username_to_check = request.form['user_name']
            password_to_check = request.form['password']

            admin_username = 'admin'
            admin_password = '$j&_2!KJDS9qErMf'

            if username_to_check != admin_username or password_to_check != admin_password:
                login_message = 'Username or password incorrect. Please try again.'
                return render_template('login.html', login_message=login_message)
            else:
                session['username'] = admin_username
                return redirect(url_for("inner"))
        if 'username' in session:
            return render_template("login.html", username=session["username"])
        else:
            return render_template("login.html")


@app.route('/logout.html', methods=["GET", "POST"])
def logout():
    if 'username' in session:
        session.pop('username', None)
        return redirect(url_for('inner'))
    else:
        return redirect(url_for('inner'))


@app.route('/contact.html', methods=["GET", "POST"])
def contact():
    if request.method == "POST":
        message = request.form['message']
        sender_email = request.form['email']
        subject = request.form['subject']
        first_name = request.form['fname']
        last_name = request.form['lname']

        msg = Message(
            'Hello',
            sender='sagarmathacontactform@gmail.com',
            recipients=['dzolwlive@gmail.com']
        )
        msg.body = f"Mail from: {first_name} {last_name} \n Sender email: {sender_email} \n Subject: {subject} \n Message: {message}"
        mail.send(msg)
        flash('Dziękujemy za Twoją wiadomość! / Thank You for your message!')
        return render_template("contact.html")
    return render_template("contact.html")


@app.route('/menu.html')
def menu():
    sections_to_show = []
    all_settings = connection.open_json_settings()
    if all_settings["menuLunch"] == 'show':
        sections_to_show.append('menuLunch')
    if all_settings["menuMain"] == 'show':
        sections_to_show.append('menuMain')

    chapters = []

    menu_full = connection.reader_csv("menu_list.csv")

    menu = []
    for position in menu_full:
        item = dict(position)
        menu.append(item)

    chapters = []
    saved_chapters = connection.reader_csv("chapters.csv")
    for row in saved_chapters:
        chapters_dict = dict(row)

    for key, value in chapters_dict.items():
        chapters.append(value)

    chapters_col_1 = chapters[0:5]
    chapters_col_2 = chapters[6:13]
    chapters_col_3 = chapters[13:]

    return render_template("menu.html", menu=menu, chapters_col_1=chapters_col_1, chapters_col_2=chapters_col_2,
                           chapters_col_3=chapters_col_3, sections_to_show=sections_to_show)


@app.route('/about.html')
def about():
    sections_to_show = []
    all_settings = connection.open_json_settings()
    if all_settings["aboutIntroSection"] == 'show':
        sections_to_show.append('aboutIntroSection')
    if all_settings["aboutTestimonialSection"] == 'show':
        sections_to_show.append('aboutTestimonialSection')
    if all_settings["aboutStaffSection"] == 'show':
        sections_to_show.append('aboutStaffSection')

    return render_template("about.html", sections_to_show=sections_to_show)


@app.route('/inner.html', methods=['POST', 'GET'])
def inner():
    if 'username' in session:
        if request.method == 'POST':
            saved_menu = request.form
            titles = ['chapter', 'id', 'eng_name', 'pol_name', 'price']
            form_values = []
            menu_to_csv = []

            for value in saved_menu.values():
                form_values.append(value)

            menu_values_in_row = [form_values[i:i + len(titles)] for i in range(0, len(form_values), len(titles))]

            for row in menu_values_in_row:
                dict_row = dict(zip(titles, row))
                menu_to_csv.append(dict_row)

            connection.writer(menu_to_csv)

            return redirect(url_for('menu'))
        else:
            menu_full = connection.reader_csv("menu_list.csv")

            menu_inner = []
            for position in menu_full:
                item = dict(position)
                menu_inner.append(item)

            chapters = []
            saved_chapters = connection.reader_csv("chapters.csv")
            for row in saved_chapters:
                chapters_dict = dict(row)

            for key, value in chapters_dict.items():
                if value is not None:
                    chapters.append(value)
                else:
                    continue

            return render_template("inner.html", menu=menu_inner, chapters=chapters)
    else:
        return redirect(url_for('login'))


@app.route('/backup', methods=['POST', 'GET'])
def load_backup_menu():
    if request.method == 'POST':
        menu_full = connection.reader_csv("menu_list_backup.csv")

        menu_inner = []
        for position in menu_full:
            item = dict(position)
            menu_inner.append(item)

        chapters = []
        saved_chapters = connection.reader_csv("chapters.csv")
        for row in saved_chapters:
            chapters_dict = dict(row)

        for key, value in chapters_dict.items():
            if value is not None:
                chapters.append(value)
            else:
                continue

        return render_template("inner.html", menu=menu_inner, chapters=chapters)


if __name__ == '__main__':
    app.run(
        host='0.0.0.0',
        port=8000,
        debug=True,
    )