Compare commits

4 Commits

Author SHA1 Message Date
n07070
1e46de4b3c Update line lenght of docstring 2026-05-22 10:58:41 +02:00
n07070
e9bf334abd Update raspberry pi class to print via the print queue 2026-05-22 10:58:26 +02:00
n07070
1ba3d1bee3 Add docstring & comments, remove dead code 2026-05-22 10:43:38 +02:00
n07070
20f1145b60 Add comments about the code structure 2026-05-22 10:43:09 +02:00
4 changed files with 49 additions and 30 deletions

View File

@@ -79,6 +79,16 @@ Your contributions are very much welcome ! You can either request an account on
Please also say if you had a printer to test your code, and which printer you've been using.
### Code structure
The app is written about the Flask framework. You can start by looking at the code in the `src/` folder, in the `main.py` file. There, you will see that a few classes are initialized. In general, they are two parts to the program :
The Web pages and the API, which are the user-facing parts. This is with what the users will interact, and define how the program is going to be used. The web pages are renderer from the `include/` folder where Jinj2 templates are defined.
The Worker and Printer Queue are the internal parts. When a new thing needs to be printed, usually sent from the Web or API interfaces, a new Task in the type of the document is created, and added to a print queue. Then, a Worker thread looks up the state of the queue every so often and picks jobs to execute on the printers connected to the system.
The last part of the program is the Raspberry Pi class, that handles to Photomaton mode, which handles button presses, and LED indicator and a flash.
### Linting
If you want to contribute code, please make sure to lint the project before commiting. This helps the code keep a general structure, and avoids some commons erros and mistakes.

View File

@@ -101,9 +101,6 @@ app.config["TEMPLATES_AUTO_RELOAD"] = True
# Printer connection
# Uses the class defined in the printer.py file
printer = Printer(app, 0x04B8, 0x0E28)
# printers = Printer(app)
# printers.discover_printers()
# printers.init()
printer.init_printer()
# Find out if we are running on a Raspberry Pi
@@ -122,20 +119,19 @@ RASPBERRY_PI_CONNECTED = rpi.is_raspberry_pi()
# Queue creation
print_queue = PrintQueue(app)
# Web & API routes
# Web & API management
web = Web(app, print_queue)
# Start worker thread
worker = PrintWorker(app, print_queue, printer, socketio)
worker.start()
# The rate limit
limiter = Limiter(
get_remote_address, app=app, default_limits=["1500 per day", "500 per hour"]
)
# General routes
@app.route("/")
@limiter.limit("1/second", override_defaults=False)
def index():
@@ -153,8 +149,6 @@ def webcam():
# Form treatement
@app.route("/web/print/sms", methods=["POST"])
@limiter.limit("6/minute", override_defaults=False)
def web_print_sms():
@@ -185,7 +179,7 @@ def web_print_sms():
return redirect(url_for("index"))
# end try
flash("The SMS has been printed !", "info")
flash("The SMS has been added to the print queue !", "info")
return redirect(url_for("index"))
@@ -227,7 +221,7 @@ def web_print_img():
flash("The image could not be printed because : " + str(e), "error")
return redirect(url_for("index"))
flash("Picture printed !", "info")
flash("Picture added to the print queue !", "info")
return redirect(url_for("index"))
@@ -296,8 +290,7 @@ def api_print_image():
return "No image submitted", 400
file = request.files["img"]
# If the user does not select a file, the browser submits an
# empty file without a filename.
# If the user submits an empty file without a filename.
if file.filename == "":
app.logger.error("Submitted file has no filename !")
return "Submitted file has no filename !", 400
@@ -337,12 +330,18 @@ def api_worker_state():
@app.route("/api/worker/start")
def api_worker_start():
"""
Enable to worker. This starts to process the print queue.
"""
worker.start_worker()
return jsonify(worker.current_state())
@app.route("/api/worker/stop")
def api_worker_stop():
"""
Stops the print queue. This stops the processing of the print queue.
"""
worker.stop_worker()
return jsonify(worker.current_state())

