torch-subscriber-simple/torch_sub/torch_sub.py

import json
import os

import paho.mqtt.subscribe as mqtt

database_file = "clients.json"


# noinspection PyUnusedLocal
def update_client_record(client, userdata, message):

    if not os.path.exists(database_file):
        with open(database_file, 'w') as database_blank:
            database_blank.write("{}")

    with open(database_file, 'r') as infile:
        database = json.load(infile)

    payload = message.payload.decode('utf-8')
    response = json.loads(payload)

    database[response['clientId']] = response

    with open(database_file, 'w') as outfile:
        json.dump(database, outfile)


def subscribe(broker_hostname, broker_port, topic="torch", ca_file=None, cert_file=None, key_file=None):
    mqtt.callback(update_client_record,
                  topic,
                  hostname=broker_hostname,
                  port=broker_port,
                  tls={
                      'ca_certs': ca_file,
                      'certfile': cert_file,
                      'keyfile': key_file
                  })