n07070/littleprynter
2
0
Fork 0
Code Issues 15 Pull Requests 1 Projects Releases Activity

Update linting with black

Browse Source
This commit is contained in:
n07070
2026-05-17 14:24:07 +02:00
parent 641b8a2d1f
commit 85c10a47b0
4 changed files with 214 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

128
src/printer.py
View File

@@ -34,31 +34,33 @@ class Printer(object):
self.device_id = device_id self.device_id = device_id
self.vendor_id = vendor_id self.vendor_id = vendor_id
self.usb_args = {} self.usb_args = {}
self.usb_args['idVendor'] = self.device_id self.usb_args["idVendor"] = self.device_id
self.usb_args['idProduct'] = self.vendor_id self.usb_args["idProduct"] = self.vendor_id
def check_paper(self) -> bool: def check_paper(self) -> bool:
# Let's check paper status # Let's check paper status
self.app.logger.debug('Checking paper status...') self.app.logger.debug("Checking paper status...")
self.printer.open(self.usb_args) self.printer.open(self.usb_args)
status = self.printer.paper_status() status = self.printer.paper_status()
match status: match status:
case 0: case 0:
self.app.logger.error('Printer has no more paper, aborting...') self.app.logger.error("Printer has no more paper, aborting...")
self.printer.close() self.printer.close()
raise Exception("No more paper in the printer") raise Exception("No more paper in the printer")
case 1: case 1:
self.app.logger.warning('Printer needs paper to be changed very soon ! ') self.app.logger.warning(
"Printer needs paper to be changed very soon ! "
)
self.printer.close() self.printer.close()
case 2: case 2:
self.app.logger.debug('Printer has paper, good to go') self.app.logger.debug("Printer has paper, good to go")
self.printer.close() self.printer.close()
def init_printer(self): def init_printer(self):
# Is the printer online ? Is the communication with the printer successfull ? # Is the printer online ? Is the communication with the printer successfull ?
waiting_elapsed = 30 waiting_elapsed = 30
self.app.logger.debug('Waiting for printer to get online...') self.app.logger.debug("Waiting for printer to get online...")
while not self.ready: while not self.ready:
try: try:
@@ -66,56 +68,83 @@ class Printer(object):
# or else the device will appear as busy. # or else the device will appear as busy.
p = Usb(self.device_id, self.vendor_id, 0, profile="TM-P80") p = Usb(self.device_id, self.vendor_id, 0, profile="TM-P80")
except USBNotFoundError as e: except USBNotFoundError as e:
self.app.logger.error("The USB device is not plugged in, trying again : " + str(e)) self.app.logger.error(
"The USB device is not plugged in, trying again : " + str(e)
)
pass pass
try: try:
if p.is_online(): if p.is_online():
self.ready = True self.ready = True
self.app.logger.debug('Printer online !') self.app.logger.debug("Printer online !")
except Exception as e: except Exception as e:
pass pass
sleep(1) sleep(1)
waiting_elapsed -= 1 waiting_elapsed -= 1
if waiting_elapsed < 1: if waiting_elapsed < 1:
self.app.logger.error('Printer took more than 30 seconds to get online, aborting...') self.app.logger.error(
waiting_elapsed = 30 # Reset the waiting time for the next print. "Printer took more than 30 seconds to get online, aborting..."
)
waiting_elapsed = 1 # Reset the waiting time for the next print.
return False return False
# Setting up the printing options. # Setting up the printing options.
p.set(align='center', font='a', bold=False, underline=0, width=1, height=1, density=9, invert=False, smooth=False, flip=False, double_width=False, double_height=False, custom_size=False) p.set(
align="center",
font="a",
bold=False,
underline=0,
width=1,
height=1,
density=9,
invert=False,
smooth=False,
flip=False,
double_width=False,
double_height=False,
custom_size=False,
)
# Beware : if we print every time the printer becomes ready, it means # Beware : if we print every time the printer becomes ready, it means
# we are printing before and after every print ! # we are printing before and after every print !
self.printer = p self.printer = p
self.ready = True; self.ready = True
self.printer.close(); self.printer.close()
self.check_paper() self.check_paper()
return True return True
def print_sms(self, msg, signature="",bold=False): def print_sms(self, msg, signature="", bold=False):
clean_msg = str(msg) + "\n" clean_msg = str(msg) + "\n"
clean_signature = str(signature) clean_signature = str(signature)
if len(clean_msg) > 4096: if len(clean_msg) > 4096:
self.app.logger.warning("Could not print message of this length: " + str(len(clean_msg))) self.app.logger.warning(
raise Exception("Could not print message of this length :" + str(len(clean_msg)) + ", needs to be below 4096 caracters long.") "Could not print message of this length: " + str(len(clean_msg))
)
raise Exception(
"Could not print message of this length :"
+ str(len(clean_msg))
+ ", needs to be below 4096 caracters long."
)
if len(signature) > 256: if len(signature) > 256:
self.app.logger.warning("Could not print signature of this length: " + str(len(clean_signature))) self.app.logger.warning(
raise Exception("Could not print signature of this length :" + str(len(clean_signature)) + ", needs to be below 256 caracters long.") "Could not print signature of this length: " + str(len(clean_signature))
)
raise Exception(
"Could not print signature of this length :"
+ str(len(clean_signature))
+ ", needs to be below 256 caracters long."
)
try: try:
self.printer.open(self.usb_args); self.printer.open(self.usb_args)
self.printer.set(align='center', font='a', bold=bold) self.printer.set(align="center", font="a", bold=bold)
self.printer.textln(clean_msg ) self.printer.textln(clean_msg)
if clean_signature: if clean_signature:
self.printer.textln(clean_signature ) self.printer.textln(clean_signature)
self.printer.close() self.printer.close()
except Exception as e: except Exception as e:
self.app.logger.error("Unable to print because : " + str(e)) self.app.logger.error("Unable to print because : " + str(e))
@@ -123,16 +152,26 @@ class Printer(object):
self.app.logger.info("Printed text") self.app.logger.info("Printed text")
return True return True
def print_img(self, path, sign="",center=True,process=False): def print_img(self, path, sign="", center=True, process=False):
clean_signature = str(sign) clean_signature = str(sign)
if len(sign) > 256: if len(sign) > 256:
self.app.logger.warning("Could not print signature of this length: " + str(len(clean_signature))) self.app.logger.warning(
raise Exception("Could not print signature of this length :" + str(len(clean_signature)) + ", needs to be below 256 caracters long.") "Could not print signature of this length: " + str(len(clean_signature))
)
raise Exception(
"Could not print signature of this length :"
+ str(len(clean_signature))
+ ", needs to be below 256 caracters long."
)
if not os.path.isfile(str(path)): if not os.path.isfile(str(path)):
self.app.logger.warning("File does not exist : " + str(path)) self.app.logger.warning("File does not exist : " + str(path))
raise Exception('The file path for this image :' + str(path) + " wasn't found. Please try again.") raise Exception(
"The file path for this image :"
+ str(path)
+ " wasn't found. Please try again."
)
else: else:
self.app.logger.debug("Printing file from " + str(path)) self.app.logger.debug("Printing file from " + str(path))
@@ -146,10 +185,9 @@ class Printer(object):
else: else:
self.app.logger.warning("Not proccessing the image") self.app.logger.warning("Not proccessing the image")
try: try:
self.printer.open(self.usb_args) self.printer.open(self.usb_args)
self.printer.image(path,center=center) self.printer.image(path, center=center)
self.printer.close() self.printer.close()
self.app.logger.debug("Printed an image : " + str(path)) self.app.logger.debug("Printed an image : " + str(path))
os.remove(path) os.remove(path)
@@ -175,8 +213,6 @@ class Printer(object):
self.app.logger.info("Printed a QR") self.app.logger.info("Printed a QR")
return True return True
def cut(self): def cut(self):
try: try:
self.printer.open(self.usb_args) self.printer.open(self.usb_args)
@@ -194,7 +230,7 @@ class Printer(object):
def process_image(self, path): def process_image(self, path):
brightness_factor = 1.5 # Used only if image is too dark brightness_factor = 1.5 # Used only if image is too dark
brightness_threshold = 100 # Brightness threshold (0–255) brightness_threshold = 100 # Brightness threshold (0–255)
contrast_factor = 0.6 # Less than 1.0 = lower contrast contrast_factor = 0.6 # Less than 1.0 = lower contrast
max_width = 575 max_width = 575
max_height = 1000 max_height = 1000
@@ -215,13 +251,21 @@ def process_image(self, path):
# Compute brightness of original image (grayscale average) # Compute brightness of original image (grayscale average)
grayscale = original_img.convert("L") grayscale = original_img.convert("L")
avg_brightness = np.array(grayscale).mean() avg_brightness = np.array(grayscale).mean()
self.app.logger.debug("Average brightness of the image : " + str(avg_brightness) ) self.app.logger.debug(
"Average brightness of the image : " + str(avg_brightness)
)
# Dynamically compute brightness factor if too dark # Dynamically compute brightness factor if too dark
if avg_brightness < brightness_threshold: if avg_brightness < brightness_threshold:
brightness_factor = 1 + (brightness_threshold - avg_brightness) / brightness_threshold brightness_factor = (
brightness_factor = min(max(brightness_factor, 1.1), 2.5) # Clamp between 1.1 and 2.5 1 + (brightness_threshold - avg_brightness) / brightness_threshold
self.app.logger.debug(f"Image too dark, increasing brightness by a factor of {brightness_factor:.2f}") )
brightness_factor = min(
max(brightness_factor, 1.1), 2.5
) # Clamp between 1.1 and 2.5
self.app.logger.debug(
f"Image too dark, increasing brightness by a factor of {brightness_factor:.2f}"
)
enhancer = ImageEnhance.Brightness(original_img) enhancer = ImageEnhance.Brightness(original_img)
original_img = enhancer.enhance(brightness_factor) original_img = enhancer.enhance(brightness_factor)
@@ -232,11 +276,13 @@ def process_image(self, path):
# Final resize check # Final resize check
if original_img.height > max_height: if original_img.height > max_height:
raise ValueError("Image is too long, sorry! Keep it below 575×1000 pixels.") raise ValueError("Image is too long, sorry! Keep it below 575×1000 pixels.")
self.app.logger.error("Image is too long, sorry! Keep it below 575×1000 pixels.") self.app.logger.error(
"Image is too long, sorry! Keep it below 575×1000 pixels."
)
return False return False
# Convert to JPEG and save # Convert to JPEG and save
jpeg_path = os.path.splitext(path)[0] + "_processed.jpg" jpeg_path = os.path.splitext(path)[0] + "_processed.jpg"
original_img.save(jpeg_path, format='JPEG', quality=95, optimize=True) original_img.save(jpeg_path, format="JPEG", quality=95, optimize=True)
return jpeg_path return jpeg_path

