Files
board-mate/rpi/main.py
2025-12-30 03:56:31 +01:00

96 lines
2.7 KiB
Python

#!/usr/bin/env python3
import os
from flask import Flask
from dotenv import load_dotenv
from controllers.mqtt_forwarder import MQTTForwarder
from hardware.light.lora_light_sensor_reader import LoraLightSensorReader
from hardware.screen.screen import Screen
from hardware.rfid.reader import RfidReader
from services.detection_service import DetectionService
from services.mqtt_service import MQTTService
load_dotenv()
app = Flask(__name__)
client_id = "pi-1"
local_broker_address = os.environ.get("LOCAL_BROKER_ADDRESS", "127.0.0.1")
local_broker_port = int(os.environ.get("LOCAL_BROKER_PORT", 1883))
local_username = os.environ.get("LOCAL_USERNAME")
local_password = os.environ.get("LOCAL_PASSWORD")
api_broker_address = os.environ.get("API_BROKER_ADDRESS", "127.0.0.1")
api_broker_port = int(os.environ.get("API_BROKER_PORT", 1883))
api_username = os.environ.get("API_USERNAME")
api_password = os.environ.get("API_PASSWORD")
screen = Screen()
rfid_reader = RfidReader("/dev/serial0", 9600)
light_sensor_reader = LoraLightSensorReader("/dev/ttyUSB1", 9600)
local_broker = MQTTService(
local_broker_address,
local_broker_port,
client_id="system",
username=local_username,
password=local_password,
)
api_broker = MQTTService(
api_broker_address,
api_broker_port,
client_id=client_id,
username=api_username,
password=api_password,
)
detection_service = DetectionService()
@app.route("/party/start", methods=['POST'])
def start_party():
print("Party started!")
pass
def start_brokers():
forwarder = MQTTForwarder(client_id, local_broker, api_broker)
forwarder.start(f"/system/sensor/rfid", f"/customer/telemetry/rfid")
forwarder = MQTTForwarder(client_id, local_broker, api_broker)
forwarder.start(f"/system/sensor/light", f"/customer/telemetry/light")
def start_serial_devices():
screen.enableBackground()
screen.displayMessage("Waiting for scan...")
rfid_reader.subscribe(lambda uid: local_broker.publish("/system/sensor/rfid", str(uid), 1))
rfid_reader.start()
light_sensor_reader.subscribe(lambda light_value: local_broker.publish("/system/sensor/light", str(light_value), 0))
light_sensor_reader.start()
if __name__ == "__main__":
try :
print("Starting app...")
start_brokers()
print("Brokers started")
#start_serial_devices()
print("Serial devices started")
app.run(host="0.0.0.0", port=5000, debug=False)
except KeyboardInterrupt:
print("Keyboard interrupt. Stopping app...")
local_broker.disconnect()
api_broker.disconnect()
exit()
except Exception as e:
print(e)
local_broker.disconnect()
api_broker.disconnect()
exit()