label_printer/app.py

# SPDX-License-Identifier: GPL-3.0-or-later
from flask import Flask, render_template, request, redirect, url_for, session, send_from_directory
from PIL import Image
from gen_image import gen_image
from config import *
from datetime import *
from print import *
from process_image import *
from file_handler import *
from colorama import Fore, Style
from fonts import fonts
import os

text_image_filename = "text_image.png"
uploaded_image_filename = "uploaded_image.png"

app = Flask(__name__)


app.secret_key = "blahaj"



def render_text_template(info=None, info_color=None, scrollDown=None):
    return render_template('text.html', filename=session["text image path"], cut=session["cut"], info=info, info_color=info_color, scrollDown=scrollDown, fonts=session["fonts"])



def render_image_template(info=None, info_color=None):
    return render_template('image.html', filename=session["uploaded image path"], cut=session["cut"], info=info, info_color=info_color, label_width=label_width)


@app.route('/', methods=['GET', 'POST'])
def base():
    check_for_new_user(session)
    return render_template('default.html')

# gets triggered when the user_data folder for the user is created
@new_user_handler
def on_new_user(session):
    print(Fore.BLUE + "new user with ID:", session["ID"], Style.RESET_ALL)

    session["text image path"] = None
    session["uploaded image path"] = None
    session["cut"] = True
    session["cut"] = True
    session["fonts"] = fonts()

class Image_preset:
    def __init__(self, path, dithering=True, threshold_if_not_dithering=0.5):
        self.path = path
        self.dithering = dithering
        self.threshold_if_not_dithering = threshold_if_not_dithering


image_presets = {
    "Abandoned": Image_preset("resources/abandoned.jpg", False, 0.5)
}

@app.route('/image', methods=['GET', 'POST'])
def image():
    check_for_new_user(session)

    if request.method == 'POST':
        if 'image' not in request.files:
            return render_image_template()

        
        if "dropdown" in request.form and request.form["dropdown"] != "file":
            path = image_presets[request.form["dropdown"]].path
            dither = image_presets[request.form["dropdown"]].dithering
            threshold_if_not_dithering = image_presets[request.form["dropdown"]].threshold_if_not_dithering
            class file:
                filename = os.path.basename(path)
                stream = open(path, "rb")
        else:
            file = request.files['image']
            dither = "dithering" in request.form
            threshold_if_not_dithering = float(request.form["threshold"]) * 0.01


        if file.filename == '':
            return render_image_template()
        else:
            extension = os.path.splitext(file.filename)[1]

            try:
                message, status, img = process_image(file.stream, dither, threshold_if_not_dithering)
                file.stream.close()
                session["uploaded image path"] = uploaded_image_filename
                img.save(get_file_path(session, session["uploaded image path"]))
            except Exception as e:
                message = f"{e}"
                status = "Error"


            if status == "Error":
                session["uploaded image path"] = None
                return render_image_template("Error: " + message, "red")

            elif status == "Info":
                return render_image_template("Info: " + message, "black")
            
    return render_image_template()


#TODO: maybe merge some stuff with text-print
@app.route('/image-print', methods=['GET', 'POST'])
def image_print():
    check_for_new_user(session)

    if session["uploaded image path"] == None:
        print(Fore.YELLOW + "Warning, file doesn't exist" + Style.RESET_ALL)
        return render_template('image.html', filename=session["uploaded image path"], cut=session["cut"])

    if request.method == 'POST':
        image = format_image_to_label(get_file_path(session, session["uploaded image path"]))
        print_image(image)

        if 'cut' in request.form:
            print("printing and cutting")
            cut_paper()
            session["cut"] = True;
        else:
            print("printing")
            session["cut"] = False;

    return render_image_template()



@app.route('/text-form', methods=['GET', 'POST'])
def text_form():
    check_for_new_user(session)

    if request.method == 'POST':
        line_count = int(request.form["lineCount"])
        lines = []
        for i in range(0, line_count):
            lines.append({
                "string": request.form["text" + str(i)],
                "size": int(request.form["size" + str(i)]),
                "pos": int(request.form["y_pos" + str(i)]),
                "bold": "bold" + str(i) in request.form
            })
        print(request.form)
            
        # Clear previously saved font
        for font in session["fonts"]:
            session["fonts"][font]["selected"] = False

        # If an invalid font has been submitted, just default to the first one
        chosen_font = request.form["font"]
        if chosen_font in session["fonts"]:
            session["fonts"][chosen_font]["selected"] = True
        else:
            first_font = next(iter(session["fonts"]))
            session["fonts"][first_font]["selected"] = True

        # Select the first font that has been marked as selected
        chosen_font = None
        for font in session["fonts"]:
            f = session["fonts"][font]
            if f["selected"]:
                chosen_font = f

        # If the font is still None, something has gone wrong
        if chosen_font is None:
            return

        print(chosen_font)

        message, status, img = gen_image(label_width, lines, chosen_font)

        if status == "Error":
            session["text image path"] = None
            return render_text_template(message, "red")

        get_folder_path(session)

        session["text image path"] = text_image_filename
        img.save(get_file_path(session, session["text image path"]))

        print()
        print(session["text image path"])

    return render_text_template(scrollDown = True)


@app.route('/text', methods=['GET', 'POST'])
def text():
    check_for_new_user(session)

    return render_text_template()


@app.route('/text-print', methods=['GET', 'POST'])
def text_print():
    check_for_new_user(session)

    if request.method == 'POST':
        if session["text image path"] == None:
            print(Fore.YELLOW + "Warning, file doesn't exist" + Style.RESET_ALL)
            return render_text_template()

        image = format_image_to_label(get_file_path(session, session["text image path"]))
        print_image(image)

        if 'cut' in request.form:
            print("printing and cutting")
            cut_paper()
            session["cut"] = True;
        else:
            print("printing")
            session["cut"] = False;

    return render_text_template()


@app.route('/user_data/<filename>')
def user_data(filename):
    return send_from_directory(get_folder_path(session), filename)



if __name__ == "__main__":
    app.run(debug=True)