forked from emilevs/label_printer
151 lines
4.7 KiB
Python
151 lines
4.7 KiB
Python
from flask import Flask, render_template, request, redirect, url_for, session
|
|
from PIL import Image
|
|
from gen_image import gen_image
|
|
from printer_info import *
|
|
from datetime import datetime
|
|
import os
|
|
|
|
valid_height=800
|
|
|
|
app = Flask(__name__)
|
|
app.config['UPLOAD_FOLDER'] = 'static/uploads'
|
|
app.config['TEXT_FOLDER'] = 'static/text'
|
|
app.secret_key = "blahaj"
|
|
|
|
# Ensure the upload directory exists
|
|
if not os.path.exists(app.config['UPLOAD_FOLDER']):
|
|
os.makedirs(app.config['UPLOAD_FOLDER'])
|
|
if not os.path.exists(app.config['TEXT_FOLDER']):
|
|
os.makedirs(app.config['TEXT_FOLDER'])
|
|
|
|
|
|
@app.route('/image', methods=['GET', 'POST'])
|
|
def image():
|
|
if request.method == 'POST':
|
|
if 'image' not in request.files:
|
|
return redirect(request.url)
|
|
file = request.files['image']
|
|
|
|
if file.filename == '':
|
|
return redirect(request.url)
|
|
if file:
|
|
extension = os.path.splitext(file.filename)[1]
|
|
filepath = os.path.join(app.config['UPLOAD_FOLDER'], "upload" + extension)
|
|
|
|
file.save(filepath)
|
|
|
|
# Process image #
|
|
height = Image.open(filepath).height
|
|
|
|
valid = False
|
|
if height == valid_height:
|
|
valid = True
|
|
#################
|
|
|
|
return render_template('image.html', filename="upload"+extension, height=height, valid=valid)
|
|
|
|
return render_template('image.html')
|
|
|
|
templates = {
|
|
"DNH": {
|
|
"text1": {
|
|
"string": "Do Not Hack",
|
|
"size": 50,
|
|
"pos": 0
|
|
},
|
|
"text2": {
|
|
"string": "bottom text",
|
|
"size": 25,
|
|
"pos": 55
|
|
},
|
|
"cut": True
|
|
},
|
|
"Food": {
|
|
"text1": {
|
|
"string": "Nickname",
|
|
"size": 50,
|
|
"pos": 0
|
|
},
|
|
"text2": {
|
|
"string": "",
|
|
"size": 25,
|
|
"pos": 55
|
|
},
|
|
"cut": True
|
|
}
|
|
}
|
|
|
|
@app.route('/text-template', methods=['GET', 'POST'])
|
|
def text_template():
|
|
if request.method == 'POST':
|
|
template = templates[request.form["template"]]
|
|
if request.form["template"] == "Food":
|
|
template["text2"]["string"] = datetime.now().strftime('%Y-%m-%d')
|
|
|
|
session["text1"] = template["text1"]
|
|
session["text2"] = template["text2"]
|
|
session["cut"] = template["cut"]
|
|
|
|
return render_template('text.html', filename="text.png", text1=session["text1"], text2=session["text2"], cut=session["cut"])
|
|
|
|
return render_template('text.html', filename="text.png", text1=session["text1"], text2=session["text2"], cut=session["cut"])
|
|
|
|
@app.route('/text-form', methods=['GET', 'POST'])
|
|
def text_form():
|
|
if request.method == 'POST':
|
|
session["text1"] = {
|
|
"string": request.form["string1"],
|
|
"size": int(request.form["size1"]),
|
|
"pos": int(request.form["pos1"]),
|
|
}
|
|
session["text2"] = {
|
|
"string": request.form["string2"],
|
|
"size": int(request.form["size2"]),
|
|
"pos": int(request.form["pos2"]),
|
|
}
|
|
|
|
img = gen_image(label_width, session["text1"], session["text2"])
|
|
filepath = os.path.join(app.config['TEXT_FOLDER'], "text.png")
|
|
img.save(filepath)
|
|
print(session)
|
|
|
|
return render_template('text.html', filename="text.png", text1=session["text1"], text2=session["text2"], cut=session["cut"])
|
|
|
|
return render_template('text.html', filename="text.png", text1=session["text1"], text2=session["text2"], cut=session["cut"])
|
|
|
|
|
|
|
|
@app.route('/text', methods=['GET', 'POST'])
|
|
def text():
|
|
session["text1"] = templates["DNH"]["text1"]
|
|
session["text2"] = templates["DNH"]["text2"]
|
|
session["cut"] = templates["DNH"]["cut"]
|
|
return render_template('text.html', filename="text.png", text1=session["text1"], text2=session["text2"], cut=session["cut"])
|
|
|
|
|
|
@app.route('/text-print', methods=['GET', 'POST'])
|
|
def tex_print():
|
|
if request.method == 'POST':
|
|
if 'cut' in request.form:
|
|
print("printing and cutting")
|
|
session["cut"] = True;
|
|
else:
|
|
print("printing")
|
|
session["cut"] = False;
|
|
|
|
return render_template('text.html', filename="text.png", text1=session["text1"], text2=session["text2"], cut=session["cut"])
|
|
|
|
return render_template('text.html', filename="text.png", text1=session["text1"], text2=session["text2"], cut=session["cut"])
|
|
|
|
|
|
@app.route('/uploads/<filename>')
|
|
def uploaded_file(filename):
|
|
return redirect(url_for('static', filename='uploads/' + filename))
|
|
|
|
@app.route('/text/<filename>')
|
|
def generated_file(filename):
|
|
return redirect(url_for('static', filename='text/' + filename))
|
|
|
|
if __name__ == "__main__":
|
|
app.run(debug=True)
|
|
|