library.py

import json
import requests
from datetime import datetime
import sys
import os


def udpateFileNum(i):
    global fileNum
    fileNum = i

def getIniParameters(sFile):
    """
    Read parameters in ini file
    @param sFile Path to the ini file to read
    @return Dictionary with sections and parameters contained in the INI file
    """
    import os,sys
    sCurrentPath = os.path.abspath(os.path.dirname(sys.argv[0]))
    os.chdir(sCurrentPath)
    sIniFile = sFile
    import configparser as cp
    cfg = cp.ConfigParser()

    with open(sIniFile) as f:
        cfg.read_file(f)

    # https://stackoverflow.com/a/28990982
    return {s:dict(cfg.items(s)) for s in cfg.sections()}

def printlog(s, sep = ": "):
    """
    Print message with date and time and flush the console
    @see https://www.turnkeylinux.org/blog/unix-buffering
    """
    sDateTime = datetime.now().strftime("%Y/%m/%d %H:%M:%S")
    print(sDateTime + sep + s)
    sys.stdout.flush()

def PostThingsboard(JsDATA, DeviceId):
    dPrm = getIniParameters("commissioning.ini")
    headers_AccessT = {
    'content-type': "application/json",
    'accept': "application/json"
    }

    printlog("1/6 - Send request to Access Token")
    #print(dPrm['THINGS']['url']+'/api/auth/login/public')
    try:
        ResponseAccessToken = requests.post(dPrm['THINGS']['url']+'/api/auth/login', headers=headers_AccessT, data='{"username":"' + dPrm['THINGS']['username'] + '", "password":"' + dPrm['THINGS']['password'] + '"}')
    except requests.exceptions.RequestException as e:
        printlog("Error " + e)
        return

    if 'status' in ResponseAccessToken.text:
        printlog(ResponseAccessToken.text)
        return
    else:
        printlog("2/6 - Request response ok")

    try:
        AccessToken=eval(ResponseAccessToken.text)['token']
    except:
        printlog("Error retrieving token in " + ResponseAccessToken.text)
        return

    headers_DeviceT = {
            'Accept': 'application/json',
            'X-Authorization': "Bearer " +AccessToken
            }

    printlog("3/6 - Send request to Device token")

    try:
        ResponseDeviceToken = requests.get(dPrm['THINGS']['url']+'/api/device/'+DeviceId+'/credentials', headers=headers_DeviceT)
    except requests.exceptions.RequestException as e:
        printlog("Error " + e)
        return

    if 'status' in ResponseDeviceToken.text:
        printlog(ResponseAccessToken.text)
        return
    else:
        printlog("4/6 - Request response ok")

    null = None

    try:
        DeviceToken=str(eval(ResponseDeviceToken.text)['credentialsId'])
    except:
        printlog("Error retrieving credentials Id")

    printlog("5/6 - Send request telemetry")
    try:
        r = requests.post(dPrm['THINGS']['url']+'/api/v1/'+DeviceToken+'/telemetry', json=JsDATA)
    except requests.exceptions.RequestException as e:
        printlog("Error " + e)

    printlog("6/6 - Request code response "+ ("ok" if r.status_code == 200 else str(r.text)))

def on_log( client, userdata, level, buf ):
    printlog( "log: "+ buf)

def on_connect( client, userdata, flags, rc ):
    printlog( "Connexion MQTT: code retour= "+ str(rc) )
    printlog( "Connexion MQTT: Statut= " +("OK" if rc==0 else "echec") + "\n" )
    #print("\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\n")

def on_message(client, userdata, message):
    """
    application that processes data when a message is
    received and starts the post procedure on ThingsBoard
    """
    msg = json.loads(message.payload.decode("utf-8" ))
    printlog( "Receipt message MQTT... (application : " + msg["end_device_ids"]["application_ids"]["application_id"] + ", pid : "+ str(os.getpid()) + ")" )
    printlog( "Topic : "+ str(message.topic) )
    #printlog("dev_if : " +y["dev_id"])
    #printlog("Latitude: " +y["payload_fields"]["latitude"])
    #printlog("Longitude: " +y["payload_fields"]["longitude"])
    #printlog("detection: %s" +y["payload_fields"]["detection"])

    #printlog("data: "+str(msg["payload_fields"]))
    printlog("data: "+str(msg["uplink_message"]["decoded_payload"]))
    PostThingsboard(msg["uplink_message"]["decoded_payload"], msg["end_device_ids"]["device_id"])