RLY8Class.py

import socket
import json
ADDR = ("192.168.0.202", 2000)


class RLY8:

    def __init__(self, ADDR=ADDR):
        self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        self.s.settimeout(1) # Set the timeout to 1 second
        self.s.connect(ADDR)

        # get name
        self.getName()

        # get netconfig
        self.s.sendall(b'{"get":"netconfig"}')
        self.netconf = self.returnResponseJSON()

        # get version
        self.s.sendall(b'{"get":"version"}')
        self.version = self.returnResponseJSON()['version']

        # get baudrate
        self.s.sendall(b'{"get":"baudrate"}')
        self.baudrate = self.returnResponseJSON()['baudrate']

        # get relays' status
        self.getRelayStatus()

    def getName(self):
        self.s.sendall(b'{"get":"name"}')
        self.name = self.returnResponseJSON()['name']

    def setName(self, name):
        self.s.sendall((f'{{"name":"{name}"}}').encode())
        if self.verifyResponse():
            self.name = name

    def setBaudrate(self, baudrate):
        self.s.sendall((f'{{"baudrate":"{baudrate}"}}').encode())
        if self.verifyResponse():
            self.baudrate = baudrate

    def setRelayStatus(self, relaynr, status):
        self.s.sendall((f'{{"relay{relaynr}":"{status}"}}').encode())
        # must run
        self.returnResponseJSON()
        # reread relay statuses
        self.getRelayStatus()

    def getRelayStatus(self):
        self.s.sendall(b'{"get":"relayStatus"}')
        self.status = self.returnResponseJSON()
        # separate relay statuses
        self.relay1 = self.status['relay1']
        self.relay2 = self.status['relay2']
        self.relay3 = self.status['relay3']
        self.relay4 = self.status['relay4']
        self.relay5 = self.status['relay5']
        self.relay6 = self.status['relay6']
        self.relay7 = self.status['relay7']
        self.relay8 = self.status['relay8']

    # verify 'ok', 'error' responses
    def verifyResponse(self):
        if self.returnResponseJSON()['resp'] == 'ok':
            return True
        return False

    # get socket response and jsonify
    def returnResponseJSON(self):
        result = self.s.recv(1024).decode().split('\x00', 1)[0]
        return json.loads(result)