Modbus

# Modbus.py
# Implementacion de funciones modbus sobre TCP/IP Usando Sockets

import sys
import socket
import array
from struct import *
import time
from PyQt4 import QtCore, QtGui
from PyQt4.QtCore import *
import ConfigParser

class Modbus(QObject):
    debug_mode = False

    def __init__(self,parent = None):
        QObject.__init__(self,parent)
        self.parent = parent

        self.s = socket.socket()

        self.config = ConfigParser.ConfigParser()
        self.config.readfp(open('main.conf'))
        self.conectar()

    def conectar(self):
        try:
            self.s = socket.socket()

            self.s.settimeout(4)
            print "connect("+str(self.config.get('Modbus', 'IP'))+","+str(self.config.get('Modbus', 'PORT'))
            self.s.connect((str(self.config.get('Modbus', 'IP')), int(self.config.get('Modbus', 'PORT'))))
            print "Done"
            """
            self.client = ModbusClient(str("192.168.0.201"),8000) """

        except socket.error, e: ## Entra en un loop de reintentos hasta que se comunica
            if self.criticalMessage(e) == 1:
                print("Aborte")
            else:
                self.conectar()

    def getInt32Bits(self, dir = 0,  var = 1):
        try:
            mensaje ="\x00\x00\x00\x00\x00\x06\x01\x03"+pack('!h',dir )+pack('!h',int(var) )
            self.s.send(mensaje)
            time.sleep(1)
            respuesta=""
            MSGLEN = (11)
            while len(respuesta) < MSGLEN:
                chunk = self.s.recv(MSGLEN-len(respuesta))
                respuesta = respuesta + chunk

        except socket.error, e: ## Entra en un loop de reintentos hasta que se comunica
            if self.criticalMessage(e) == 1:
                print "Aborte"
            else:
                self.getInt32Bits(dir,var)

        if(self.debug_mode==True):
            print "Modbus.py - getInt32Bits("+str(dir)+")"
            print "TX: "+str(repr(mensaje))
            print "RX: "+str(repr(respuesta))
        return unpack('!h', respuesta[9]+respuesta[10])[0]

    def getInt32BitsArray(self, dir = 0,  var = 1):  # IMPLEMENTAR, Arreglar!
        try:
            CountA = 0
            original = array.array('f')
            mensaje ="\x00\x00\x00\x00\x00\x06\x01\x03"+pack('!h',dir)+pack('!h',var)
            self.s.send(mensaje)

            respuesta = self.s.recv(9+var*2)

            print repr(respuesta)
            while CountA<=(var-1)*2:
                original.append(unpack('!h', respuesta[9+CountA]+respuesta[10+CountA])[0])
                CountA = CountA + 2


        except socket.error,  e:
            if self.criticalMessage(e) == 1:
                print "Aborte"
            else:
                self.getInt32Bits(dir,var)
        return original

    def setInt32Bits(self, dir = 0,  var =0):
        #print ("setInt32Bits")
        mensaje ="\x00\x00\x00\x00\x00\x06\x01\x06"+pack('!h',dir)+pack('!h',var)
        try:
            self.s.send(mensaje)
        except socket.error, e:
            print "Error"
            return "Error"

        #respuesta = self.s.recv(11)
        respuesta=""
        MSGLEN = (12)
        while len(respuesta) < MSGLEN:
            chunk = self.s.recv(MSGLEN-len(respuesta))
            if chunk == '':
                raise RuntimeError, "conexion interrumpida"
            respuesta = respuesta + chunk

        if(self.debug_mode==True):
            print "Modbus.py - setInt32Bits("+str(dir)+","+str(var)+")"
            print "TX: "+str(repr(mensaje))
            print "RX: "+str(repr(respuesta))

    def setInt32BitsArray(self, dir = 0,  var =0):
        #print ("setInt32BitsArray")
        mensaje = "\x00\x00\x00\x00"+pack('!h',7+len(var)*2)+"\x01\x10"+pack('!h',dir)+pack('!h',len(var))+pack('B',len(var)*2)
        for countA in range(len(var)):
            mensaje = mensaje + pack('!h',var[countA])
        self.s.send(mensaje)

    def cerrar(self):
        self.s.close()

    def setText(self, dir = 0,  var =0):

        cadena =  self.netorderString(var)
        print (7+len(cadena))
        mensaje = "\x00\x00\x00\x00"+pack('!h',7+len(cadena))+"\x01\x10"+pack('!h',dir)+pack('!h',len(cadena)/2)+pack('B',len(cadena))+cadena
        self.s.send(mensaje)

        print repr(mensaje)

    def netorderString(self, original):
        original = array.array('c', original)
        countA = 0
        if (len(original.tostring())%2 != 0):
            original.append(" ")

        while countA <= len(original.tostring())-1:
            var = original[countA]
            original[countA] = original[countA+1]
            original[countA+1] = var
            #print countA
            #print str(original)
            countA = countA + 2

        return original.tostring()

    def criticalMessage(self,message):
        reply = QtGui.QMessageBox.critical(None, "MODBUS ERROR",
                "Error de comunicacion:\n"+str(message),
                QtGui.QMessageBox.Abort | QtGui.QMessageBox.Retry )
        if reply == QtGui.QMessageBox.Abort:
            return 1    #Retorno Abortar
        else:
            return 0    #Retorno Reintentar

#mo = Modbus()

#mo.setInt32BitsArray(115, (2, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 16, 17, 18, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 16, 17, 18, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18))
#while (True):
#    print str(mo.getInt32Bits(297)) + str(mo.getInt32Bits(298)) + str(mo.getInt32Bits(299))
#while True:
#    print mo.getInt32BitsArray(297, 3)
#    time.sleep(0.5)
#mo.cerrar()