Dans un précédent billet (Bouton connecté Zigbee, connexion MQTT en Python avec Eclipse Paho-MQTT), je jouais déjà avec un bouton-poussoir connecté en Zigbee. Je me contentais alors d'afficher la détection du clic sur le bouton dans un terminal.Cette fois, je crée un serveur qui sert une page Web avec une checkbox HTML décorée en CSS pour représenter un interrupteur ON/OFF. Le but est de faire basculer cet interrupteur à chaque clic sur le bouton sans fil Zigbee.
[Python FastAPI] [Zigbee2MQTT] Comment contrôler un interrupteur HTML/CSS avec un bouton sans fil Zigbee
Le serveur est piloté en Python et le framework web FastAPI. Le serveur gère aussi la connexion MQTT en se mettant en attente des messages à chaque clic du bouton Zigbee. Des messages JSON du type {'state' : true} ou {'state' : false} sont envoyés alternativement par le serveur au client à chaque clic du bouton Zigbee grâce à une liaison WebSocket.
index.html :
| Code HTML : | Sélectionner tout |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | <!DOCTYPE html> <html> <head> <title>Switch Control</title> <link rel="stylesheet" href="/static/style.css"> </head> <body> <div class="toggle"> <input type="checkbox" id="btn" disabled> <label for="btn"> <span class="track"> <span class="txt"></span> </span> <span class="thumb">|||</span> </label> </div> <script> // Création d'une connexion WebSocket à l'URL spécifiée const ws = new WebSocket("ws://192.168.0.20:8000/ws"); // Fonction appelée lors de la réception d'un message ws.onmessage = function(event) { const data = JSON.parse(event.data); // Met à jour l'état de la checkbox en fonction de la donnée reçue document.getElementById("btn").checked = data.state; }; </script> </body> </html> |
main.py :
| Code Python : | Sélectionner tout |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 | from fastapi import FastAPI, WebSocket from fastapi.staticfiles import StaticFiles from fastapi.responses import HTMLResponse import asyncio from button_client import ButtonSNZB01PClient app = FastAPI() app.mount("/static", StaticFiles(directory="static"), name="static") # instanciation client MQTT : @IP du broker, port et topic mqtt_client = ButtonSNZB01PClient("raspberrypi.local", 1883, "zigbee2mqtt/fleb-SNZB-01P-switch") @app.on_event("startup") def startup_event(): """ Fonction appelée au démarrage de l'application. Connecte le client MQTT et démarre la boucle MQTT. """ print("Appel de la fonction startup") mqtt_client.connect() print("Connexion MQTT effectuée") mqtt_client.start_mqtt_loop() print("Boucle MQTT démarrée") @app.get("/", response_class=HTMLResponse) async def get(): """ Endpoint GET pour la racine de l'application. Lit et renvoie le contenu du fichier index.html. """ print("Lecture du fichier index.html") with open("index.html", "r") as file: html_content = file.read() print("index.html lu, envoi du contenu au client") return HTMLResponse(content=html_content) @app.websocket("/ws") async def websocket_endpoint(websocket: WebSocket): """ Endpoint WebSocket pour gérer les connexions WebSocket. Accepte la connexion WebSocket et maintient la connexion ouverte. """ print("connexion WebSocket établie à l'URL /ws") await websocket.accept() print("Connexion Websocket acceptée") mqtt_client.websocket = websocket print("Maintien de la connexion Websocket ouverte") try: while True: await asyncio.sleep(1) except Exception as e: print(f"Erreur de connexion WebSocket : {e}") finally: mqtt_client.websocket = None |
button_client.py :
| Code Python : | Sélectionner tout |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 | import paho.mqtt.client as mqtt import json import asyncio import threading class ButtonSNZB01PClient: """ Client MQTT pour le bouton Sonoff SNZB-01P. """ def __init__(self, broker, port, topic, state=False): """ Initialise le client MQTT. :param broker: Adresse ip du broker MQTT. :param port: Port du broker MQTT. :param topic: Topic MQTT à souscrire. :param state: État initial du switch. """ self.broker = broker self.port = port self.topic = topic self.state = state self.client = mqtt.Client() self.websocket = None # fonctions de rappel self.client.on_connect = self.on_connect self.client.on_message = self.on_message def connect(self): """ Connecte le client au broker MQTT. """ print("Appel de la méthode connect du broker") self.client.connect(self.broker, self.port, 60) print("Tentative de connexion au broker...") def on_connect(self, client, userdata, flags, reason_code, properties=None): """ Fonction de rappel appelée lors de la connexion au broker MQTT. """ if reason_code == 0: print("Connexion au broker réussie !") self.client.subscribe(self.topic) print(f"Abonné au topic : {self.topic}") else: print(f"Echec de la connexion au broker, code d'erreur : {reason_code}") def on_message(self, client, userdata, msg): """ Fonction de rappel appelée lors de la réception d'un message MQTT. """ key = 'action' try: message = json.loads(msg.payload.decode()) if message.get(key) == "single": self.state = not self.state print(f"Changement d'état du switch détecté : {self.state}") # Envoyer l'état du switch via WebSocket print("Envoi de l'état du switch via WebSocket") if self.websocket: asyncio.run(self.send_switch_state()) else: print(f"Erreur de clé : {key}") self.disconnect() except json.JSONDecodeError: print(f"Réception de JSON non conforme sur le topic {msg.topic}") self.disconnect() async def send_switch_state(self): """ Envoie l'état du switch via WebSocket. """ if self.websocket: print(f"Envoi du JSON via websocket : {{'state' : {self.state}}}") await self.websocket.send_json({"state": self.state}) def start_mqtt_loop(self): """ Démarre la boucle MQTT dans un thread séparé. """ print("Démarrage de la boucle MQTT dans un thread séparé") thread = threading.Thread(target=self.client.loop_forever) thread.start() def disconnect(self): """ Déconnecte le client du broker MQTT. """ self.client.disconnect() print("Client MQTT déconnecté.") |
/static/style.css, d'après https://codepen.io/josetxu/pen/Poxwpme :
Au démarrage de l'application, la fonction startup_event connecte le client MQTT (ButtonSNZB01_client) au broker MQTT. La boucle MQTT est démarrée et fonctionne en arrière-plan, en attente de messages MQTT du topic qui mettront à jour l'état de la checkbox de la page Web en conséquence.
Quand un client se connecte depuis un navigateur, le serveur Web sert la page Web au client qui se connecte à l'endpoint Websocket /ws. Le serveur accepte la connexion Websocket et la maintient. La connexion Websocket est assignée au client MQTT.
Lorsque l'utilisateur clique sur le bouton Zigbee, un message {'action' : 'single', ...} est envoyé par le broker au client MQTT. Si le message contient l'action 'single' (appui simple du bouton), l'état de l'interrupteur est inversé (méthode on_message). La méthode send_switch_state est appelée pour envoyer le nouvel état de l'interrupteur via Websocket. Le client Websocket reçoit le nouvel état et met à jour la checkbox de la page Web.
Conclusion : Le code de cette application de démonstration n'est sans doute pas optimisé de la meilleure des façons, mais il montre tout de même comment gérer de façon asynchrone les interactions entre le client Web, FastAPI, le client MQTT et le broker MQTT, afin d'éviter les situations bloquantes. Je vais donc poursuivre l'expérience...