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

126
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)
@@ -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

125
src/raspberry.py
View File

@@ -6,6 +6,7 @@ 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,25 +235,29 @@ 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()

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))