loco-tester.py

#!/usr/bin/python3

import paho.mqtt.client as mqtt
import time
import sys

assert(len(sys.argv) >= 2)
motors = ['', 'motors/a', 'motors/b', 'motors/c', 'motors/d']

laspeed = int(sys.argv[1])

m1 = 'motors/a'
m2 = 'motors/b'
m3 = motors[3]
m4 = motors[4]
#power = int((int(sys.argv[2])+100)/200 * 255)
power = int(sys.argv[2])
client = mqtt.Client()
client.connect('127.0.0.1')

client.publish("loco/{}".format(m1), payload=power, qos=0, retain=False)
client.publish("loco/{}".format(m2), payload=power, qos=0, retain=False)
client.publish("loco/{}".format(m3), payload=power, qos=0, retain=False)
client.publish("loco/{}".format(m4), payload=power, qos=0, retain=False)
# time.sleep(0.01)
print("motor: {} power: {}".format(m1, power))
print("motor: {} power: {}".format(m2, power))
print("motor: {} power: {}".format(m3, laspeed))

if (len(sys.argv) > 2):
    if (float(sys.argv[3]) > 0):
        time.sleep(float(sys.argv[3]))
        client.publish("loco/{}".format(m1), payload=0x00, qos=0, retain=False)
        client.publish("loco/{}".format(m2), payload=0x00, qos=0, retain=False)
        client.publish("loco/{}".format(m3), payload=0x00, qos=0, retain=False)
        client.publish("loco/{}".format(m4), payload=0x00, qos=0, retain=False)
        print("stopping")