adapt to ttnv3 and black
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
Tobias Brunner 2021-10-18 07:47:23 +02:00
parent 157371ff59
commit 328ebe7dd2

155
lgt92.py
View file

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