io-watch.py

# Ugly code following!     
import time
import sys
import array
import pymodbus
import serial
import argparse

from pymodbus.pdu import ModbusRequest
from pymodbus.client import ModbusSerialClient as SerialModbusClient
from pymodbus.client import ModbusTcpClient as TCPModbusClient
from pymodbus.transaction import ModbusRtuFramer
#import logging
#logging.basicConfig()
##log = logging.getLogger()
#log.setLevel(logging.DEBUG)


baudrate = 38400
parity = 'N'
port = "/dev/ttyUSB0"
slave=0 # the slave save this request is targeting

global newAdr

global oldDiscretes
global oldCoils
global coilsChanged
global discretesChanged

def probe():
        try:
                result0 = client.read_coils(2000,1,slave=unit)
                client.close()
        except:
                print ("Serial error. Is "+port+" available?")
                client.close()
        else:
                try:
                        result0.string
                except:
                        client.close()
                        return True
                else:
                        client.close()
                        print("Client unreachable!")
                        return False

def printStates(outs,ins):
    sys.stdout.write("\n")
    sys.stdout.write("Device address: "+str(unit)+"\n")
    sys.stdout.write("Outputs        Inputs\n")
    for x in range(0,16):
        sys.stdout.write(outs[x])
        if(outs[x]=="On "):
            sys.stdout.write(" ")
        sys.stdout.write("            "+ins[x])
        if(ins[x]=="On "):
            sys.stdout.write(" ")
        sys.stdout.write("\n")
    for x in range(0,19):
        sys.stdout.write("\033[F")

    sys.stdout.flush()


def monitor():

#       client = ModbusClient(method = "rtu", port = port, stopbits = 1, bytesize = 8, parity = parity, baudrate = baudrate)
        #if True:
        try:
                #client = ModbusClient(method = "rtu", port = port, stopbits = 1, bytesize = 8, parity = parity, baudrate = baudrate, timeout=0.5)
                #connection = client.connect()
                
                oldCoils = client.read_coils(0,16,slave=unit)
                oldDiscretes = client.read_discrete_inputs(0,16,slave=unit)
                discretesChanged = [False]*16
                coilsChanged = [False]*16
                
                blahvar = 0
                killvar = 1
                while killvar == 1:
                    if True:#try:
                        coils = client.read_coils(0,16,slave=unit)
                        discretes = client.read_discrete_inputs(0,16,slave=unit)
                        outs = ["Off"]*16
                        ins = ["Off"]*16
                        for x in range(0,16):
                            if(coils.bits[x] != oldCoils.bits[x]):
                                coilsChanged[x] = True
                            if(coils.bits[x] == True):
                                outs[x]="On "
                            if(coilsChanged[x] == True):
                                outs[x]=u"\u001b[31m"+outs[x]+u"\u001b[0m"

                        for x in range(0,16):
                            if(discretes.bits[x] != oldDiscretes.bits[x]):
                                discretesChanged[x] = True
                            if(discretes.bits[x] == True):
                                ins[x]="On "
                            if(discretesChanged[x] == True):
                                ins[x]=u"\u001b[31m"+ins[x]+u"\u001b[0m"
                    #except:
                       # print("Polling error!")
                       # outs = ["Err"]*16
                       # ins = ["Err"]*16
                    
                    printStates(outs,ins)
                    #blahvar=blahvar+4
                    #print(blahvar)
                    #kin=sys.stdin.readline()
                    #if(kin>0):
                       # killvar=9
                      #  sys.exit(1)

#sys.stdout.write("das"+u"\u001b[31m"+"blah"+u"\u001b[0m"+"hundszahn")
                        
                    
                client.close()
        except Exception as e:
                print("Polling error: "+str(e))
        client.close()

parser = argparse.ArgumentParser()

parser.add_argument("address", type=int, help="device address")
parser.add_argument("port", type=str, help="serial port like /dev/ttyUSB0")
args = parser.parse_args()
unit = args.address

client = SerialModbusClient(method = "rtu", port = args.port, stopbits = 1, bytesize = 8, parity = parity, baudrate = baudrate, timeout=1.0)
try:
    connection = client.connect()
except:
    print("Cannot open serial port. Is "+port+" available?")
    exit()



if probe():
    monitor()