Compare commits
4 Commits
d2f670bb68
...
1e46de4b3c
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
1e46de4b3c | ||
|
|
e9bf334abd | ||
|
|
1ba3d1bee3 | ||
|
|
20f1145b60 |
10
README.md
10
README.md
@@ -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.
|
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
|
### 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.
|
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.
|
||||||
|
|||||||
23
src/main.py
23
src/main.py
@@ -101,9 +101,6 @@ app.config["TEMPLATES_AUTO_RELOAD"] = True
|
|||||||
# Printer connection
|
# Printer connection
|
||||||
# Uses the class defined in the printer.py file
|
# Uses the class defined in the printer.py file
|
||||||
printer = Printer(app, 0x04B8, 0x0E28)
|
printer = Printer(app, 0x04B8, 0x0E28)
|
||||||
# printers = Printer(app)
|
|
||||||
# printers.discover_printers()
|
|
||||||
# printers.init()
|
|
||||||
printer.init_printer()
|
printer.init_printer()
|
||||||
|
|
||||||
# Find out if we are running on a Raspberry Pi
|
# Find out if we are running on a Raspberry Pi
|
||||||
@@ -122,20 +119,19 @@ RASPBERRY_PI_CONNECTED = rpi.is_raspberry_pi()
|
|||||||
# Queue creation
|
# Queue creation
|
||||||
print_queue = PrintQueue(app)
|
print_queue = PrintQueue(app)
|
||||||
|
|
||||||
# Web & API routes
|
# Web & API management
|
||||||
web = Web(app, print_queue)
|
web = Web(app, print_queue)
|
||||||
|
|
||||||
# Start worker thread
|
# Start worker thread
|
||||||
worker = PrintWorker(app, print_queue, printer, socketio)
|
worker = PrintWorker(app, print_queue, printer, socketio)
|
||||||
worker.start()
|
worker.start()
|
||||||
|
|
||||||
|
# The rate limit
|
||||||
limiter = Limiter(
|
limiter = Limiter(
|
||||||
get_remote_address, app=app, default_limits=["1500 per day", "500 per hour"]
|
get_remote_address, app=app, default_limits=["1500 per day", "500 per hour"]
|
||||||
)
|
)
|
||||||
|
|
||||||
# General routes
|
# General routes
|
||||||
|
|
||||||
|
|
||||||
@app.route("/")
|
@app.route("/")
|
||||||
@limiter.limit("1/second", override_defaults=False)
|
@limiter.limit("1/second", override_defaults=False)
|
||||||
def index():
|
def index():
|
||||||
@@ -153,8 +149,6 @@ def webcam():
|
|||||||
|
|
||||||
|
|
||||||
# Form treatement
|
# Form treatement
|
||||||
|
|
||||||
|
|
||||||
@app.route("/web/print/sms", methods=["POST"])
|
@app.route("/web/print/sms", methods=["POST"])
|
||||||
@limiter.limit("6/minute", override_defaults=False)
|
@limiter.limit("6/minute", override_defaults=False)
|
||||||
def web_print_sms():
|
def web_print_sms():
|
||||||
@@ -185,7 +179,7 @@ def web_print_sms():
|
|||||||
return redirect(url_for("index"))
|
return redirect(url_for("index"))
|
||||||
|
|
||||||
# end try
|
# 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"))
|
return redirect(url_for("index"))
|
||||||
|
|
||||||
|
|
||||||
@@ -227,7 +221,7 @@ def web_print_img():
|
|||||||
flash("The image could not be printed because : " + str(e), "error")
|
flash("The image could not be printed because : " + str(e), "error")
|
||||||
return redirect(url_for("index"))
|
return redirect(url_for("index"))
|
||||||
|
|
||||||
flash("Picture printed !", "info")
|
flash("Picture added to the print queue !", "info")
|
||||||
return redirect(url_for("index"))
|
return redirect(url_for("index"))
|
||||||
|
|
||||||
|
|
||||||
@@ -296,8 +290,7 @@ def api_print_image():
|
|||||||
return "No image submitted", 400
|
return "No image submitted", 400
|
||||||
|
|
||||||
file = request.files["img"]
|
file = request.files["img"]
|
||||||
# If the user does not select a file, the browser submits an
|
# If the user submits an empty file without a filename.
|
||||||
# empty file without a filename.
|
|
||||||
if file.filename == "":
|
if file.filename == "":
|
||||||
app.logger.error("Submitted file has no filename !")
|
app.logger.error("Submitted file has no filename !")
|
||||||
return "Submitted file has no filename !", 400
|
return "Submitted file has no filename !", 400
|
||||||
@@ -337,12 +330,18 @@ def api_worker_state():
|
|||||||
|
|
||||||
@app.route("/api/worker/start")
|
@app.route("/api/worker/start")
|
||||||
def api_worker_start():
|
def api_worker_start():
|
||||||
|
"""
|
||||||
|
Enable to worker. This starts to process the print queue.
|
||||||
|
"""
|
||||||
worker.start_worker()
|
worker.start_worker()
|
||||||
return jsonify(worker.current_state())
|
return jsonify(worker.current_state())
|
||||||
|
|
||||||
|
|
||||||
@app.route("/api/worker/stop")
|
@app.route("/api/worker/stop")
|
||||||
def api_worker_stop():
|
def api_worker_stop():
|
||||||
|
"""
|
||||||
|
Stops the print queue. This stops the processing of the print queue.
|
||||||
|
"""
|
||||||
worker.stop_worker()
|
worker.stop_worker()
|
||||||
return jsonify(worker.current_state())
|
return jsonify(worker.current_state())
|
||||||
|
|
||||||
|
|||||||
@@ -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 io # To check if we are on a Raspberry Pi
|
||||||
import subprocess
|
import subprocess
|
||||||
import os
|
import os
|
||||||
@@ -5,6 +13,7 @@ from time import sleep, gmtime, strftime
|
|||||||
from flask_socketio import SocketIO
|
from flask_socketio import SocketIO
|
||||||
from gpiozero import Button, LED, DigitalOutputDevice
|
from gpiozero import Button, LED, DigitalOutputDevice
|
||||||
from PIL import Image
|
from PIL import Image
|
||||||
|
from task import TextTask, ImageTask, CutTask
|
||||||
|
|
||||||
class Raspberry():
|
class Raspberry():
|
||||||
"""
|
"""
|
||||||
@@ -18,7 +27,7 @@ class Raspberry():
|
|||||||
|
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
printer,
|
print_queue,
|
||||||
app,
|
app,
|
||||||
socketio,
|
socketio,
|
||||||
button_gpio_port_number,
|
button_gpio_port_number,
|
||||||
@@ -26,7 +35,7 @@ class Raspberry():
|
|||||||
flash_gpio_port_number,
|
flash_gpio_port_number,
|
||||||
is_flash_present,
|
is_flash_present,
|
||||||
):
|
):
|
||||||
self.printer = printer
|
self.print_queue = print_queue
|
||||||
self.socketio = socketio
|
self.socketio = socketio
|
||||||
self.app = app
|
self.app = app
|
||||||
|
|
||||||
@@ -37,9 +46,13 @@ class Raspberry():
|
|||||||
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):
|
||||||
|
"""
|
||||||
|
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
|
# 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", encoding="utf-8") as cpuinfo:
|
||||||
found = False
|
found = False
|
||||||
for line in cpuinfo:
|
for line in cpuinfo:
|
||||||
if line.startswith("Hardware"):
|
if line.startswith("Hardware"):
|
||||||
@@ -140,7 +153,7 @@ class Raspberry():
|
|||||||
f"Unable to take a picture. Error: {e.stderr.decode()}"
|
f"Unable to take a picture. Error: {e.stderr.decode()}"
|
||||||
)
|
)
|
||||||
return False
|
return False
|
||||||
except Exception as e:
|
except RuntimeError as e:
|
||||||
# Catch any unexpected errors
|
# Catch any unexpected errors
|
||||||
self.app.logger.error(f"Unexpected error while taking picture: {str(e)}")
|
self.app.logger.error(f"Unexpected error while taking picture: {str(e)}")
|
||||||
return False
|
return False
|
||||||
@@ -197,7 +210,7 @@ class Raspberry():
|
|||||||
output_path = image_path # Overwrite the original image if no output path is given
|
output_path = image_path # Overwrite the original image if no output path is given
|
||||||
image.save(output_path)
|
image.save(output_path)
|
||||||
|
|
||||||
except Exception as e:
|
except RuntimeError as e:
|
||||||
self.app.logger.error(f"Error overlaying logo: {e}")
|
self.app.logger.error(f"Error overlaying logo: {e}")
|
||||||
return False
|
return False
|
||||||
|
|
||||||
@@ -224,7 +237,7 @@ class Raspberry():
|
|||||||
|
|
||||||
image.save(output_path)
|
image.save(output_path)
|
||||||
|
|
||||||
except Exception as e:
|
except RuntimeError as e:
|
||||||
self.app.logger.error(f"Error cropping image to square: {e}")
|
self.app.logger.error(f"Error cropping image to square: {e}")
|
||||||
return False
|
return False
|
||||||
|
|
||||||
@@ -241,7 +254,7 @@ class Raspberry():
|
|||||||
try:
|
try:
|
||||||
self.flash.on()
|
self.flash.on()
|
||||||
self.take_picture()
|
self.take_picture()
|
||||||
except Exception as e:
|
except RuntimeError as e:
|
||||||
self.app.logger.error(
|
self.app.logger.error(
|
||||||
"Could not take a picture after the button press : " + str(e)
|
"Could not take a picture after the button press : " + str(e)
|
||||||
)
|
)
|
||||||
@@ -250,15 +263,11 @@ class Raspberry():
|
|||||||
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.print_queue.enqueue(ImageTask(self.image_path,signature="",process=True))
|
||||||
self.printer.print_img(self.image_path, process=True)
|
self.print_queue.enqueue(TextTask(content="Imprimé par LittlePrynter", signature=""))
|
||||||
self.printer.print_sms("")
|
time = strftime("%Y-%m-%d %H:%M", gmtime())
|
||||||
self.printer.print_sms("With Love From Société.Vide", signature="", bold=True)
|
self.print_queue.enqueue(TextTask(content=time, signature=""))
|
||||||
self.printer.print_sms("Printed by LittlePrynter", signature="")
|
self.print_queue.enqueue(CutTask())
|
||||||
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.led.off()
|
self.led.off()
|
||||||
self.app.logger.debug("Done printing picture")
|
self.app.logger.debug("Added a photomaton picture to the print queue")
|
||||||
return True
|
return True
|
||||||
|
|||||||
@@ -6,7 +6,8 @@ from task import TextTask, ImageTask, CutTask
|
|||||||
|
|
||||||
|
|
||||||
class Web(object):
|
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):
|
def __init__(self, app, print_queue):
|
||||||
super(Web).__init__()
|
super(Web).__init__()
|
||||||
|
|||||||
Reference in New Issue
Block a user