Commit 2
This commit is contained in:
parent
d377fa89a4
commit
9e702af8c6
13 changed files with 639 additions and 133 deletions
197
python/mqtt-systray.py
Normal file → Executable file
197
python/mqtt-systray.py
Normal file → Executable file
|
|
@ -7,7 +7,7 @@ import signal
|
|||
import paho.mqtt.client as mqtt
|
||||
|
||||
gi.require_version('Gtk', '3.0')
|
||||
from gi.repository import Gtk
|
||||
from gi.repository import Gtk, Gdk
|
||||
|
||||
gi.require_version('AppIndicator3', '0.1')
|
||||
from gi.repository import AppIndicator3 as AppIndicator
|
||||
|
|
@ -15,9 +15,6 @@ from gi.repository import AppIndicator3 as AppIndicator
|
|||
from PIL import Image, ImageDraw, ImageFont
|
||||
import yaml
|
||||
|
||||
mqttprefix = "devices"
|
||||
mqtthost = '192.168.67.1'
|
||||
|
||||
oldtmpfile = ''
|
||||
oldalertcount = 0
|
||||
menu = Gtk.Menu()
|
||||
|
|
@ -26,17 +23,18 @@ fp = open("mqtt-systray.yaml")
|
|||
conf = yaml.safe_load(fp)
|
||||
fp.close()
|
||||
|
||||
mqttprefix = conf['config']['mqtt_prefix']
|
||||
mqtthost = conf['config']['mqtt_host']
|
||||
|
||||
appindicator = AppIndicator.Indicator.new(
|
||||
'mqttalert',
|
||||
'',
|
||||
AppIndicator.IndicatorCategory.APPLICATION_STATUS)
|
||||
|
||||
def create_image(width, height, count):
|
||||
def create_image(width, height, count, color_circle):
|
||||
# Generate an image and draw a pattern
|
||||
image = Image.new('RGB', (width, height), 'white')
|
||||
dc = ImageDraw.Draw(image)
|
||||
color_circle = 'green'
|
||||
if count > 0: color_circle = 'red'
|
||||
#dc.circle((width // 2, height // 2), height // 2, fill=color2)
|
||||
radius = height // 2
|
||||
ellipse_xy = (0, 0, radius*2, radius*2)
|
||||
|
|
@ -54,42 +52,76 @@ def quit():
|
|||
Gtk.main_quit()
|
||||
|
||||
def signal_handler(sig, frame):
|
||||
print('You pressed Ctrl+C!')
|
||||
#print('You pressed Ctrl+C!')
|
||||
quit()
|
||||
|
||||
def gtkquit(source):
|
||||
quit()
|
||||
quit()
|
||||
|
||||
def get_alert(devicea):
|
||||
for sensor in conf[devicea]['values'].keys():
|
||||
if 'value' in conf[devicea]['values'][sensor] and (conf[devicea]['values'][sensor]['value'] > conf[devicea]['values'][sensor]['max'] or conf[devicea]['values'][sensor]['value'] < conf[devicea]['values'][sensor]['min']):
|
||||
return True
|
||||
return False
|
||||
def gtkdashboard(source):
|
||||
Gtk.show_uri_on_window(None, conf['config']['dashboard_url'],int(Gdk.CURRENT_TIME))
|
||||
|
||||
def get_alert2(devicea, sensora):
|
||||
def get_alert_text_warning(devicea, sensora):
|
||||
alert_text = ''
|
||||
if 'value' in conf[devicea]['values'][sensora]:
|
||||
if conf[devicea]['values'][sensora]['value'] > conf[devicea]['values'][sensora]['max']:
|
||||
alert_text = '!!HIGH!! '
|
||||
elif conf[devicea]['values'][sensora]['value'] < conf[devicea]['values'][sensora]['min']:
|
||||
alert_text = '!!LOW!! '
|
||||
if 'value' in conf['devices'][devicea]['values'][sensora]:
|
||||
if 'max_warning' in conf['devices'][devicea]['values'][sensora] and conf['devices'][devicea]['values'][sensora]['value'] > conf['devices'][devicea]['values'][sensora]['max_warning']:
|
||||
alert_text = '!!high!! '
|
||||
elif 'min_warning' in conf['devices'][devicea]['values'][sensora] and conf['devices'][devicea]['values'][sensora]['value'] < conf['devices'][devicea]['values'][sensora]['min_warning']:
|
||||
alert_text = '!!low!! '
|
||||
return alert_text
|
||||
|
||||
def get_alert_text_critical(devicea, sensora):
|
||||
alert_text = ''
|
||||
if 'value' in conf['devices'][devicea]['values'][sensora]:
|
||||
if 'max_critical' in conf['devices'][devicea]['values'][sensora] and conf['devices'][devicea]['values'][sensora]['value'] > conf['devices'][devicea]['values'][sensora]['max_critical']:
|
||||
alert_text = '!!HIGH!! '
|
||||
elif 'min_critical' in conf['devices'][devicea]['values'][sensora] and conf['devices'][devicea]['values'][sensora]['value'] < conf['devices'][devicea]['values'][sensora]['min_critical']:
|
||||
alert_text = '!!LOW!! '
|
||||
return alert_text
|
||||
|
||||
def get_alert_text(devicea, sensora):
|
||||
alert_text = ''
|
||||
if get_alert_text_critical(devicea, sensora) != '':
|
||||
alert_text = get_alert_text_critical(devicea, sensora)
|
||||
elif get_alert_text_warning(devicea, sensora) != '':
|
||||
alert_text = get_alert_text_warning(devicea, sensora)
|
||||
return alert_text
|
||||
|
||||
def alert_count_critical():
|
||||
count = 0
|
||||
for device in conf['devices']:
|
||||
for sensor in conf['devices'][device]['values'].keys():
|
||||
if get_alert_text_critical(device, sensor) != '': count = count + 1
|
||||
return count
|
||||
|
||||
def alert_count_warning():
|
||||
count = 0
|
||||
for device in conf['devices']:
|
||||
for sensor in conf['devices'][device]['values'].keys():
|
||||
if get_alert_text_warning(device, sensor) != '': count = count + 1
|
||||
return count
|
||||
|
||||
def alert_count():
|
||||
count = 0
|
||||
for device in conf:
|
||||
for sensor in conf[device]['values'].keys():
|
||||
if get_alert2(device, sensor) != '': count = count + 1
|
||||
return count
|
||||
count = 0
|
||||
for device in conf['devices']:
|
||||
for sensor in conf['devices'][device]['values'].keys():
|
||||
if get_alert_text_critical(device, sensor) != '' or get_alert_text_warning(device, sensor) != '': count = count + 1
|
||||
return count
|
||||
|
||||
def get_icon():
|
||||
global oldtmpfile
|
||||
if oldtmpfile!='': os.remove(oldtmpfile)
|
||||
icon_path = tempfile.mktemp()
|
||||
oldtmpfile=icon_path
|
||||
with open(icon_path, 'wb') as f:
|
||||
create_image(64, 64, alert_count()).save(f, 'PNG')
|
||||
return icon_path
|
||||
global oldtmpfile
|
||||
if oldtmpfile!='': os.remove(oldtmpfile)
|
||||
icon_path = tempfile.mktemp()
|
||||
oldtmpfile=icon_path
|
||||
color = 'green'
|
||||
if alert_count_critical() > 0:
|
||||
color = 'red'
|
||||
elif alert_count_warning() > 0:
|
||||
color = 'yellow'
|
||||
|
||||
with open(icon_path, 'wb') as f:
|
||||
create_image(64, 64, alert_count(), color).save(f, 'PNG')
|
||||
return icon_path
|
||||
|
||||
#def menun2(deviceg):
|
||||
# itemn2 = []
|
||||
|
|
@ -107,17 +139,20 @@ def get_icon():
|
|||
|
||||
def build_menu():
|
||||
menu = Gtk.Menu()
|
||||
for device1 in conf:
|
||||
devname = conf[device1]['name']
|
||||
conf[device1]['menuitem'] = Gtk.MenuItem(devname)
|
||||
for device1 in conf['devices']:
|
||||
devname = conf['devices'][device1]['name']
|
||||
conf['devices'][device1]['menuitem'] = Gtk.MenuItem(label=devname)
|
||||
submenu = Gtk.Menu()
|
||||
for sensor in conf[device1]['values'].keys():
|
||||
conf[device1]['values'][sensor]['menuitem'] = Gtk.MenuItem(sensor)
|
||||
submenu.append(conf[device1]['values'][sensor]['menuitem'])
|
||||
conf[device1]['menuitem'].set_submenu(submenu)
|
||||
menu.append(conf[device1]['menuitem'])
|
||||
for sensor in conf['devices'][device1]['values'].keys():
|
||||
conf['devices'][device1]['values'][sensor]['menuitem'] = Gtk.MenuItem(label=conf['devices'][device1]['values'][sensor]['name'])
|
||||
submenu.append(conf['devices'][device1]['values'][sensor]['menuitem'])
|
||||
conf['devices'][device1]['menuitem'].set_submenu(submenu)
|
||||
menu.append(conf['devices'][device1]['menuitem'])
|
||||
|
||||
item_quit = Gtk.MenuItem('Quit')
|
||||
item_dashboard = Gtk.MenuItem(label='Open Dashboard')
|
||||
item_dashboard.connect('activate', gtkdashboard)
|
||||
menu.append(item_dashboard)
|
||||
item_quit = Gtk.MenuItem(label='Quit')
|
||||
item_quit.connect('activate', gtkquit)
|
||||
menu.append(item_quit)
|
||||
|
||||
|
|
@ -160,59 +195,59 @@ def build_menu():
|
|||
# return menu
|
||||
|
||||
def update_menu():
|
||||
appindicator.set_menu(build_menu())
|
||||
appindicator.set_menu(build_menu())
|
||||
|
||||
def set_value(topic, value):
|
||||
global oldalertcount
|
||||
topicds = topic.rsplit("/", 1)
|
||||
devices=topicds[0]
|
||||
devices = devices.removeprefix(mqttprefix+'/')
|
||||
conf[devices]['values'][topicds[1]]['value'] = float(value)
|
||||
print("menu")
|
||||
#update_menu()
|
||||
global oldalertcount
|
||||
topicds = topic.rsplit("/", 1)
|
||||
devices=topicds[0]
|
||||
devices = devices.removeprefix(mqttprefix+'/')
|
||||
conf['devices'][devices]['values'][topicds[1]]['value'] = float(value)
|
||||
#update_menu()
|
||||
|
||||
alert_text = get_alert2(devices, topicds[1])
|
||||
text='{}{}: {}'.format(alert_text, topicds[1], conf[devices]['values'][topicds[1]]['value'])
|
||||
conf[devices]['values'][topicds[1]]['menuitem'].set_label(text)
|
||||
#conf[devices]['values'][topicds[1]]['menuitem'].set_label("{}".format(conf[devices]['values'][topicds[1]]['value']))
|
||||
devname = conf[devices]['name']
|
||||
#print(get_alert(devices))
|
||||
#if get_alert(devices): devname = "! " + devname
|
||||
#if get_alert(devices): devname = "!!ALERT!! " + devname
|
||||
#conf[devices]['menuitem'].set_label(devname)
|
||||
alert_text = get_alert_text(devices, topicds[1])
|
||||
text='{}{}: {} {}'.format(alert_text, conf['devices'][devices]['values'][topicds[1]]['name'], conf['devices'][devices]['values'][topicds[1]]['value'], conf['devices'][devices]['values'][topicds[1]]['unit'])
|
||||
conf['devices'][devices]['values'][topicds[1]]['menuitem'].set_label(text)
|
||||
#conf['devices'][devices]['values'][topicds[1]]['menuitem'].set_label("{}".format(conf['devices'][devices]['values'][topicds[1]]['value']))
|
||||
devname = conf['devices'][devices]['name']
|
||||
#print(get_alert(devices))
|
||||
#if get_alert(devices): devname = "! " + devname
|
||||
#if get_alert(devices): devname = "!!ALERT!! " + devname
|
||||
#conf['devices'][devices]['menuitem'].set_label(devname)
|
||||
|
||||
if oldalertcount != alert_count():
|
||||
oldalertcount = alert_count()
|
||||
appindicator.set_icon(get_icon())
|
||||
os.system("notify-send '{}' '{}'".format(conf[devices]['name'], text))
|
||||
if oldalertcount != (alert_count()):
|
||||
oldalertcount = alert_count()
|
||||
appindicator.set_icon_full(get_icon(), "mqtt-icon")
|
||||
os.system("notify-send '{}' '{}'".format(conf['devices'][devices]['name'], text))
|
||||
|
||||
def subscribe_mqtt(mqttc):
|
||||
for device in conf:
|
||||
#print(conf[device])
|
||||
for sensor in conf[device]['values'].keys():
|
||||
#print(sensor)
|
||||
tosubscribe = "{}/{}/{}".format(mqttprefix, device, sensor)
|
||||
mqttc.subscribe(tosubscribe, 0)
|
||||
print(tosubscribe)
|
||||
for device in conf['devices']:
|
||||
#print(conf[device])
|
||||
for sensor in conf['devices'][device]['values'].keys():
|
||||
#print(sensor)
|
||||
tosubscribe = "{}/{}/{}".format(mqttprefix, device, sensor)
|
||||
mqttc.subscribe(tosubscribe, 0)
|
||||
print(tosubscribe)
|
||||
|
||||
def on_connect(mqttc, obj, flags, rc):
|
||||
print("rc: " + str(rc))
|
||||
#def on_subscribe(mqttc, obj, mid, granted_qos):
|
||||
# print("Subscribed: " + str(mid) + " " + str(granted_qos))
|
||||
def on_message(client, userdata, msg):
|
||||
print(msg.topic+" "+str(msg.payload))
|
||||
set_value(msg.topic, msg.payload)
|
||||
print(msg.topic+" "+str(msg.payload))
|
||||
set_value(msg.topic, msg.payload)
|
||||
|
||||
|
||||
signal.signal(signal.SIGINT, signal_handler)
|
||||
|
||||
appindicator.set_icon(get_icon())
|
||||
appindicator.set_icon_full(get_icon(), "mqtt-icon")
|
||||
appindicator.set_title('MQTT Alert')
|
||||
appindicator.set_status(AppIndicator.IndicatorStatus.ACTIVE)
|
||||
appindicator.set_menu(build_menu())
|
||||
|
||||
set_value("model1/1/1/temperature1C", "10.1")
|
||||
print(conf)
|
||||
set_value("model1/1/1/temperature1C", "25.1")
|
||||
#set_value("model2/2/2/humidity2", "75")
|
||||
#print(conf)
|
||||
|
||||
mqttc = mqtt.Client()
|
||||
mqttc.on_message = on_message
|
||||
|
|
@ -220,13 +255,13 @@ mqttc.on_connect = on_connect
|
|||
#mqttc.on_subscribe = on_subscribe
|
||||
connected = False
|
||||
while connected == False:
|
||||
try:
|
||||
mqttc.connect(mqtthost, 1883, 60)
|
||||
except:
|
||||
time.sleep(5)
|
||||
connected=False
|
||||
else:
|
||||
connected=True
|
||||
try:
|
||||
mqttc.connect(mqtthost, 1883, 60)
|
||||
except:
|
||||
time.sleep(5)
|
||||
connected=False
|
||||
else:
|
||||
connected=True
|
||||
mqttc.loop_start()
|
||||
|
||||
subscribe_mqtt(mqttc)
|
||||
|
|
|
|||
|
|
@ -1,22 +1,35 @@
|
|||
model1/1/1:
|
||||
name: "toto"
|
||||
values:
|
||||
temperature1C:
|
||||
name: "Temperature"
|
||||
min: 5
|
||||
max: 20
|
||||
humidity1:
|
||||
name: "Humidity"
|
||||
min: 30
|
||||
max: 70
|
||||
model2/2/2:
|
||||
name: "tata"
|
||||
values:
|
||||
temperature2C:
|
||||
name: "Temperature"
|
||||
min: 5
|
||||
max: 20
|
||||
humidity2:
|
||||
name: "Humidity"
|
||||
min: 30
|
||||
max: 70
|
||||
config:
|
||||
mqtt_host: "192.168.67.1"
|
||||
mqtt_prefix: "devices"
|
||||
dashboard_url: "https://github.com/NorthernMan54/rtl_433_ESP"
|
||||
devices:
|
||||
model1/1/1:
|
||||
name: "toto"
|
||||
values:
|
||||
temperature1C:
|
||||
name: "Temperature"
|
||||
unit: "°C"
|
||||
humidity1:
|
||||
name: "Humidity"
|
||||
unit: "%"
|
||||
min_critical: 20
|
||||
min_warning: 30
|
||||
max_warning: 70
|
||||
max_critical: 80
|
||||
model2/2/2:
|
||||
name: "tata"
|
||||
values:
|
||||
temperature2C:
|
||||
name: "Temperature"
|
||||
unit: "°C"
|
||||
min_critical: 0
|
||||
min_warning: 5
|
||||
max_warning: 20
|
||||
max_critical: 30
|
||||
humidity2:
|
||||
name: "Humidity"
|
||||
unit: "%"
|
||||
min_critical: 20
|
||||
min_warning: 30
|
||||
max_warning: 70
|
||||
max_critical: 80
|
||||
|
|
|
|||
|
|
@ -11,12 +11,19 @@ TOPIC_KEYS = [ "type", "model", "subtype", "channel", "id" ]
|
|||
#default="devices[/type][/model][/subtype][/channel][/id]"
|
||||
prefix = "home/rtl_433"
|
||||
|
||||
broker = "localhost"
|
||||
broker = "192.168.67.1"
|
||||
port = 1883
|
||||
|
||||
altitude = 150
|
||||
|
||||
serialdev = "/dev/ttyACM0"
|
||||
serialspeed = 115200
|
||||
|
||||
def get_humidex(tdb, rh):
|
||||
return round(tdb + 5 / 9 * ((6.112 * 10 ** (7.5 * tdb / (237.7 + tdb)) * rh / 100) - 10))
|
||||
|
||||
def get_sealevel_pressure(pres, altitude, temp):
|
||||
return (pres / pow(1 - ((0.0065 *altitude) / (temp + (0.0065 *altitude) + 273.15)), 5.257));
|
||||
|
||||
def generate_topic(jsonin):
|
||||
topic = prefix+'/devices'
|
||||
|
|
@ -59,7 +66,15 @@ while True:
|
|||
print("Error JSON, received {}".format(input))
|
||||
continue
|
||||
#print(data)
|
||||
mqtt.publish(prefix+'/events', input.rstrip())
|
||||
#mqtt.publish(prefix+'/events', input.rstrip())
|
||||
|
||||
if 'humidity' in data and 'temperature_C' in data:
|
||||
data['humidex'] = get_humidex(data['temperature_C'], data['humidity'])
|
||||
if 'pressure' in data:
|
||||
data['pressure'] = get_sealevel_pressure(data['pressure'], altitude, data['temperature_C'])
|
||||
data['pressure'] = round(data['pressure'], 1)
|
||||
|
||||
mqtt.publish(prefix+'/events', json.dumps(data))
|
||||
prefix_device = generate_topic(data)
|
||||
publish(data, prefix_device)
|
||||
|
||||
|
|
|
|||
|
|
@ -82,6 +82,16 @@ def on_message(mqttc, obj, msg):
|
|||
if 'humidity' not in recorddb[ref] or recorddb[ref]['humidity'] != value:
|
||||
measures += create_measure("humidity", currentts, tags, value)
|
||||
recorddb[ref]['humidity'] = value
|
||||
if 'moisture' in jsonin:
|
||||
value = int(jsonin['moisture'])
|
||||
if 'moisture' not in recorddb[ref] or recorddb[ref]['moisture'] != value:
|
||||
measures += create_measure("moisture", currentts, tags, value)
|
||||
recorddb[ref]['moisture'] = value
|
||||
if 'pressure' in jsonin:
|
||||
value = float(jsonin['pressure'])
|
||||
if 'pressure' not in recorddb[ref] or recorddb[ref]['pressure'] != value:
|
||||
measures += create_measure("pressure", currentts, tags, value)
|
||||
recorddb[ref]['humidity'] = value
|
||||
if 'battery_ok' in jsonin:
|
||||
value = int(jsonin['battery_ok'])
|
||||
if 'battery_ok' not in recorddb[ref] or recorddb[ref]['battery_ok'] != value:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue