194 wiersze
6.4 KiB
Python
194 wiersze
6.4 KiB
Python
import paho.mqtt.client as mqtt
|
|
import json
|
|
import os
|
|
import logging
|
|
import signal
|
|
import requests
|
|
from datetime import datetime
|
|
from influxdb import InfluxDBClient
|
|
from dotenv import find_dotenv, load_dotenv
|
|
|
|
load_dotenv(find_dotenv())
|
|
SRC_MQTT_HOST = os.getenv("SRC_MQTT_HOST")
|
|
SRC_MQTT_USER = os.getenv("SRC_MQTT_USER")
|
|
SRC_MQTT_PASS = os.getenv("SRC_MQTT_PASS")
|
|
DST_MQTT_HOST = os.getenv("DST_MQTT_HOST")
|
|
DST_MQTT_USER = os.getenv("DST_MQTT_USER")
|
|
DST_MQTT_PASS = os.getenv("DST_MQTT_PASS")
|
|
DST_INFLUX_HOST = os.getenv("DST_INFLUX_HOST")
|
|
DST_INFLUX_USER = os.getenv("DST_INFLUX_USER")
|
|
DST_INFLUX_PASS = os.getenv("DST_INFLUX_PASS")
|
|
DST_INFLUX_DB = os.getenv("DST_INFLUX_DB")
|
|
HC_PING_URL = os.getenv("HC_PING_URL")
|
|
VERSION = "v2.0"
|
|
|
|
OT_TOPIC="owntracks/tobru/dragino"
|
|
OT_TID="dragino"
|
|
|
|
def on_connect_ttn(client, userdata, flags, rc):
|
|
logging.info("connected to ttn %s - %s", SRC_MQTT_HOST, str(rc))
|
|
client.subscribe("+/devices/+/up")
|
|
|
|
def on_connect_ot(client, userdata, flags, rc):
|
|
logging.info("connected to ot %s - %s", DST_MQTT_HOST, str(rc))
|
|
|
|
def on_publish_ot(client, userdata, rc):
|
|
logging.info("published data to ot")
|
|
|
|
def on_log(client, userdata, level, buf):
|
|
logging_level = mqtt.LOGGING_LEVEL[level]
|
|
logging.log(logging_level, buf)
|
|
#logging.info("got a log message level %s: %s", level, str(buf))
|
|
if "PINGRESP" in str(buf):
|
|
# report to https://healthchecks.io to tell that the connection is alive
|
|
requests.get(HC_PING_URL)
|
|
|
|
# The callback for when a PUBLISH message is received from the server.
|
|
def on_message_ttn(client, userdata, msg):
|
|
data = json.loads(msg.payload)
|
|
logging.info("message from ttn received for %s - #%s", data["dev_id"], data["counter"])
|
|
|
|
# retrieve info about gateway
|
|
gtw_id = data["metadata"]["gateways"][0]["gtw_id"]
|
|
try:
|
|
gtw_info = requests.get("https://www.thethingsnetwork.org/gateway-data/gateway/"+gtw_id).json()
|
|
logging.info("received via gw %s, %s, owned by %s",
|
|
data["metadata"]["gateways"][0]["gtw_id"],
|
|
gtw_info[gtw_id]["description"],
|
|
gtw_info[gtw_id]["owner"],
|
|
)
|
|
except:
|
|
logging.info("received via gw %s", gtw_id)
|
|
|
|
# max is 4 volts, 3 volts is considered empty
|
|
batpercent = round((data["payload_fields"]["BatV"] - 3) * 100)
|
|
|
|
if data["payload_fields"]["ALARM_status"]:
|
|
logging.info("Red button pushed!")
|
|
|
|
logging.info("Motion detection: %s", data["payload_fields"]["MD"])
|
|
logging.info("LED status for position: %s", data["payload_fields"]["LON"])
|
|
logging.info("Firmware version: %s", data["payload_fields"]["FW"])
|
|
|
|
got_fix = False
|
|
if data["payload_fields"]["Latitude"] == 0:
|
|
logging.info("no GPS data (Latitude) present")
|
|
# set GPS data to 0 for InfluxDB
|
|
data["payload_fields"]["Latitude"] = 0.0
|
|
data["payload_fields"]["Longitude"] = 0.0
|
|
else:
|
|
logging.info("GPS data (Latitude) present: lat %s, lon %s",
|
|
data["payload_fields"]["Latitude"],
|
|
data["payload_fields"]["Longitude"]
|
|
)
|
|
got_fix = True
|
|
# transform received data into OwnTracks format
|
|
ot_data = json.dumps({
|
|
"_type": "location",
|
|
"lat": data["payload_fields"]["Latitude"],
|
|
"lon": data["payload_fields"]["Longitude"],
|
|
"batt": batpercent,
|
|
"t": "p",
|
|
"tid": OT_TID,
|
|
"tst": int(datetime.timestamp(datetime.now())),
|
|
"conn": "m",
|
|
})
|
|
|
|
# publish to owntracks
|
|
logging.info("publishing data to owntracks via mqtt to topic %s", OT_TOPIC)
|
|
client_ot.publish(OT_TOPIC, payload=ot_data, retain=True, qos=1)
|
|
|
|
# write to influxdb
|
|
logging.info("writing data to influxdb")
|
|
influxdb.write_points(
|
|
[{
|
|
"measurement": "dragino",
|
|
"tags": {
|
|
"device": "lgt92",
|
|
},
|
|
"fields": {
|
|
"bat": float(data["payload_fields"]["BatV"]),
|
|
"pitch": float(data["payload_fields"]["Pitch"]),
|
|
"roll": float(data["payload_fields"]["Roll"]),
|
|
"lat": float(data["payload_fields"]["Latitude"]),
|
|
"lon": float(data["payload_fields"]["Longitude"]),
|
|
"alarm": int(data["payload_fields"]["ALARM_status"]),
|
|
"counter": data["counter"],
|
|
"airtime": data["metadata"]["airtime"],
|
|
"rssi": data["metadata"]["gateways"][0]["rssi"],
|
|
"fix": got_fix,
|
|
}
|
|
}]
|
|
)
|
|
|
|
logging.info("data processing done")
|
|
|
|
def shutdown():
|
|
logging.info("disconnecting from mqtt")
|
|
client_ot.disconnect()
|
|
client_ot.loop_stop()
|
|
client_ttn.disconnect()
|
|
client_ttn.loop_stop()
|
|
|
|
def handleSIGTERM(signalNumber, frame):
|
|
logging.info("got SIGTERM")
|
|
shutdown()
|
|
return
|
|
|
|
if __name__ == '__main__':
|
|
|
|
signal.signal(signal.SIGTERM, handleSIGTERM)
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s - %(message)s',
|
|
datefmt='%Y-%m-%d %H:%M:%S %Z'
|
|
)
|
|
|
|
logging.info("Starting ioteer lgt92. "+VERSION)
|
|
|
|
# Prepare InfluxDB
|
|
influxdb = InfluxDBClient(
|
|
host=DST_INFLUX_HOST,
|
|
port=443,
|
|
database=DST_INFLUX_DB,
|
|
username=DST_INFLUX_USER,
|
|
password=DST_INFLUX_PASS,
|
|
ssl=True,
|
|
verify_ssl=True,
|
|
)
|
|
|
|
# Prepare MQTT for The Things Network
|
|
client_ttn = mqtt.Client()
|
|
client_ttn.enable_logger()
|
|
client_ttn.on_connect = on_connect_ttn
|
|
client_ttn.on_message = on_message_ttn
|
|
client_ttn.on_log = on_log
|
|
client_ttn.username_pw_set(SRC_MQTT_USER,SRC_MQTT_PASS)
|
|
client_ttn.tls_set()
|
|
client_ttn.connect(SRC_MQTT_HOST, 8883, 60)
|
|
|
|
# Prepare MQTT for OwnTracks
|
|
ot_lwt = json.dumps({
|
|
"_type": "lwt",
|
|
"tst": int(datetime.timestamp(datetime.now())),
|
|
})
|
|
client_ot = mqtt.Client()
|
|
client_ot.enable_logger()
|
|
client_ot.on_connect = on_connect_ot
|
|
client_ot.on_publish = on_publish_ot
|
|
client_ot.on_log = on_log
|
|
client_ot.username_pw_set(DST_MQTT_USER,DST_MQTT_PASS)
|
|
client_ot.tls_set()
|
|
client_ot.will_set(OT_TOPIC, payload=ot_lwt, qos=1, retain=True)
|
|
client_ot.connect(DST_MQTT_HOST, 8883, 60)
|
|
|
|
try:
|
|
# Connect to MQTT and react to messages
|
|
client_ot.loop_start()
|
|
client_ttn.loop_forever()
|
|
except KeyboardInterrupt:
|
|
shutdown()
|
|
logging.info("tschuess")
|
|
|