17 Commits

Author SHA1 Message Date
n07070
3769e17444 Lint project 2026-02-02 12:37:15 +01:00
n07070
98299cc281 Add information on contributions 2026-01-04 12:08:05 +01:00
n07070
d261bb0fdc Add 3d models of the printer and raspberry pi 2026-01-02 17:40:29 +01:00
n07070
46fda975d4 Ajout des logos de extase club 2025-11-13 16:59:52 +01:00
nono
670ab495d8 Update code to better handle the raspberry pi mode 2025-10-17 17:33:53 +02:00
nono
34bffe93af Add systemd service 2025-10-17 17:33:37 +02:00
nono
8ecb93656c Change signature 2025-10-16 18:39:29 +02:00
nono
eac737b889 Remove flash from the printer class, update web class 2025-10-16 18:36:16 +02:00
nono
dfeb1be0f0 update raspberry pi code 2025-10-16 18:35:57 +02:00
nono
0eefafdeb2 Add configuration ports for the flash 2025-10-16 18:35:35 +02:00
nono
ce70d498ca Add requirements 2025-10-16 18:35:27 +02:00
nono
991f6794c7 Update readme with more deps 2025-10-16 18:35:16 +02:00
nono
2b8a57b14a Update gitignore 2025-06-24 23:28:24 +02:00
nono
117f069351 Rework the webcam detection 2025-06-24 23:27:28 +02:00
nono
981eba100d Update little printer requirements 2025-06-23 23:08:32 +02:00
nono
cf557c89cd Add configuration section for the rapsberry pi 2025-06-19 10:52:15 +02:00
nono
2cc0b40b90 Ajout de quelques changements pour acces libre 2025-06-18 11:57:34 +02:00
21 changed files with 578 additions and 3031 deletions

View File

