Refactor forwarder and implement clock service
This commit is contained in:
76
rpi/main.py
76
rpi/main.py
@@ -1,56 +1,86 @@
|
||||
#!/usr/bin/env python3
|
||||
import os
|
||||
|
||||
from flask import Flask
|
||||
from dotenv import load_dotenv
|
||||
import broker_starter as bs
|
||||
from flask import Flask
|
||||
|
||||
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.forwarder_service import ForwarderService
|
||||
from services.mqtt_service import MQTTService
|
||||
|
||||
load_dotenv()
|
||||
|
||||
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")
|
||||
|
||||
app = Flask(__name__)
|
||||
|
||||
screen = Screen()
|
||||
rfid_reader = RfidReader("/dev/serial0", 9600)
|
||||
light_sensor_reader = LoraLightSensorReader("/dev/ttyUSB1", 9600)
|
||||
|
||||
detection_service = DetectionService()
|
||||
|
||||
@app.route("/party/start", methods=['POST'])
|
||||
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,
|
||||
)
|
||||
|
||||
forward_service = ForwarderService(local_broker, api_broker)
|
||||
|
||||
@app.route("/command/party/start", methods=['POST'])
|
||||
def start_party():
|
||||
print("Party started!")
|
||||
pass
|
||||
|
||||
def start_serial_devices():
|
||||
screen.enableBackground()
|
||||
screen.displayMessage("Waiting for scan...")
|
||||
|
||||
rfid_reader.subscribe(lambda uid: bs.local_broker.publish("/system/sensor/rfid", str(uid), 1))
|
||||
rfid_reader.start()
|
||||
|
||||
light_sensor_reader.subscribe(lambda light_value: bs.local_broker.publish("/system/sensor/light", str(light_value), 0))
|
||||
light_sensor_reader.start()
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
try :
|
||||
print("Starting app...")
|
||||
forward_service.register_forwarder(client_id, "/system/sensor/rfid", f"/customer/telemetry/rfid")
|
||||
forward_service.register_forwarder(client_id, "/system/sensor/light", f"/customer/telemetry/light")
|
||||
forward_service.register_forwarder(client_id, "/system/sensor/gps", f"/customer/telemetry/gps")
|
||||
|
||||
bs.start()
|
||||
print("Brokers started")
|
||||
rfid_reader.subscribe(lambda uid: local_broker.publish("/system/sensor/rfid", str(uid), 1))
|
||||
light_sensor_reader.subscribe(lambda light_value: local_broker.publish("/system/sensor/light", str(light_value), 0))
|
||||
|
||||
start_serial_devices()
|
||||
print("Serial devices started")
|
||||
screen.enableBackground()
|
||||
screen.displayMessage("Waiting for scan...")
|
||||
|
||||
forward_service.start_all()
|
||||
|
||||
rfid_reader.start()
|
||||
light_sensor_reader.start()
|
||||
|
||||
print("App started...")
|
||||
app.run(host="0.0.0.0", port=5000, debug=False)
|
||||
|
||||
except KeyboardInterrupt:
|
||||
print("Keyboard interrupt. Stopping app...")
|
||||
bs.stop()
|
||||
forward_service.stop_all()
|
||||
exit()
|
||||
|
||||
except Exception as e:
|
||||
print(e)
|
||||
bs.stop()
|
||||
exit()
|
||||
forward_service.stop_all()
|
||||
exit()
|
||||
Reference in New Issue
Block a user