d3vyce/teleinfo-exporter
1
0
Fork 0
mirror of https://github.com/d3vyce/teleinfo-exporter.git synced 2025-04-04 09:10:48 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: MQTT brocker disconnection when client connection timeout

Browse Source
This commit is contained in:
d3vyce 2024-02-16 18:09:34 +01:00
parent 90f9414d2d
commit 0160aa6197
1 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/teleinfo_exporter/app.py
View File

@ -3,6 +3,7 @@
import json import json
import random import random
import string import string
import time
import bcrypt import bcrypt
import configargparse import configargparse
@ -92,6 +93,7 @@ teleinfo_contract_type = Gauge("teleinfo_contract_type", "contract type", ["type
def on_connect(client, userdata, flags, rc): # pylint: disable=unused-argument def on_connect(client, userdata, flags, rc): # pylint: disable=unused-argument
client.subscribe("teleinfo/tele/SENSOR")
if rc == 0: if rc == 0:
print("Connected to broker") print("Connected to broker")
else: else:
@ -146,6 +148,17 @@ def on_message(client, userdata, message): # pylint: disable=unused-argument
teleinfo_contract_type.labels(message["TIC"]["OPTARIF"]).set(0) teleinfo_contract_type.labels(message["TIC"]["OPTARIF"]).set(0)
def on_disconnect(client, userdata, rc):
print("Diconnected from broker, reconnecting...")
while True:
try:
if not client.reconnect():
break
except ConnectionRefusedError:
pass
time.sleep(1)
@app.before_request @app.before_request
@auth.login_required() @auth.login_required()
def global_auth(): def global_auth():
@ -162,7 +175,6 @@ def verify_password(username, password):
return username return username
return None return None
@app.route("/metrics") @app.route("/metrics")
def metrics(): def metrics():
return make_wsgi_app() return make_wsgi_app()
@ -186,7 +198,6 @@ def main():
p.add("--http_cert", help="HTTP Server Certificate", env_var="HTTP_CERT") p.add("--http_cert", help="HTTP Server Certificate", env_var="HTTP_CERT")
p.add("--http_key", help="HTTP Server Key", env_var="HTTP_KEY") p.add("--http_key", help="HTTP Server Key", env_var="HTTP_KEY")
options = p.parse_args() options = p.parse_args()
print(options)
if options.auth_user and options.auth_hash: if options.auth_user and options.auth_hash:
app.config["USERS"] = {options.auth_user: options.auth_hash.encode()} app.config["USERS"] = {options.auth_user: options.auth_hash.encode()}
@ -198,13 +209,12 @@ def main():
if options.broker_user and options.broker_password: if options.broker_user and options.broker_password:
client.username_pw_set(options.broker_user, password=options.broker_password) client.username_pw_set(options.broker_user, password=options.broker_password)
client.connect(options.broker_host, port=options.broker_port)
client.on_connect = on_connect client.on_connect = on_connect
client.on_message = on_message client.on_message = on_message
client.connect(options.broker_host, port=options.broker_port) client.on_disconnect = on_disconnect
client.loop_start() client.loop_start()
client.subscribe(options.broker_topic)
if options.http_cert and options.http_key: if options.http_cert and options.http_key:
ssl_context = (options.http_cert, options.http_key) ssl_context = (options.http_cert, options.http_key)
else: else: