ota.py

try:
    import configparser
except ImportError:
    import ConfigParser as configparser

Import("env")

config = configparser.ConfigParser()
config.read("platformio.ini")
configkast = config.get("env", "target_kast").strip().splitlines(False)
brokerconfig = config.get("env", "broker").strip()

# Python callback
def on_upload(source, target, env):
    print("on_upload")
    fota(str(source[0]), configkast, env)

def on_ota(*args, **kwargs):
    print("on_ota")
    fota("$BUILD_DIR/firmware.bin", configkast, env)

def fota(file, kasten, env):
    for kast in kasten:
        print("Uploading %s to element %s..." % (file, kast))
        env.Execute("cat %s | mosquitto_pub -h %s -t mendeleev/%s/ota -s" % (file, brokerconfig, kast))

env.Replace(UPLOADCMD=on_upload)
env.AlwaysBuild(env.Alias("ota",
    "$BUILD_DIR/firmware.bin",
    on_ota))