View File

@@ -1,3 +1,11 @@
"""
This class executes when we are on a raspberry Pi.
It handles the press of a button via GPIO,
activates a flash and prints out the picture
that was taken via a USB Webcam.
"""
import io # To check if we are on a Raspberry Pi
import subprocess
import os
@@ -5,6 +13,7 @@ from time import sleep, gmtime, strftime
from flask_socketio import SocketIO
from gpiozero import Button, LED, DigitalOutputDevice
from PIL import Image
from task import TextTask, ImageTask, CutTask
class Raspberry():
"""
@@ -18,7 +27,7 @@ class Raspberry():
def __init__(
self,
printer,
print_queue,
app,
socketio,
button_gpio_port_number,
@@ -26,7 +35,7 @@ class Raspberry():
flash_gpio_port_number,
is_flash_present,
):
self.printer = printer
self.print_queue = print_queue
self.socketio = socketio
self.app = app
@@ -37,9 +46,13 @@ class Raspberry():
self.image_path = self.app.config["UPLOAD_FOLDER"] + "/image.jpg"
def is_raspberry_pi(self, raise_on_errors=False):
"""
Checking if we are on a Raspberry Pi by checking
information on the /proc/cpuinfo file
"""
# Check if we are running on a raspberry pi
try:
with io.open("/proc/cpuinfo", "r") as cpuinfo:
with io.open("/proc/cpuinfo", "r", encoding="utf-8") as cpuinfo:
found = False
for line in cpuinfo:
if line.startswith("Hardware"):
@@ -140,7 +153,7 @@ class Raspberry():
f"Unable to take a picture. Error: {e.stderr.decode()}"
)
return False
except Exception as e:
except RuntimeError as e:
# Catch any unexpected errors
self.app.logger.error(f"Unexpected error while taking picture: {str(e)}")
return False
@@ -197,7 +210,7 @@ class Raspberry():
output_path = image_path # Overwrite the original image if no output path is given
image.save(output_path)
except Exception as e:
except RuntimeError as e:
self.app.logger.error(f"Error overlaying logo: {e}")
return False
@@ -224,7 +237,7 @@ class Raspberry():
image.save(output_path)
except Exception as e:
except RuntimeError as e:
self.app.logger.error(f"Error cropping image to square: {e}")
return False
@@ -241,7 +254,7 @@ class Raspberry():
try:
self.flash.on()
self.take_picture()
except Exception as e:
except RuntimeError as e:
self.app.logger.error(
"Could not take a picture after the button press : " + str(e)
)
@@ -250,15 +263,11 @@ class Raspberry():
self.app.logger.debug("Printing picture")
self.led.on()
self.crop_to_square(self.image_path)
self.printer.print_img("src/static/images/extase-club.png", process=True)
self.printer.print_img(self.image_path, process=True)
self.printer.print_sms("")
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("n07070.xyz", signature="")
self.printer.print_sms(strftime("%Y-%m-%d %H:%M", gmtime()), signature="")
self.printer.qr("https://n07070.xyz/articles/littleprynter")
self.printer.cut()
self.print_queue.enqueue(ImageTask(self.image_path,signature="",process=True))
self.print_queue.enqueue(TextTask(content="Imprimé par LittlePrynter", signature=""))
time = strftime("%Y-%m-%d %H:%M", gmtime())
self.print_queue.enqueue(TextTask(content=time, signature=""))
self.print_queue.enqueue(CutTask())
self.led.off()
self.app.logger.debug("Done printing picture")
self.app.logger.debug("Added a photomaton picture to the print queue")
return True

View File

@@ -6,7 +6,8 @@ from task import TextTask, ImageTask, CutTask
class Web(object):
"""Web is the class that gets all of the information from web calls ( API and Web page ) and provides checks before sending stuff to printing"""
"""Web is the class that gets all of the information from web calls
( API and Web page ) and provides checks before sending stuff to printing"""
def __init__(self, app, print_queue):
super(Web).__init__()