129
src/raspberry.py
View File

@@ -2,10 +2,11 @@ from flask_socketio import SocketIO
from gpiozero import Button, LED, DigitalOutputDevice from gpiozero import Button, LED, DigitalOutputDevice
from time import sleep, gmtime, strftime from time import sleep, gmtime, strftime
from PIL import Image from PIL import Image
import io # To check if we are on a Raspberry Pi import io # To check if we are on a Raspberry Pi
import subprocess import subprocess
import os import os
class Raspberry(object): class Raspberry(object):
""" """
This class will manage three things : This class will manage three things :
@@ -15,7 +16,17 @@ class Raspberry(object):
- Flash an indicator light - Flash an indicator light
""" """
def __init__(self, printer, app, socketio, button_gpio_port_number, indicator_gpio_port_number, flash_gpio_port_number, is_flash_present,):
def __init__(
self,
printer,
app,
socketio,
button_gpio_port_number,
indicator_gpio_port_number,
flash_gpio_port_number,
is_flash_present,
):
self.printer = printer self.printer = printer
self.socketio = socketio self.socketio = socketio
self.app = app self.app = app
@@ -24,67 +35,71 @@ class Raspberry(object):
self.is_flash_present = is_flash_present self.is_flash_present = is_flash_present
self.button_gpio = button_gpio_port_number self.button_gpio = button_gpio_port_number
self.led_gpio = indicator_gpio_port_number self.led_gpio = indicator_gpio_port_number
self.image_path = self.app.config['UPLOAD_FOLDER'] + '/image.jpg' self.image_path = self.app.config["UPLOAD_FOLDER"] + "/image.jpg"
def is_raspberry_pi(self, raise_on_errors=False): def is_raspberry_pi(self, raise_on_errors=False):
# Check if we are running on a raspberry pi # Check if we are running on a raspberry pi
try: try:
with io.open('/proc/cpuinfo', 'r') as cpuinfo: with io.open("/proc/cpuinfo", "r") as cpuinfo:
found = False found = False
for line in cpuinfo: for line in cpuinfo:
if line.startswith('Hardware'): if line.startswith("Hardware"):
found = True found = True
label, value = line.strip().split(':', 1) label, value = line.strip().split(":", 1)
value = value.strip() value = value.strip()
if value not in ( if value not in (
'BCM2708', "BCM2708",
'BCM2709', "BCM2709",
'BCM2711', "BCM2711",
'BCM2835', "BCM2835",
'BCM2836' "BCM2836",
): ):
self.app.logger.debug('This system does not appear to be a Raspberry Pi.') self.app.logger.debug(
"This system does not appear to be a Raspberry Pi."
)
return False return False
if not found: if not found:
self.app.logger.error('Couldn\'t get sufficient hardware information from /proc/cpuinfo, Unable to determine if we are on a Raspberry Pi.') self.app.logger.error(
"Couldn't get sufficient hardware information from /proc/cpuinfo, Unable to determine if we are on a Raspberry Pi."
)
return False return False
except IOError: except IOError:
self.app.logger.error('Unable to open `/proc/cpuinfo`.') self.app.logger.error("Unable to open `/proc/cpuinfo`.")
return False return False
self.app.logger.debug('It seems we are on a Raspberry Pi') self.app.logger.debug("It seems we are on a Raspberry Pi")
try: try:
self.initialise_gpio() self.initialise_gpio()
except Exception as e: except Exception as e:
self.app.logger.debug('Could not init GPIO : ' + str(e)) self.app.logger.debug("Could not init GPIO : " + str(e))
raise e raise e
return True return True
def initialise_gpio(self): def initialise_gpio(self):
self.app.logger.debug('Initializing GPIO') self.app.logger.debug("Initializing GPIO")
self.led = LED(self.led_gpio) self.led = LED(self.led_gpio)
self.app.logger.debug('Activated indicator LED') self.app.logger.debug("Activated indicator LED")
self.indicator_countdown(iters=3) self.indicator_countdown(iters=3)
self.button = Button(self.button_gpio, pull_up=True, bounce_time=0.1) self.button = Button(self.button_gpio, pull_up=True, bounce_time=0.1)
self.button.when_pressed = self.on_button_pressed self.button.when_pressed = self.on_button_pressed
self.app.logger.debug('Activated button') self.app.logger.debug("Activated button")
# The "flash" is a relay-controlled device ( light bulb for example ) # The "flash" is a relay-controlled device ( light bulb for example )
self.flash = DigitalOutputDevice(self.flash_gpio) self.flash = DigitalOutputDevice(self.flash_gpio)
self.flash_toggle() self.flash_toggle()
self.app.logger.debug('Activated flash') self.app.logger.debug("Activated flash")
def indicator_countdown(self,iters=10,multi=10): def indicator_countdown(self, iters=10, multi=10):
for i in range(iters,0,-1): for i in range(iters, 0, -1):
self.led.on() self.led.on()
sleep(i/multi) sleep(i / multi)
self.led.off() self.led.off()
sleep(i/multi) sleep(i / multi)
def indicator_led(self,timing=0.2,l=5): def indicator_led(self, timing=0.2, l=5):
for i in range(l): for i in range(l):
self.app.logger.debug("LED turned on") self.app.logger.debug("LED turned on")
self.led.on() self.led.on()
@@ -103,22 +118,28 @@ class Raspberry(object):
def take_picture(self): def take_picture(self):
# Validate if the image path is valid # Validate if the image path is valid
if not os.path.isdir(os.path.dirname(self.image_path)): if not os.path.isdir(os.path.dirname(self.image_path)):
self.app.logger.error(f"Invalid directory for image path: {self.image_path}") self.app.logger.error(
f"Invalid directory for image path: {self.image_path}"
)
return False return False
try: try:
result = subprocess.run( result = subprocess.run(
['fswebcam', '--no-banner', '-r', '1920x1080', self.image_path], ["fswebcam", "--no-banner", "-r", "1920x1080", self.image_path],
check=True, # Will raise CalledProcessError if the command fails check=True, # Will raise CalledProcessError if the command fails
stdout=subprocess.PIPE, # Capture standard output stdout=subprocess.PIPE, # Capture standard output
stderr=subprocess.PIPE # Capture error output stderr=subprocess.PIPE, # Capture error output
) )
# Optionally log the command output # Optionally log the command output
self.app.logger.debug(f"Image captured successfully: {result.stdout.decode()}") self.app.logger.debug(
f"Image captured successfully: {result.stdout.decode()}"
)
except subprocess.CalledProcessError as e: except subprocess.CalledProcessError as e:
# Log error output if available # Log error output if available
self.app.logger.error(f"Unable to take a picture. Error: {e.stderr.decode()}") self.app.logger.error(
f"Unable to take a picture. Error: {e.stderr.decode()}"
)
return False return False
except Exception as e: except Exception as e:
# Catch any unexpected errors # Catch any unexpected errors
@@ -132,32 +153,42 @@ class Raspberry(object):
return True return True
def overlay_logo(
def overlay_logo(self, image_path, logo_path, output_path=None, position='bottom_right', margin=10): self,
image_path,
logo_path,
output_path=None,
position="bottom_right",
margin=10,
):
try: try:
image = Image.open(image_path).convert("RGBA") image = Image.open(image_path).convert("RGBA")
logo = Image.open(logo_path).convert("RGBA") logo = Image.open(logo_path).convert("RGBA")
# Resize logo if it's too big (logo will be 30% the width of the image) # Resize logo if it's too big (logo will be 30% the width of the image)
logo_ratio = 0.30 # You can change the ratio if you want the logo bigger or smaller logo_ratio = (
0.30 # You can change the ratio if you want the logo bigger or smaller
)
logo_width = int(image.width * logo_ratio) logo_width = int(image.width * logo_ratio)
logo_height = int(logo.height * (logo_width / logo.width)) logo_height = int(logo.height * (logo_width / logo.width))
logo = logo.resize((logo_width, logo_height), Image.Resampling.LANCZOS) logo = logo.resize((logo_width, logo_height), Image.Resampling.LANCZOS)
# Calculate position based on the chosen location # Calculate position based on the chosen location
if position == 'bottom_right': if position == "bottom_right":
x = image.width - logo.width - margin x = image.width - logo.width - margin
y = image.height - logo.height - margin y = image.height - logo.height - margin
elif position == 'top_left': elif position == "top_left":
x, y = margin, margin x, y = margin, margin
elif position == 'top_right': elif position == "top_right":
x = image.width - logo.width - margin x = image.width - logo.width - margin
y = margin y = margin
elif position == 'bottom_left': elif position == "bottom_left":
x = margin x = margin
y = image.height - logo.height - margin y = image.height - logo.height - margin
else: else:
raise ValueError("Invalid position. Choose from 'bottom_right', 'top_left', 'top_right', or 'bottom_left'.") raise ValueError(
"Invalid position. Choose from 'bottom_right', 'top_left', 'top_right', or 'bottom_left'."
)
# Composite the logo onto the image # Composite the logo onto the image
image.paste(logo, (x, y), logo) # Use logo as its own alpha mask image.paste(logo, (x, y), logo) # Use logo as its own alpha mask
@@ -204,27 +235,31 @@ class Raspberry(object):
self.app.logger.debug("Button has been pressed") self.app.logger.debug("Button has been pressed")
self.led.on() self.led.on()
self.app.logger.debug("Counting down") self.app.logger.debug("Counting down")
self.indicator_countdown(iters=4,multi=20) # The indicator will flash a countdown LED self.indicator_countdown(
iters=4, multi=20
) # The indicator will flash a countdown LED
self.app.logger.debug("Taking picture") self.app.logger.debug("Taking picture")
try: try:
self.flash.on() self.flash.on()
self.take_picture() self.take_picture()
except Exception as e: except Exception as e:
self.app.logger.error("Could not take a picture after the button press : " + str(e)) self.app.logger.error(
"Could not take a picture after the button press : " + str(e)
)
finally: finally:
self.flash.off() self.flash.off()
self.app.logger.debug("Printing picture") self.app.logger.debug("Printing picture")
self.led.on() self.led.on()
self.crop_to_square(self.image_path) self.crop_to_square(self.image_path)
self.printer.print_img('src/static/images/extase-club.png',process=True) self.printer.print_img("src/static/images/extase-club.png", process=True)
self.printer.print_img(self.image_path,process=True) self.printer.print_img(self.image_path, process=True)
self.printer.print_sms("") self.printer.print_sms("")
self.printer.print_sms("With Love From Société.Vide",signature="",bold=True) self.printer.print_sms("With Love From Société.Vide", signature="", bold=True)
self.printer.print_sms("Printed by LittlePrynter",signature="") self.printer.print_sms("Printed by LittlePrynter", signature="")
self.printer.print_sms("n07070.xyz",signature="") self.printer.print_sms("n07070.xyz", signature="")
self.printer.print_sms(strftime("%Y-%m-%d %H:%M", gmtime()),signature="") self.printer.print_sms(strftime("%Y-%m-%d %H:%M", gmtime()), signature="")
self.printer.qr("https://n07070.xyz/articles/littleprynter") self.printer.qr("https://n07070.xyz/articles/littleprynter")
self.printer.cut() self.printer.cut()
self.led.off() self.led.off()
self.app.logger.debug("Done printing picture") self.app.logger.debug("Done printing picture")
return True return True