@@ -19,56 +19,44 @@ To make this project work, you will need :
- Some electric wires. - Some electric wires.
- Some knowledge of the command line, - Some knowledge of the command line,
- Some knowledge of Python. - Some knowledge of Python.
- You will need to install [Poetry](https://python-poetry.org/) to manage the depedencies of the projet.
- 3h of your time, 5h if things need debugging. - 3h of your time, 5h if things need debugging.
- You will need to following packages: `fswebcam`, `libjpeg-dev` ,`zlib1g-dev`,`libffi-dev`,`git`, `virtualenv`,`pip` and `python` >= 3.8.6. - `fswebcam`, `libjpeg-dev` ,`zlib1g-dev`,`libffi-dev`,`git`, `virtualenv`,`pip` and `python` >= 3.8.6.
- A webcam for the webcam page to work. Will work on a smartphone. Not required. - A webcam for the webcam page to work. Will work on a smartphone. Not required.
## Context ## Context
### Testing your printer
For the EPSON TM-T20III, you can get the CUPS driver from [here](https://download.epson-biz.com/modules/pos/index.php?page=soft&scat=32). It's not specific to the printer I've been using, so you can try with other printers. For the EPSON TM-T20III, you can get the CUPS driver from [here](https://download.epson-biz.com/modules/pos/index.php?page=soft&scat=32). It's not specific to the printer I've been using, so you can try with other printers.
With that, you can try out your printer and print normal text, images or pdf documents for example. For the Adafruit printer, start by following the guide [here](https://learn.adafruit.com/networked-thermal-printer-using-cups-and-raspberry-pi) to install the CUPS software needed to print images. If you want, you can install it via the command line, [following this guide](https://help.ubuntu.com/lts/serverguide/cups.html). With that, you can try out your printer and print normal text, images or pdf documents for example. LittlePrynter itself does not require any other software than the ones installed with it, i.e the ones listed in the `requirements.txt` file.
For the Adafruit printer, start by following the guide [here](https://learn.adafruit.com/networked-thermal-printer-using-cups-and-raspberry-pi) to install the CUPS software needed to print images. If you want, you can install it via the command line, [following this guide](https://help.ubuntu.com/lts/serverguide/cups.html).
You can also get some information from [here](https://learn.adafruit.com/mini-thermal-receipt-printer) and [here](https://learn.adafruit.com/instant-camera-using-raspberry-pi-and-thermal-printer) if you're stuck. You can also get some information from [here](https://learn.adafruit.com/mini-thermal-receipt-printer) and [here](https://learn.adafruit.com/instant-camera-using-raspberry-pi-and-thermal-printer) if you're stuck.
### How LittlePrynter works
LittlePrynter itself does not require any other software than the ones installed with it, i.e the ones listed in the `pyproject.toml` file.
The version for the EPSON TM-T20III uses a library called `python-escpos`, which doesn't need a CUPS driver or anything else. It's included in the Python dependencies. The version for the EPSON TM-T20III uses a library called `python-escpos`, which doesn't need a CUPS driver or anything else. It's included in the Python dependencies.
The project only supports the EPSON printer, but you can try to adapt it for other printers. For example, using `adafruit-thermal`, or `brother-ql`. The project only supports the EPSON printer, but you can try to adapt it for other printers using the `adafruit-thermal` branch, but I won't offer support for it.
### Install & setup the project : ### Install & setup the project :
Theses commands will copy the software on your computer, go into the directory, then activate a virtual environnement and install all of the project's dependecies.
``` ```
$ git clone https://git.n07070.xyz/n07070/LittlePrynter $ git clone https://git.n07070.xyz/n07070/LittlePrynter
$ virtualenv LittlePrynter
$ cd LittlePrynter $ cd LittlePrynter
$ eval "$(poetry env activate)" $ source bin/activate
$ poetry install $ pip install -r requirements.txt
``` ```
> tip : when you're done, you can get out of the poetry environnement either by closing your terminal, or by running `deactivate`. > tip : when you're done, you can get out of the virtualenv either by closing your terminal, or by running `deactivate`.
### Configure LittlePrynter
You should see a folder named `configuration`. Enter it, and duplicate the file named `config.toml.sample`, and rename the copy to `config.toml`. Now, edit this file by following the comments in the file itself. You should see a folder named `configuration`. Enter it, and duplicate the file named `config.toml.sample`, and rename the copy to `config.toml`. Now, edit this file by following the comments in the file itself.
You should also setup the proper `udev` permissions to access the printers via USB, following the configuration found [here](https://python-escpos.readthedocs.io/en/latest/user/installation.html#setup-udev-for-usb-printers).
You can now start the web server with You can now start the web server with
``` ```
$ export FLASK_APP=src/main.py $ export FLASK_APP=src/main.py
$ flask run --cert=adhoc $ flask run
``` ```
The `--cert=adhoc` argument will make LittlePrinter accessible in HTTPS with a [self-signed SSL certificate](https://en.wikipedia.org/wiki/Self-signed_certificate). This provides a free and easy way to use HTTPS, but does not provide any trust value. It is, on the other hand, a good way to access the webcam in developpement mode or on a local network, because a HTTPS connexion is required by browsers to access the needed APIs.
This command should start a web server with which you can test your configuration. If you plan on exposing your printer to the Internet, and give it an IP / URL, _please, please, please_, don't run it this way. Look at Flask's documentation and read about running a production server. It's a little more work, but it will prevent your computer/server being hacked in too easily. This command should start a web server with which you can test your configuration. If you plan on exposing your printer to the Internet, and give it an IP / URL, _please, please, please_, don't run it this way. Look at Flask's documentation and read about running a production server. It's a little more work, but it will prevent your computer/server being hacked in too easily.
Voilà ! Voilà !
@@ -79,34 +67,6 @@ 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
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.
To do so, you can run the following command, which will modify your files to be in a certain coding style ;
```
black src/
```
Beware that this command *will* re-write files, so doing `git add <file>` and then `black src/` and then `git diff` to see what the linter has done is a good idea.
Then, you can run another command, called Pylint ( [documentation](https://www.pylint.org/) ) which will rate your code. Try to get 10/10 : an excellent code works better and make you a better programmer !
```
pylint src/
```
## Screenshots ## Screenshots
![](src/static/images/homepage.png) ![](src/static/images/homepage.png)
@@ -129,7 +89,6 @@ If you liked this project, feel free to support my work !
- [Github repo with CUPS drivers for the Adafruit Thermal Printer ( zj-58 )](https://github.com/klirichek/zj-58) - [Github repo with CUPS drivers for the Adafruit Thermal Printer ( zj-58 )](https://github.com/klirichek/zj-58)
- [A link to buy one in Europe](https://rlx.sk/sk/various-boards/1829-mini-thermal-receipt-printer-adafruit-597.html) - [A link to buy one in Europe](https://rlx.sk/sk/various-boards/1829-mini-thermal-receipt-printer-adafruit-597.html)
- [Another link to buy one, direct from factory](https://www.cashinotech.com/csn-a2-58mm-mini-panel-thermal-receipt-printer_p11.html) - [Another link to buy one, direct from factory](https://www.cashinotech.com/csn-a2-58mm-mini-panel-thermal-receipt-printer_p11.html)
- [Diagnostickoeur](https://framagit.org/stickoeur/diagnostickoeur), another printer software mainly around Brother QL printers.
## Licence ## Licence

View File

@@ -1,8 +1,5 @@
# Configuration file the LittlePrynter # Configuration file the LittlePrynter
[defaults]
signature = "Anonymous"
# Printer settings # Printer settings
[printer] [printer]
vendor_id = 0x04b8 vendor_id = 0x04b8

View File

@@ -1,8 +1,5 @@
# Configuration file the LittlePrynter # Configuration file the LittlePrynter
[defaults]
signature = "Anonymous"
# Printer settings # Printer settings
[printer] [printer]
vendor_id = "0x04b8" vendor_id = "0x04b8"

View File

@@ -6,7 +6,7 @@ After=network.target
Type=simple Type=simple
WorkingDirectory=/home/pi/littleprynter/ WorkingDirectory=/home/pi/littleprynter/
Environment=FLASK_APP=src/main.py Environment=FLASK_APP=src/main.py
ExecStart=/home/pi/littleprynter/bin/flask run --host 0.0.0.0 --no-reload ExecStart=/home/pi/littleprynter/bin/flask run --host 0.0.0.0 --debug --no-reload
[Install] [Install]
WantedBy=multi-user.target WantedBy=multi-user.target

1
pip-selfcheck.json Normal file
View File

@@ -0,0 +1 @@
{"last_check":"2018-10-10T18:33:22Z","pypi_version":"18.1"}

1886
poetry.lock generated

File diff suppressed because it is too large Load Diff

View File

@@ -1,56 +0,0 @@
[project]
name = "littleprynter"
version = "0.1.0"
description = "A web interface and API to print on thermal printers"
authors = [
{name = "n07070",email = "contact@n07070.xyz"}
]
license = "AGPLv3"
readme = "README.md"
requires-python = ">=3.13"
dependencies = [
"flask (>=3.1.3,<4.0.0)",
"numpy (>=2.3.4)",
"toml (>=0.10.2,<0.11.0)",
"flask-socketio (>=5.6.1,<6.0.0)",
"flask-limiter (>=4.1.1,<5.0.0)",
"gpiozero (>=2.0.1)",
"cryptography (>=48.0.0,<49.0.0)",
"pyusb (>=1.3.1,<2.0.0)",
"pyserial (>=3.5,<4.0)",
"qrcode (<9.0)",
"python-barcode (>=0.16.1)",
"setuptools (==81)",
"cffi (>=2.0.0,<3.0.0)",
"configobj (>=5.0.9,<6.0.0)",
"future (>=1.0.0,<2.0.0)",
"h11 (>=0.16.0,<0.17.0)",
"idna (>=3.15,<4.0)",
"itsdangerous (>=2.2.0,<3.0.0)",
"jinja2 (>=3.1.6,<4.0.0)",
"jsonpointer (>=3.1.1,<4.0.0)",
"jsonschema (>=4.26.0,<5.0.0)",
"limits (>=5.8.0,<6.0.0)",
"markupsafe (>=3.0.3,<4.0.0)",
"pillow (>=12.2.0,<13.0.0)",
"pycryptodomex (>=3.23.0,<4.0.0)",
"python-dateutil (>=2.9.0.post0,<3.0.0)",
"python-engineio (>=4.13.1,<5.0.0)",
"python-socketio (>=5.16.1,<6.0.0)",
"pyyaml (>=6.0.3,<7.0.0)",
"requests (>=2.34.2,<3.0.0)",
"python-escpos (>=3.1,<4.0)",
"pep8 (>=1.7.1,<2.0.0)",
"pylint (>=4.0.5,<5.0.0)",
"brother-ql-inventree (>=1.3,<2.0)",
]
[tool.poetry]
package-mode = false
[tool.poetry.dependencies]
python-escpos = {allow-prereleases = true}
[build-system]
requires = ["poetry-core>=2.0.0,<3.0.0"]
build-backend = "poetry.core.masonry.api"

261
requirements.txt Normal file
View File

@@ -0,0 +1,261 @@
appdirs==1.4.4
apt-listchanges==4.8
argcomplete==3.6.2
arrow==1.3.0
attrs==25.3.0
babel==2.17.0
bcrypt==4.2.0
bidict==0.23.1
blinker==1.9.0
certifi==2025.1.31
cffi==1.17.1
chardet==5.2.0
charset-normalizer==3.4.2
click==8.3.0
cloud-init==25.2
colorzero==2.0
configobj==5.0.9
cryptography==43.0.0
Deprecated==1.2.18
distlib==0.3.9
distro==1.9.0
escpos==2.0.0
filelock==3.18.0
Flask==3.1.2
Flask-Limiter==4.0.0
Flask-SocketIO==5.5.1
fqdn==1.5.1
future==1.0.0
gpiod==2.2.0
gpiozero==2.0.1
h11==0.16.0
idna==3.10
isoduration==20.11.0
itsdangerous==2.2.0
Jinja2==3.1.6
jsonpatch==1.32
jsonpointer==2.4
jsonschema==4.19.2
jsonschema-specifications==2023.12.1
lgpio==0.2.2.0
limits==5.6.0
linkify-it-py==2.0.3
markdown-it-py==3.0.0
MarkupSafe==2.1.5
mdurl==0.1.2
numpy==2.3.4
oauthlib==3.2.2
olefile==0.47
ordered-set==4.1.0
packaging==25.0
pillow==11.1.0
platformdirs==4.3.7
ply==3.11
pycparser==2.22
pycryptodomex==3.20.0
Pygments==2.18.0
PyJWT==2.10.1
pyserial==3.5
python-apt==3.0.0
python-barcode==0.16.1
python-dateutil==2.9.0
python-engineio==4.12.3
python-socketio==5.14.1
pyusb==1.3.1
PyYAML==6.0.2
qrcode==8.2
referencing==0.36.2
requests==2.32.3
rfc3339-validator==0.1.4
rfc3986-validator==0.1.1
rfc3987==1.3.8
rich==13.9.4
rpds-py==0.21.0
rpi-keyboard-config==1.0
rpi-lgpio==0.6
setuptools==80.9.0
simple-websocket==1.1.0
six==1.17.0
smbus2==0.4.3
spidev==3.6
ssh-import-id==5.10
toml==0.10.2
types-aiofiles==24.1
types-antlr4-python3-runtime==4.13
types-assertpy==1.1
types-atheris==2.3
types-aws-xray-sdk==2.14
types-beautifulsoup4==4.12
types-bleach==6.2
types-boltons==24.1
types-braintree==4.31
types-cachetools==5.5
types-caldav==1.3
types-capturer==3.0
types-cffi==1.16
types-chevron==0.14
types-click-default-group==1.2
types-click-spinner==0.1
types-colorama==0.4
types-commonmark==0.9
types-console-menu==0.8
types-corus==0.10
types-croniter==5.0.1
types-dateparser==1.2
types-decorator==5.1
types-defusedxml==0.7
types-Deprecated==1.2.15
types-docker==7.1
types-dockerfile-parse==2.0
types-docutils==0.21
types-editdistance==0.8
types-entrypoints==0.4
types-ExifRead==3.0
types-fanstatic==1.4
types-first==2.0
types-flake8==7.1
types-flake8-bugbear==24.12.12
types-flake8-builtins==2.5
types-flake8-docstrings==1.7
types-flake8-rst-docstrings==0.3
types-flake8-simplify==0.21
types-flake8-typing-imports==1.16
types-Flask-Cors==5.0
types-Flask-Migrate==4.0
types-Flask-SocketIO==5.4
types-fpdf2==2.8.2
types-gdb==15.0
types-gevent==24.11
types-google-cloud-ndb==2.3
types-greenlet==3.1
types-hdbcli==2.23
types-html5lib==1.1
types-httplib2==0.22
types-humanfriendly==10.0
types-hvac==2.3
types-ibm-db==3.2.4
types-icalendar==6.1
types-influxdb-client==1.45
types-inifile==0.4
types-JACK-Client==0.5
types-Jetson.GPIO==2.1
types-jmespath==1.0
types-jsonschema==4.23
types-jwcrypto==1.5
types-keyboard==0.13
types-ldap3==2.9
types-libsass==0.23
types-lupa==2.2
types-lzstring==1.0
types-m3u8==6.0
types-Markdown==3.7
types-mock==5.1
types-mypy-extensions==1.0
types-mysqlclient==2.2
types-nanoid==2.0.0
types-netaddr==1.3
types-netifaces==0.11
types-networkx==3.4.2
types-oauthlib==3.2
types-objgraph==3.6
types-olefile==0.47
types-openpyxl==3.1.5
types-opentracing==2.4
types-paramiko==3.5
types-parsimonious==0.10
types-passlib==1.7
types-passpy==1.0
types-peewee==3.17.8
types-pep8-naming==0.14
types-pexpect==4.9
types-pika-ts==1.3
types-polib==1.2
types-portpicker==1.6
types-protobuf==5.29.1
types-psutil==6.1
types-psycopg2==2.9.10
types-pyasn1==0.6
types-pyaudio==0.2
types-PyAutoGUI==0.9
types-pycocotools==2.0
types-pycurl==7.45.4
types-pyfarmhash==0.4
types-pyflakes==3.2
types-pygit2==1.15
types-Pygments==2.18
types-pyinstaller==6.11
types-pyjks==20.0
types-PyMySQL==1.1
types-pynput==1.7.7
types-pyOpenSSL==24.1
types-pyRFC3339==2.0.1
types-PyScreeze==1.0.1
types-pyserial==3.5
types-pysftp==0.2
types-pytest-lazy-fixture==0.6
types-python-crontab==3.2
types-python-datemath==3.0.1
types-python-dateutil==2.9
types-python-http-client==3.3.7
types-python-jenkins==1.8
types-python-jose==3.3
types-python-nmap==0.7
types-python-xlib==0.33
types-pytz==2024.2
types-pywin32==308
types-pyxdg==0.28
types-PyYAML==6.0
types-qrbill==1.1
types-qrcode==8.0
types-regex==2024.11.6
types-reportlab==4.2.5
types-requests==2.32
types-requests-oauthlib==2.0
types-retry==0.9
types-RPi.GPIO==0.7
types-s2clientprotocol==5
types-seaborn==0.13.2
types-Send2Trash==1.8
types-setuptools==75.6
types-shapely==2.0
types-simplejson==3.19
types-singledispatch==4.1
types-six==1.17
types-slumber==0.7
types-str2bool==1.1
types-tabulate==0.9
types-tensorflow==2.18.0
types-TgCrypto==1.2
types-toml==0.10
types-toposort==1.10
types-tqdm==4.67
types-translationstring==1.4
types-tree-sitter-languages==1.10
types-ttkthemes==3.2
types-ujson==5.10
types-unidiff==0.7
types-untangle==1.2
types-usersettings==1.1
types-uWSGI==2.0
types-vobject==0.9.9
types-waitress==3.0.1
types-WebOb==1.8
types-whatthepatch==1.0
types-workalendar==17.0
types-WTForms==3.2.1
types-wurlitzer==3.1
types-xdgenvpy==3.0
types-xmltodict==0.14
types-zstd==1.5
types-zxcvbn==4.4
typing_extensions==4.15.0
uc-micro-py==1.0.3
uritemplate==4.1.1
urllib3==2.3.0
viivakoodi==0.8.0
virtualenv==20.31.2
webcolors==1.13
Werkzeug==3.1.3
wheel==0.46.1
wrapt==1.17.3
wsproto==1.2.0

7
run.sh Executable file
View File

@@ -0,0 +1,7 @@
virtualenv .
pip install -r requirements.txt
export FLASK_APP=src/main.py
export FLASK_ENV=development
flask run --host 192.168.0.42 --debugger --eager-loading

View File

@@ -39,15 +39,12 @@ from flask_limiter.util import get_remote_address
from printer import Printer # The wrapper for the printer class from printer import Printer # The wrapper for the printer class
from raspberry import Raspberry # The Raspberry pi control Class from raspberry import Raspberry # The Raspberry pi control Class
from web import Web # Wrapper for the web routes and API from web import Web # Wrapper for the web routes and API
from print_queue import PrintQueue
from worker import PrintWorker
# We create the main Flask object # Variables
app = Flask(__name__) app = Flask(__name__)
socketio = SocketIO(app, cors_allowed_origins="*") socketio = SocketIO(app)
ALLOWED_EXTENSIONS = {"png", "jpg", "jpeg", "gif"}
# Global variables
ALLOWED_EXTENSIONS = {"png", "jpg", "jpeg", "gif", "webp"}
# Load the configuration file # Load the configuration file
try: try:
@@ -85,10 +82,9 @@ except FileExistsError:
app.logger.debug("Directory %s already exists.", UPLOAD_FOLDER) app.logger.debug("Directory %s already exists.", UPLOAD_FOLDER)
except PermissionError: except PermissionError:
app.logger.error("Permission denied: Unable to create %s", UPLOAD_FOLDER) app.logger.error("Permission denied: Unable to create %s", UPLOAD_FOLDER)
sys.exit(77)
# Output the config file # Output the config file
if os.getenv("FLASK_DEBUG"): if os.getenv("LIPY_DEBUG") is True:
pprint.pprint(configuration_file) pprint.pprint(configuration_file)
# We define the app module used by Flask # We define the app module used by Flask
@@ -116,22 +112,19 @@ rpi = Raspberry(
RASPBERRY_PI_CONNECTED = rpi.is_raspberry_pi() RASPBERRY_PI_CONNECTED = rpi.is_raspberry_pi()
# Queue creation
print_queue = PrintQueue(app)
# Web & API management # Web & API routes
web = Web(app, print_queue)
# Start worker thread web = Web(app, printer)
worker = PrintWorker(app, print_queue, printer, socketio)
worker.start() if __name__ == "__main__":
app.run(debug=True, use_reloader=False, host="0.0.0.0", ssl_context="adhoc")
# 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
@app.route("/") @app.route("/")
@limiter.limit("1/second", override_defaults=False) @limiter.limit("1/second", override_defaults=False)
def index(): def index():
@@ -148,83 +141,6 @@ def webcam():
return render_template("webcam.html") return render_template("webcam.html")
# Form treatement
@app.route("/web/print/sms", methods=["POST"])
@limiter.limit("6/minute", override_defaults=False)
def web_print_sms():
"""Prints a short message on a printer"""
app.logger.debug("Printing an sms via web method")
try:
txt = request.form["txt"]
except werkzeug.exceptions.BadRequestKeyError as e:
app.logger.error("Whoops, we are missing the txt input field. : %s ", str(e))
flash("Whoops, no forms submitted or missing signature : " + str(e), "error")
return redirect(url_for("index"))
try:
# comment: We try to get a signature
sign = request.form["signature"]
except werkzeug.exceptions.BadRequestKeyError as e:
app.logger.warning(
"No signature found for this print, using default signature : %s ", str(e)
)
sign = configuration_file["defaults"]["signature"]
try:
web.print_sms(txt, sign)
except RuntimeError as e:
app.logger.error("Whoops, we could not print an SMS because : %s ", str(e))
flash("Whoops, we could not print an SMS because :" + str(e), "error")
return redirect(url_for("index"))
# end try
flash("The SMS has been added to the print queue !", "info")
return redirect(url_for("index"))
@app.route("/web/print/img", methods=["POST"])
@limiter.limit("1/second", override_defaults=False)
def web_print_img():
"""Prints an image on a printer"""
app.logger.debug("Printing an image with web method")
try:
# comment: We try to get a signature
sign = request.form["signature"]
except werkzeug.exceptions.BadRequestKeyError as e:
app.logger.warning(
"No signature found for this print, using default signature : %s", str(e)
)
sign = configuration_file["defaults"]["signature"]
# check if the post request has the file part
if "img" not in request.files:
app.logger.error("Whoops, no images submitted : %s ", str(e))
app.logger.error("Error getting the files : %s", str(e))
flash("Whoops, no images submitted : " + str(e), "error")
return redirect(url_for("index"))
file = request.files["img"]
# If the user does not select a file, the browser submits an
# empty file without a filename.
if file.filename == "":
app.logger.error("Submitted file has no filename !")
flash("Submitted file has no filename !", "error")
return redirect(url_for("index"))
try:
app.logger.debug("Sending the image to the printer.")
web.print_image(file, sign)
except RuntimeError as e:
app.logger.error("The image could not be printed because : %s ", str(e))
flash("The image could not be printed because : " + str(e), "error")
return redirect(url_for("index"))
flash("Picture added to the print queue !", "info")
return redirect(url_for("index"))
# API routes # API routes
# The api has the following methods # The api has the following methods
# api/print/{sms,img,letter,qr,barcode} # api/print/{sms,img,letter,qr,barcode}
@@ -246,115 +162,85 @@ def api_index():
@limiter.limit("6/minute", override_defaults=False) @limiter.limit("6/minute", override_defaults=False)
def api_print_sms(): def api_print_sms():
"""Prints a short message on a printer""" """Prints a short message on a printer"""
app.logger.debug("Printing an sms via API") app.logger.debug("Printing an sms")
try: try:
txt = request.form["txt"] txt = request.form["txt"]
except werkzeug.exceptions.BadRequestKeyError as e:
app.logger.error("Whoops, we are missing the txt input field. : %s ", str(e))
return str(e), 400
try:
# comment: We try to get a signature
sign = request.form["signature"] sign = request.form["signature"]
except werkzeug.exceptions.BadRequestKeyError as e: except werkzeug.exceptions.BadRequestKeyError as e:
app.logger.warning( app.logger.error(
"No signature found for this print, using default signature. : %s", str(e) "Whoops, no forms submitted or missing signature : %s ", str(e)
) )
sign = configuration_file["defaults"]["signature"] flash("Whoops, no forms submitted or missing signature : %s", str(e))
try: return redirect(url_for("index"))
# comment: We try to print the SMS
web.print_sms(txt, sign) web.print_sms(txt, sign)
except RuntimeError as e: return redirect(url_for("index"))
return str(e), 500
# end try
return "OK", 200
@app.route("/api/print/img", methods=["POST"]) @app.route("/api/print/img", methods=["POST"])
@limiter.limit("6/minute", override_defaults=False) @limiter.limit("6/minute", override_defaults=False)
def api_print_image(): def api_print_image():
"""Prints an image on a printer""" """Prints an image on a printer"""
app.logger.debug("Printing an image via API") app.logger.debug("Printing an image")
try: try:
# comment: We try to get a signature
sign = request.form["signature"] sign = request.form["signature"]
except werkzeug.exceptions.BadRequestKeyError as e: except Exception as e:
app.logger.warning( app.logger.error("Whoops, no forms submitted or missing signature : %s", str(e))
"No signature found for this print, using default signature. %s", str(e) flash("Whoops, no forms submitted or missing signature : %s ", str(e))
) return redirect(url_for("index"))
sign = configuration_file["defaults"]["signature"]
if request.method == "POST":
# check if the post request has the file part # check if the post request has the file part
if "img" not in request.files: if "img" not in request.files:
app.logger.error("Whoops, no images submitted.") app.logger.error("Whoops, no images submitted : %s ", str(e))
return "No image submitted", 400 app.logger.error("Error getting the files : %s", str(e))
flash("Whoops, no images submitted : %s", str(e))
return redirect(url_for("index"))
file = request.files["img"] file = request.files["img"]
# If the user submits an empty file without a filename. # If the user does not select a file, the browser submits an
# 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 flash("Submitted file has no filename !")
return redirect(url_for("index"))
try: try:
app.logger.debug("Sending the image to the printer.") app.logger.debug("Sending the image to the printer.")
web.print_image(file, sign) web.print_image(file, sign)
except RuntimeError as e: except Exception as e:
return str(e), 500 app.logger.error("The image could not be printed because : %s ", str(e))
flash("The image could not be printed because : %s ", str(e))
return redirect(url_for("index"))
return "OK", 200 else:
app.logger.error("Method not allowed")
flash("Method not allowed")
return redirect(url_for("index"))
flash("Picture printed ! ")
return redirect(url_for("index"))
@app.route("/api/camera/picture", methods=["GET"]) @app.route("/api/camera/picture")
def camera_picture(): def camera_picture():
"""Returns a picture taken by the camera on a raspberry pi""" """Returns a picture taken by the camera"""
if RASPBERRY_PI_CONNECTED: if RASPBERRY_PI_CONNECTED:
try: try:
return rpi.camera_picture() return rpi.camera_picture()
except RuntimeError as e: except Exception as e:
return jsonify({"message": "Error getting the stream : " + e}), 500 return jsonify({"message": "Error getting the stream : " + e}), 500
else: else:
return jsonify({"message": "No camera present"}), 500 return jsonify({"message": "No camera present"}), 500
@app.route("/api/queue", methods=["GET"])
def api_queue_status():
"""API endpoint for entire queue"""
return jsonify(web.get_queue_state())
@app.route("/api/worker", methods=["GET"])
def api_worker_state():
"""API endpoint to get the worker state"""
return jsonify(worker.current_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())
## Authentification
@app.route("/login") @app.route("/login")
@limiter.limit("1/second", override_defaults=False) @limiter.limit("1/second", override_defaults=False)
def login_page(): def login_page():
"""Unsued, logins""" """Unsued, logins"""
# web.login(username,password) # web.login(username,password)
return redirect(url_for("index")), 501 return redirect(url_for("index"))
@app.route("/logout") @app.route("/logout")
@@ -362,7 +248,7 @@ def login_page():
def logout_page(): def logout_page():
"""Unused, logout""" """Unused, logout"""
# web.logout(username, password) # web.logout(username, password)
return redirect(url_for("index")), 501 return redirect(url_for("index"))
@app.errorhandler(429) @app.errorhandler(429)
@@ -372,7 +258,7 @@ def ratelimit_handler(e):
"Rate limit reached, please slow down :) ( Currently at " + e.description + ")", "Rate limit reached, please slow down :) ( Currently at " + e.description + ")",
"error", "error",
) )
app.logger.debug("Rate limit reached %s ", str(e.description)) app.logger.debug("Rate limit reached %s " , str(e.description))
return redirect(url_for("index")) return redirect(url_for("index"))
@@ -388,7 +274,7 @@ def ping():
@socketio.on("ping") @socketio.on("ping")
def handle_message(data): def handle_message(data):
"""Handle sockets pings""" """Handle sockets pings"""
app.logger.debug("Received : %s ", str(data)) app.logger.debug("Received : %s " , str(data))
socketio.emit("pong", "Pong !") socketio.emit("pong", "Pong !")
@@ -400,7 +286,3 @@ def camera_status():
socketio.emit("camera_status", True) socketio.emit("camera_status", True)
else: else:
socketio.emit("camera_status", False) socketio.emit("camera_status", False)
if __name__ == "__main__":
app.run(debug=True, use_reloader=False, host="0.0.0.0", ssl_context="adhoc")

View File

@@ -1,135 +0,0 @@
"""
This class has the method by which we manage the Tasks
It's a printing queue, so we need to add, remove and get information on where
the queue is
"""
from collections import deque
# Because actually printing and adding new print job requests happen at
# diffrent times, the print queue is managed by it's own thread.
import threading
from datetime import datetime
from task import TaskType
class PrintQueue:
"""
A Double-ended Queue to manage the printing Tasks
"""
def __init__(self, app):
self.app = app
self._queue = deque()
self._lock = threading.Lock()
self._completed_tasks = {} # Store completed task info
self._task_counter = 0
self.app.logger.debug("Created a new PrintQueue")
def __len__(self) -> int:
return len(self._queue)
def enqueue(self, task):
"""Add task to right of the queue and return position"""
with self._lock:
try:
self.app.logger.info("Add task %s to queue ", task.task_id)
self._queue.append(task)
position = self._queue.index(task)
# We return the current position of the task if it was added
self.app.logger.debug(
"Added a new task %s to the queue at position %s",
task.task_id,
position,
)
return position
except Exception as e:
self.app.logger.error("Could not add a task to the queue : %s ", e)
raise e
def dequeue(self):
"""Remove and return next task ( from the left of the queue ) (thread-safe)"""
with self._lock:
return self._queue.popleft() if len(self._queue) > 0 else None
def get_position(self, task):
"""Get current position of task in queue (1-indexed)"""
with self._lock:
if task.task_id in self._completed_tasks:
return None # Task already completed
try:
# Try to get the position of a Task
return self._queue.index(task)
except ValueError as e:
raise e
# end try
def is_empty(self):
"""Check if queue is empty"""
with self._lock:
self.app.logger.debug("Checking if queue is empty")
return len(self._queue) == 0
def get_queue_state(self):
"""Return current queue state"""
with self._lock:
self.app.logger.debug("Return current queue state")
return [{"task_id": t.task_id, "status": t.status} for t in self._queue]
def get_status(self, task_id):
"""Get full status info for a task"""
with self._lock:
if task_id in self._completed_tasks:
return self._completed_tasks[task_id]
# Check in queue if it exists
for index, task in enumerate(self._queue):
if task.task_id == task_id:
# Depending on it's type, we return more info
if task.task_type == TaskType.IMAGE:
return {
"task_id": task_id,
"status": task.status,
"type": task.task_type,
"position": index,
"in_queue": True,
"content": task.content,
"signature": task.signature,
}
if task.task_type == TaskType.TEXT:
return {
"task_id": task_id,
"status": task.status,
"type": task.task_type,
"position": index,
"in_queue": True,
"image_path": str(task.image_path),
"signature": task.signature,
"process": str(task.process),
}
if task.task_type == TaskType.CUT:
return {
"task_id": task_id,
"status": task.status,
"type": task.task_type,
"position": index,
"in_queue": True,
}
return None
def mark_completed(self, task_id, task_status):
"""Mark task as completed and remove from queue"""
with self._lock:
self._completed_tasks[task_id] = {
"task_id": task_id,
"status": task_status,
"position": None,
"in_queue": False,
"completed_at": datetime.now().isoformat(),
}

View File

@@ -1,18 +1,12 @@
""" # Importing the module to mage the connection to the printer.
This class manages connexion to a Printer from escpos.printer import Usb, USBNotFoundError
""" from time import sleep, gmtime, strftime
# import brother_ql
from time import sleep
import os.path import os.path
from PIL import Image, ImageEnhance, ImageOps
from PIL import Image, ImageEnhance
import numpy as np import numpy as np
# Importing the module to manage the connection to the printer.
import escpos.printer
class Printer(object):
class Printer():
""" """
# The connection is based on the ESC/POS library # The connection is based on the ESC/POS library
@@ -33,230 +27,147 @@ class Printer():
ready = False ready = False
def __init__(self, app, device_id, vendor_id): def __init__(self, app, device_id, vendor_id):
super().__init__() super(Printer, self).__init__()
self.app = app self.app = app
self.ready = False self.ready = False
self.printer = None self.printer = None
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
On printers that support it, we check that the printer has paper 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 RuntimeError("No more paper in the printer") raise Exception("No more paper in the printer")
case 1: case 1:
self.app.logger.warning( self.app.logger.warning('Printer needs paper to be changed very soon ! ')
"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):
"""
Check if the printer online ? Is the communication with the printer successfull ?
"""
# TODO: This could happen directly when creating a new Printer class # Is the printer online ? Is the communication with the printer successfull ?
if os.getenv("FLASK_DEBUG"): waiting_elapsed = 30
waiting_elapsed = 3 self.app.logger.debug('Waiting for printer to get online...')
else:
waiting_elapsed = 10
self.app.logger.debug("Waiting for printer to get online...")
while not self.ready: while not self.ready:
try: try:
# This also calls open(), which we need to close() # This also calls open(), which we need to close()
# or else the device will appear as busy. # or else the device will appear as busy.
p = escpos.printer.Usb( p = Usb(self.device_id, self.vendor_id, 0, profile="TM-P80")
self.device_id, self.vendor_id, 0, profile="TM-P80" except USBNotFoundError as e:
) self.app.logger.error("The USB device is not plugged in, trying again : " + str(e))
except RuntimeError as e: pass
self.app.logger.error(
"The USB device is not plugged in, trying again %s : %s",
waiting_elapsed,
str(e),
)
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 RuntimeError as e: except Exception as e:
self.app.logger.error( pass
"Error while getting the printer online %s : %s",
waiting_elapsed,
str(e),
)
sleep(1) sleep(1)
waiting_elapsed -= 1 waiting_elapsed -= 1
if waiting_elapsed < 1: if waiting_elapsed < 1:
self.app.logger.error( self.app.logger.error('Printer took more than 30 seconds to get online, aborting...')
"Printer took more than 30 seconds to get online, aborting..." waiting_elapsed = 30 # Reset the waiting time for the next print.
)
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( 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)
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):
if not isinstance(msg, str):
self.app.logger.error(
"It is not possible to print a " + str(type(msg)) + ", only strings."
)
raise ValueError
# We make sure that the signature is not something too goofy
clean_msg = str(msg) + "\n" clean_msg = str(msg) + "\n"
clean_signature = str(signature) clean_signature = str(signature)
# Make checks on the size of the message being printed
if len(clean_msg) > 4096: if len(clean_msg) > 4096:
self.app.logger.warning( self.app.logger.warning("Could not print message of this length: " + str(len(clean_msg)))
"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.")
)
raise RuntimeError(
"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( self.app.logger.warning("Could not print signature of this length: " + str(len(clean_signature)))
"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.")
)
raise RuntimeError(
"Could not print signature of this length :"
+ str(len(clean_signature))
+ ", needs to be below 256 caracters long."
)
# Do the actual printing
# We would pop the next element in the queue here, if it's a sms type
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))
raise RuntimeError(
"Unable to print a SMS, the printer couldn't do it."
) from e
self.app.logger.info("Printed text") self.app.logger.info("Printed text")
return True return True
def _print_img(self, path, signature="", center=True, process=False): def print_img(self, path, sign="",center=True,process=False):
clean_signature = str(signature) clean_signature = str(sign)
if len(signature) > 256: if len(sign) > 256:
self.app.logger.warning( self.app.logger.warning("Could not print signature of this length: " + str(len(clean_signature)))
"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.")
)
raise ValueError(
"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 OSError( raise Exception('The file path for this image :' + str(path) + " wasn't found. Please try again.")
"The file path for this image :" else:
+ str(path)
+ " wasn't found. Please try again."
)
self.app.logger.debug("Printing file from " + str(path)) self.app.logger.debug("Printing file from " + str(path))
if process: if process:
try: try:
self.app.logger.debug("Proccessing the image") self.app.logger.debug("Proccessing the image")
path = _process_image(self, path) path = process_image(self, path)
except RuntimeError as e: except Exception as e:
self.app.logger.error( self.app.logger.error(str(e))
"Error while processing the image, aborting print : %s", str(e) return False
)
raise e
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.textln(signature)
self.printer.close() self.printer.close()
self.app.logger.debug("Printed an image : " + str(path)) self.app.logger.debug("Printed an image : " + str(path))
except Exception as e:
self.app.logger.error(str(e))
raise RuntimeError("Could not print the picture") from e
finally:
try:
os.remove(path) os.remove(path)
except OSError as e:
raise e
self.app.logger.debug("Removed image : " + str(path)) self.app.logger.debug("Removed image : " + str(path))
try:
self.printer.close()
except Exception as e: except Exception as e:
self.app.logger.error( self.printer.close()
"Could not close the printer connexion %s", str(e) self.app.logger.error(str(e))
) return False
raise RuntimeError("Could not close the printer connexion. ") from e
self.app.logger.info("Printed a picture") self.app.logger.info("Printed a picture")
return True return True
def _qr(self, content): def qr(self, content):
try: try:
self.printer.open(self.usb_args) self.printer.open(self.usb_args)
self.printer.qr(content, center=True) self.printer.qr(content, center=True)
self.printer.close() self.printer.close()
except RuntimeError as e: except Exception as e:
self.printer.close() self.printer.close()
self.app.logger.error(str(e)) self.app.logger.error(str(e))
return False return False
@@ -264,7 +175,9 @@ class Printer():
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)
self.printer.cut() self.printer.cut()
@@ -277,22 +190,8 @@ class Printer():
self.app.logger.info("Did a cut") self.app.logger.info("Did a cut")
return True return True
def print_task(self, task_type, data):
"""Execute actual print based on task type"""
match (task_type.value):
case "text":
self._print_sms(data["txt"], signature=data["sign"])
case "image":
self._print_img(
data["img"], signature=data["sign"], process=data["process"]
)
case "cut":
self._cut()
case _:
raise RuntimeError("This task type is not supported")
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 (0255) brightness_threshold = 100 # Brightness threshold (0255)
contrast_factor = 0.6 # Less than 1.0 = lower contrast contrast_factor = 0.6 # Less than 1.0 = lower contrast
@@ -306,32 +205,23 @@ def _process_image(self, path):
original_img = original_img.convert("RGB") original_img = original_img.convert("RGB")
# Resize while maintaining aspect ratio # Resize while maintaining aspect ratio
original_img.thumbnail((max_width, max_height), Image.Resampling.LANCZOS) original_img.thumbnail((max_width, max_height), Image.LANCZOS)
self.app.logger.debug("Resized the image") self.app.logger.debug("Resized the image")
# # Convert to grayscale for dithering # # Convert to grayscale for dithering
# dithered_img = original_img.convert("L").convert("1") # dithered_img = original_img.convert("L").convert("1") # Dithering using default method (FloydSteinberg)
# Dithering using default method (FloydSteinberg)
# self.app.logger.debug("Dithered the image") # self.app.logger.debug("Dithered the image")
# 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( self.app.logger.debug("Average brightness of the image : " + str(avg_brightness) )
"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 = ( brightness_factor = 1 + (brightness_threshold - avg_brightness) / brightness_threshold
1 + (brightness_threshold - avg_brightness) / brightness_threshold 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}")
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)
@@ -339,9 +229,14 @@ def _process_image(self, path):
# contrast_enhancer = ImageEnhance.Contrast(original_img) # contrast_enhancer = ImageEnhance.Contrast(original_img)
# original_img = contrast_enhancer.enhance(contrast_factor) # original_img = contrast_enhancer.enhance(contrast_factor)
# Final resize check
if original_img.height > max_height:
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.")
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)
self.app.logger.debug("Processed and saved image.")
return jpeg_path return jpeg_path

View File

@@ -1,21 +1,12 @@
""" from flask_socketio import SocketIO
This class executes when we are on a raspberry Pi. from gpiozero import Button, LED, DigitalOutputDevice
from time import sleep, gmtime, strftime
It handles the press of a button via GPIO, from PIL import Image
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
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(): class Raspberry(object):
""" """
This class will manage three things : This class will manage three things :
- Connecting to a USB webcam - Connecting to a USB webcam
@@ -24,18 +15,8 @@ class Raspberry():
- 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 = printer
self,
print_queue,
app,
socketio,
button_gpio_port_number,
indicator_gpio_port_number,
flash_gpio_port_number,
is_flash_present,
):
self.print_queue = print_queue
self.socketio = socketio self.socketio = socketio
self.app = app self.app = app
@@ -43,75 +24,67 @@ class Raspberry():
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):
"""
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", encoding="utf-8") 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( self.app.logger.debug('This system does not appear to be a Raspberry Pi.')
"This system does not appear to be a Raspberry Pi."
)
return False return False
if not found: if not found:
self.app.logger.error( self.app.logger.error('Couldn\'t get sufficient hardware information from /proc/cpuinfo, Unable to determine if we are on a Raspberry Pi.')
"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()
@@ -130,30 +103,24 @@ class Raspberry():
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( self.app.logger.error(f"Invalid directory for image path: {self.image_path}")
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( self.app.logger.debug(f"Image captured successfully: {result.stdout.decode()}")
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( self.app.logger.error(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 RuntimeError as e: except Exception 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
@@ -165,42 +132,32 @@ class Raspberry():
return True return True
def overlay_logo(
self, def overlay_logo(self, image_path, logo_path, output_path=None, position='bottom_right', margin=10):
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 = ( logo_ratio = 0.30 # You can change the ratio if you want the logo bigger or smaller
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( raise ValueError("Invalid position. Choose from 'bottom_right', 'top_left', 'top_right', or 'bottom_left'.")
"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
@@ -210,7 +167,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 RuntimeError as e: except Exception as e:
self.app.logger.error(f"Error overlaying logo: {e}") self.app.logger.error(f"Error overlaying logo: {e}")
return False return False
@@ -237,7 +194,7 @@ class Raspberry():
image.save(output_path) image.save(output_path)
except RuntimeError as e: except Exception 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
@@ -247,27 +204,27 @@ class Raspberry():
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( self.indicator_countdown(iters=4,multi=20) # The indicator will flash a countdown LED
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 RuntimeError as e: except Exception 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)
)
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.print_queue.enqueue(ImageTask(self.image_path,signature="",process=True)) self.printer.print_img('src/static/images/extase-club.png',process=True)
self.print_queue.enqueue(TextTask(content="Imprimé par LittlePrynter", signature="")) self.printer.print_img(self.image_path,process=True)
time = strftime("%Y-%m-%d %H:%M", gmtime()) self.printer.print_sms("")
self.print_queue.enqueue(TextTask(content=time, signature="")) self.printer.print_sms("With Love From Société.Vide",signature="",bold=True)
self.print_queue.enqueue(CutTask()) 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.led.off() self.led.off()
self.app.logger.debug("Added a photomaton picture to the print queue") self.app.logger.debug("Done printing picture")
return True return True

View File

@@ -3,36 +3,6 @@
margin-right: 20%; margin-right: 20%;
} */ } */
html, body {
height: 100%;
width: 100%;
}
.fullscreen-countdown {
position: fixed;
top: 0;
left: 0;
right: 0;
bottom: 0;
z-index: 100;
height: 100%;
width: 100%;
/* backdrop-filter: blur(10px); */
color: black;
box-sizing: initial;
}
#countdown-number {
font-weight: bolder;
font-size: 10em;
text-shadow: 0px 0px 3em white, 1px 1px 1px rgba(255, 255, 255, 1);
position: fixed;
top: 50%;
left: 50%;
transform: translate(-50%, -50%);
}
/*// Small devices (landscape phones, 576px and up)*/ /*// Small devices (landscape phones, 576px and up)*/
@media (min-width: 576px) { @media (min-width: 576px) {

View File

@@ -186,18 +186,18 @@ function setup_events(){
}, false); }, false);
switch_cameras.addEventListener('click', function(ev) { switch_cameras.addEventListener('click', function(ev) {
flip_cameras(); flip_cameras(camera_options);
}, false ); }, false );
printButton.addEventListener('click', async function(ev) { printButton.addEventListener('click', function(ev){
ev.preventDefault(); data = take_picture();
try { try {
await countDownToPicture(); // wait for countdown to finish
var data = await take_picture(); // take_picture can be async
print_picture(data); print_picture(data);
} catch (e) { } catch (e) {
alert("Failed to print a picture because : " + e); alert("Failed to print a picture because : " + e);
} }
ev.preventDefault();
}, false); }, false);
} }
@@ -212,10 +212,6 @@ function clear_canvas() {
} }
function dataURLtoFile(dataurl, filename) { function dataURLtoFile(dataurl, filename) {
if(dataurl === "") {
console.log("We didn't receive data");
}
var arr = dataurl.split(','), mime = arr[0].match(/:(.*?);/)[1], var arr = dataurl.split(','), mime = arr[0].match(/:(.*?);/)[1],
bstr = atob(arr[1]), n = bstr.length, u8arr = new Uint8Array(n); bstr = atob(arr[1]), n = bstr.length, u8arr = new Uint8Array(n);
while(n--){ while(n--){
@@ -224,18 +220,19 @@ function dataURLtoFile(dataurl, filename) {
return new File([u8arr], filename, {type:mime}); return new File([u8arr], filename, {type:mime});
} }
async function take_picture() { function take_picture(){
const context = canvas.getContext('2d'); var context = canvas.getContext('2d');
let data = null;
if (width && height) { if (width && height) {
canvas.width = width; canvas.width = width;
canvas.height = height; canvas.height = height;
context.drawImage(video, 0, 0, width, height); context.drawImage(video, 0, 0, width, height);
data = canvas.toDataURL('image/png');
var data = canvas.toDataURL('image/png');
photo.setAttribute('src', data); photo.setAttribute('src', data);
} else { } else {
clear_canvas(); clear_canvas();
} }
return data; return data;
} }
@@ -255,28 +252,15 @@ function print_picture(data){
// headers:{ // headers:{
// 'Content-Type': 'multipart/form-data' // 'Content-Type': 'multipart/form-data'
// } // }
}).then(function(response) { }).then(function(response) { console.log('Success:', response); } , true)
console.log('Reponse:', response);
if(response.status != 200 ){
alert("The picture could not be printed be : " + response.statusText)
}
} , true)
.catch(error => console.error('Error:', error), false); .catch(error => console.error('Error:', error), false);
} }
function flip_cameras(){ function flip_cameras(camera_options){
console.log("Current facing mode : " + camera_options.video.facingMode)
if ( supports_facing_mode ) { if ( supports_facing_mode ) {
console.log("Support facing mode, trying to switch !"); get_webcam((camera_options.video.facingMode === 'user' ? 'environment' : 'user'));
if (camera_options.video.facingMode == 'user'){
camera_options = { audio: false, video: { facingMode: "environment" },};
} else {
camera_options = { audio: false, video: { facingMode: "user" },};
} }
}
console.log("Switching to " + camera_options.video.facingMode );
get_webcam(camera_options);
} }
function stop_video_streams(){ function stop_video_streams(){
@@ -316,33 +300,4 @@ function no_webcam_error(){
throw new Error("Unable to get a video device, stopping the photobooth."); throw new Error("Unable to get a video device, stopping the photobooth.");
} }
async function countDownToPicture(){
return new Promise((resolve) => {
console.log("Starting countdown");
// We create full-page overlay that displays a number to countdown until the picture is taken.
const countdown_div = document.getElementById("countdown");
countdown_div.style.display = "block";
// Set the different styling attributes of the element
countdown_div.setAttribute("class", "fullscreen-countdown");
// The loop must run for 3 seconds
interval = 1000;
var loops = 3;
var x = setInterval(() => {
console.log(loops);
// Update the content of the div
document.getElementById("countdown-number").innerHTML = loops;
if (loops == 0) {
countdown_div.style.display = "none";
// The timer has finished
clearInterval(x);
resolve();
}
loops--;
}, interval);
});
}
window.addEventListener('load', startup, false); window.addEventListener('load', startup, false);

View File

@@ -1,98 +0,0 @@
"""
Here we define the types of tasks
We are using Abstract Base Classes,
like this we can define types of tasks ( text, images, ... )
that all work with the same basic options
The tasks are going to be injected into a Queue.
It's a usefull way of storing information in our
program, while making sure that things are indeed printed.
It's also a way to prevent two concurrent connexions creating
a access conflict on a single printer, like two people wanting
to print at the same time.
We can also delay and store printing tasks until a printer becomes
available if none is online.
"""
from abc import ABC, abstractmethod
## See https://docs.python.org/3/library/abc.html to learn more about this
from enum import Enum
import uuid
## You can expand this if you want to take other types of tasks into account
class TaskType(Enum):
"""
The different tasks supported by the printers
"""
TEXT = "text"
IMAGE = "image"
CUT = "cut"
class PrintTask(ABC):
"""
A print task holds information about what we are looking to print.
"""
def __init__(self, task_type):
self.task_id = self._generate_id()
self.task_type = task_type
self.status = "pending" # pending, processing, completed, failed
print("Created a new " + str(self.task_type) + " with ID " + self.task_id)
@abstractmethod
def get_print_data(self):
"""Return data formatted for printer"""
def _generate_id(self):
# Generate unique task ID
return str(uuid.uuid4())
class TextTask(PrintTask):
"""
This tasks represents a texte content, and it's signature.
"""
def __init__(self, content, signature):
super().__init__(TaskType.TEXT)
self.content = content
self.signature = signature
def get_print_data(self):
return {"txt": self.content, "sign": self.signature}
class ImageTask(PrintTask):
"""
This tasks represents a image content ( in the form of it's path ), and it's signature.
"""
def __init__(self, image_path, signature, process):
super().__init__(TaskType.IMAGE)
self.image_path = image_path
self.signature = signature
self.process = process
def get_print_data(self):
# Return image data in printer-compatible format
return {"img": self.image_path, "sign": self.signature, "process": self.process}
class CutTask(PrintTask):
"""
This class activates the cutter on the printer if it exists
"""
def __init__(self):
super().__init__(TaskType.CUT)
# There is no print data,
# the task existence in itself is indication of what to do
def get_print_data(self):
return None

View File

@@ -5,9 +5,9 @@
<div class="card"> <div class="card">
<h3 class="card-header">Print a short message</h3> <h3 class="card-header">Print a short message</h3>
<div class="card-body"> <div class="card-body">
<form class="form-group" action="/web/print/sms" method="post"> <form class="form-group" action="/api/print/sms" method="post">
<input class="form-control" type="text" name="txt" placeholder="200 chars or less " maxlength="200" required><br> <input class="form-control" type="text" name="txt" placeholder="200 chars or less " maxlength="200" required><br>
<input class="form-control" type="text" name="signature" placeholder="Signature or pseudo" maxlength="200"><br> <input class="form-control" type="text" name="signature" placeholder="Signature or pseudo" maxlength="200" required><br>
<input class="btn btn-primary float-right" type="submit" value="Imprimer" name="imprimer"> <input class="btn btn-primary float-right" type="submit" value="Imprimer" name="imprimer">
</form> </form>
</div> </div>
@@ -18,9 +18,9 @@
<div class="card"> <div class="card">
<h3 class="card-header">Print an image</h3> <h3 class="card-header">Print an image</h3>
<div class="card-body"> <div class="card-body">
<form enctype="multipart/form-data" class="form-group" action="/web/print/img" method="post"> <form enctype="multipart/form-data" class="form-group" action="/api/print/img" method="post">
<input class="form-control" type="file" name="img" required><br> <input class="form-control" type="file" name="img" required><br>
<input class="form-control" type="text" name="signature" placeholder="Signature or pseudo" maxlength="200"><br> <input class="form-control" type="text" name="signature" placeholder="Signature or pseudo" maxlength="200" required><br>
<input class="btn btn-primary float-right" type="submit" value="Imprimer" name="imprimer"> <input class="btn btn-primary float-right" type="submit" value="Imprimer" name="imprimer">
</form> </form>
</div> </div>

View File

@@ -54,10 +54,6 @@
</div> </div>
<div id="countdown">
<div id="countdown-number"></div>
</div>
<style media="screen"> <style media="screen">
canvas { canvas {
display: none; display: none;

View File

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

View File

@@ -1,115 +1,69 @@
from flask import flash from flask import Flask, request, flash
from werkzeug.utils import secure_filename from werkzeug.utils import secure_filename
from printer import Printer
import time import time
import os import os
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 """docstring for web."""
( API and Web page ) and provides checks before sending stuff to printing"""
def __init__(self, app, print_queue): def __init__(self, app, printer ):
super(Web).__init__() super(Web).__init__()
self.print_queue = print_queue self.printer = printer
self.app = app self.app = app
def print_sms(self, texte, sign: str) -> bool: def print_sms(self, texte, sign: str):
""" # TODO: verify the texte before printing it here ?
Get text and a signature, prints the text and cuts after that.
"""
self.app.logger.debug("Printing : " + str(texte) + " from " + str(sign)) self.app.logger.debug("Printing : " + str(texte) + " from " + str(sign))
try: try:
# We create two new tasks and add them directly to the queue self.printer.print_sms(texte, sign)
# TODO: this might need to be improved because self.printer.cut()
# !! there is no garantee !! that both the SMS task and the Cut task
# are added back to back, another task could be
# inserted between the two.
sms = self.print_queue.enqueue(TextTask(content=texte, signature=sign))
cut = self.print_queue.enqueue(CutTask())
except Exception as e: except Exception as e:
self.app.logger.error(e) self.app.logger.error(e)
raise RuntimeError("Could not add SMS to queue, " + str(e)) from e flash("Error while printing the SMS : "+ str(e))
self.app.logger.info("Added two new tasks at position %s and %s", sms, cut)
return True
def print_image(self, image, sign: str) -> bool: flash("You message " + str( texte ) + " has been printed :)")
"""
Get an image and a signature, prints the image and cuts after that.
""" def print_image(self, image, sign):
self.app.logger.debug("Uploading file")
try: try:
file_uploaded = self.upload_file(image) self.app.logger.debug("Uploading file from " + str(sign))
except Exception as e: if self.upload_file(image):
self.app.logger.error(e)
raise RuntimeError("Could not upload file") from e
if file_uploaded:
self.app.logger.debug("File has been uploaded, printing...") self.app.logger.debug("File has been uploaded, printing...")
try: self.printer.print_img(os.path.join(self.app.config['UPLOAD_FOLDER'], secure_filename(image.filename)), sign=sign,process=True)
img = self.print_queue.enqueue( self.printer.cut()
ImageTask(
os.path.join(
self.app.config["UPLOAD_FOLDER"],
secure_filename(image.filename),
),
signature=sign,
process=True,
)
)
cut = self.print_queue.enqueue(CutTask())
except Exception as e: except Exception as e:
raise RuntimeError("Could not add IMG to queue" + str(e)) from e self.app.logger.error(e)
flash("Could not upload file." + str(e))
self.app.logger.info("Added two new tasks at position %s and %s", img, cut) flash("Your image has been printed :)")
return True def login(username: str,password: str) -> bool:
pass
def login(self, username: str, password: str) -> bool: def logout(username: str, password: str) -> bool:
"""Not implemented""" pass
return
def logout(self, username: str, password: str) -> bool:
"""Not implemented"""
return
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 ( return '.' in filename and filename.rsplit('.', 1)[1].lower() in self.app.config['ALLOWED_EXTENSIONS']
"." 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: if image and self.allowed_file(image.filename):
if 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))
except OSError as e: self.app.logger.debug("File saved")
self.app.logger.error("Could not save file %s", e) except Exception as e:
self.app.logger.error("Could not save file")
flash(str(e),'error')
return False return False
self.app.logger.debug( self.app.logger.debug("File saved to " + str(os.path.join(self.app.config['UPLOAD_FOLDER'], filename)))
"File saved to "
+ str(os.path.join(self.app.config["UPLOAD_FOLDER"], filename))
)
return True return True
else: else:
self.app.logger.error( self.app.logger.error("Could not save file " + str(filename))
"Could not save file because the filename is forbidden"
)
return False return False
else:
self.app.logger.error(
"Could not save file, it seems to be null ? : " + str(filename)
)
return False
def get_queue_state(self):
"""Return current queue state"""
return self.print_queue.get_queue_state()

View File

@@ -1,107 +0,0 @@
# This is the main printing thread
# As explained in the task file, this is where we command
# printing to happen.
import threading
import time
class PrintWorker(threading.Thread):
def __init__(self, app, print_queue, printer, socketio=None):
super().__init__(daemon=True)
self.app = app
self.print_queue = print_queue
self.printer = printer
self.socketio = socketio # Optional
self.running = True
self.state = "idle" # idle, printing, dead, drinking-a-beer
self.app.logger.debug("Ho great, I'm alive... I'm ready to work another day...")
def run(self):
"""Background thread that processes queue items"""
self.app.logger.info("Worker started working.")
while True:
if not self.running or not self.printer.ready:
time.sleep(0.2)
continue
try:
task = self.print_queue.dequeue()
except Exception as e:
self.app.logger.error("Could not get a new task ! %s ", str(e))
raise RuntimeError(
"We could not get a new task because " + str(e)
) from e
if task:
try:
self.app.logger.info("Got a new task")
self.app.logger.debug("Got task %s", task.task_id)
self.state = "printing"
task.status = "processing"
self._emit_status(task.task_id, "processing")
print_data = task.get_print_data()
try:
self.printer.print_task(task.task_type, print_data)
except RuntimeError as e:
self.app.logger.error("Could not print : %s", str(e))
raise e
task.status = "completed"
self.print_queue.mark_completed(task.task_id, "completed")
self._emit_status(task.task_id, "completed")
except RuntimeError as e:
task.status = "failed"
self.print_queue.mark_completed(task.task_id, "failed")
self._emit_status(task.task_id, "failed", error=str(e))
print(f"Print task {task.task_id} failed: {e}")
else:
# When they are no new tasks to handle, we put the thread to sleep.
self.state = "idle"
time.sleep(0.1)
def _emit_status(self, task_id, status, error=None):
"""Emit status update via Socket.IO if available"""
if not self.socketio:
return
room = f"task_{task_id}"
data = {
"task_id": task_id,
"status": status,
"position": None, # Task no longer in queue
}
if error:
data["error"] = error
self.socketio.emit("task_status", data, room=room)
def stop_worker(self):
"""
Give the worker a break
"""
self.app.logger.debug("Giving the worker a break")
self.state = "drinking-a-beer"
self.running = False
def start_worker(self):
"""
Get the worker back to it
"""
self.app.logger.debug("Time to work !")
self.state = "idle"
self.running = True
def current_state(self):
"""
Return the worker state
"""
return {
"is_running": self.running,
"queue_size": len(self.print_queue),
"state": self.state,
}