main.py

import json
import os
from dotenv import load_dotenv
from ruuvitag_sensor.ruuvi import RuuviTagSensor
from influxdb import InfluxDBClient

load_dotenv()
client = InfluxDBClient(host=os.getenv('INFLUX_HOST'), port=os.getenv('INFLUX_PORT'), database=os.getenv('INFLUX_DB'))

def handle_data(received_data):
    fields = {}
    mac = received_data[0]
    payload = received_data[1]

    dataFormat = payload['data_format'] if 'data_format' in payload else None
    field_map = json.loads(os.getenv('RUUVI_FIELDMAP'))
    for field_label in field_map:
        fields[field_label] = payload[field_label] if field_label in payload else None
    print(fields)

    json_body = [
        {
            'measurement': 'ruuvi_measurements',
            'tags': {
                'mac': mac,
                'dataFormat': dataFormat
            },
            'fields': fields
        }
    ]
    try:
        client.write_points(json_body)
    except Exception as e:
        print(e)

macs = json.loads(os.getenv('APPROVED_MACS'))
RuuviTagSensor.get_datas(handle_data, macs)