38
src/user.py
View File

@@ -5,35 +5,37 @@ class User(object):
super(User, self).__init__() super(User, self).__init__()
self.arg = arg self.arg = arg
# @app.route('/login', methods=['POST','GET']) # @app.route('/login', methods=['POST','GET'])
# @limiter.limit("100 per minute", error_message=error_handler_limiter) # @limiter.limit("100 per minute", error_message=error_handler_limiter)
def login(): def login():
if request.method == 'POST': if request.method == "POST":
if not session.get('logged_in'): if not session.get("logged_in"):
if request.form['username'] and request.form['password']: if request.form["username"] and request.form["password"]:
# Get the json # Get the json
with open('users.json') as f: with open("users.json") as f:
users_file = json.load(f) users_file = json.load(f)
for user in users_file["users"]: for user in users_file["users"]:
if users_file["users"][user] == request.form['password']: if users_file["users"][user] == request.form["password"]:
session['logged_in'] = True session["logged_in"] = True
session['user'] = request.form['username'] session["user"] = request.form["username"]
if not session.get('logged_in'): if not session.get("logged_in"):
flash('Mot de passe ou pseudo invalide.','danger') flash("Mot de passe ou pseudo invalide.", "danger")
return redirect(url_for('login')) return redirect(url_for("login"))
else: else:
return redirect(url_for('display_index_page')) return redirect(url_for("display_index_page"))
else: else:
flash('Incorrect logins') flash("Incorrect logins")
return render_template('password.html') return render_template("password.html")
else: else:
return render_template('password.html') return render_template("password.html")
else: else:
return render_template('password.html') return render_template("password.html")
@app.route("/logout") @app.route("/logout")
def logout(): def logout():
session['logged_in'] = False session["logged_in"] = False
flash('Tu est déconnecté', 'info') flash("Tu est déconnecté", "info")
return redirect(url_for('login')) return redirect(url_for("login"))

36
src/web.py
View File

@@ -4,10 +4,11 @@ from printer import Printer
import time import time
import os import os
class Web(object): class Web(object):
"""docstring for web.""" """docstring for web."""
def __init__(self, app, printer ): def __init__(self, app, printer):
super(Web).__init__() super(Web).__init__()
self.printer = printer self.printer = printer
self.app = app self.app = app
@@ -20,10 +21,9 @@ class Web(object):
self.printer.cut() self.printer.cut()
except Exception as e: except Exception as e:
self.app.logger.error(e) self.app.logger.error(e)
flash("Error while printing the SMS : "+ str(e)) flash("Error while printing the SMS : " + str(e))
flash("You message " + str( texte ) + " has been printed :)")
flash("You message " + str(texte) + " has been printed :)")
def print_image(self, image, sign): def print_image(self, image, sign):
self.app.logger.debug("Uploading file") self.app.logger.debug("Uploading file")
@@ -31,7 +31,14 @@ class Web(object):
self.app.logger.debug("Uploading file from " + str(sign)) self.app.logger.debug("Uploading file from " + str(sign))
if self.upload_file(image): if self.upload_file(image):
self.app.logger.debug("File has been uploaded, printing...") self.app.logger.debug("File has been uploaded, printing...")
self.printer.print_img(os.path.join(self.app.config['UPLOAD_FOLDER'], secure_filename(image.filename)), sign=sign,process=True) self.printer.print_img(
os.path.join(
self.app.config["UPLOAD_FOLDER"],
secure_filename(image.filename),
),
sign=sign,
process=True,
)
self.printer.cut() self.printer.cut()
except Exception as e: except Exception as e:
self.app.logger.error(e) self.app.logger.error(e)
@@ -39,7 +46,7 @@ class Web(object):
flash("Your image has been printed :)") flash("Your image has been printed :)")
def login(username: str,password: str) -> bool: def login(username: str, password: str) -> bool:
pass pass
def logout(username: str, password: str) -> bool: def logout(username: str, password: str) -> bool:
@@ -47,22 +54,29 @@ class Web(object):
def allowed_file(self, filename) -> bool: def allowed_file(self, filename) -> bool:
self.app.logger.debug("Is the filename allowed ?") self.app.logger.debug("Is the filename allowed ?")
return '.' in filename and filename.rsplit('.', 1)[1].lower() in self.app.config['ALLOWED_EXTENSIONS'] return (
"." in filename
and filename.rsplit(".", 1)[1].lower()
in self.app.config["ALLOWED_EXTENSIONS"]
)
def upload_file(self, image)-> bool: def upload_file(self, image) -> bool:
self.app.logger.debug("Validating file") self.app.logger.debug("Validating file")
if image and self.allowed_file(image.filename): if image and self.allowed_file(image.filename):
filename = secure_filename(image.filename) filename = secure_filename(image.filename)
self.app.logger.debug("File valid") self.app.logger.debug("File valid")
try: try:
image.save(os.path.join(self.app.config['UPLOAD_FOLDER'], filename)) image.save(os.path.join(self.app.config["UPLOAD_FOLDER"], filename))
self.app.logger.debug("File saved") self.app.logger.debug("File saved")
except Exception as e: except Exception as e:
self.app.logger.error("Could not save file") self.app.logger.error("Could not save file")
flash(str(e),'error') flash(str(e), "error")
return False return False
self.app.logger.debug("File saved to " + str(os.path.join(self.app.config['UPLOAD_FOLDER'], filename))) self.app.logger.debug(
"File saved to "
+ str(os.path.join(self.app.config["UPLOAD_FOLDER"], filename))
)
return True return True
else: else:
self.app.logger.error("Could not save file " + str(filename)) self.app.logger.error("Could not save file " + str(filename))