plcSimulator.py 2.02 KB
import sys
import os

PACKAGE_PARENT = '..'
SCRIPT_DIR = os.path.dirname(os.path.realpath(os.path.join(os.getcwd(), os.path.expanduser(__file__))))
sys.path.append(os.path.normpath(os.path.join(SCRIPT_DIR, PACKAGE_PARENT)))

from utils.Device import Device
from utils.StatusManager import StatusManager

class PLCSimulator(Device, StatusManager):
    status = {"status" : "VALID", "presence" : "No", "pressure" : 1013.25, "wind_dir" : "N", "cloud" : 0, "rain" : 0}

    def __init__(self, argv):
        super().__init__(argv)
        self.setDevice("PLC")
        self.setStatusManager("plcSimulator", argv)

    def plcPrint(self, string: str):
        print("PLCSimulator : " + string)

    def handleConnection(self):
        data = self.readBytes(128)

        answer = ""
        cut_data = data.split(" ")

        if (len(cut_data) < 2):
            self.plcPrint("Received invalid message on socket : " + data)
            return (0)
        cmd_type = cut_data[0]
        cmd = cut_data[1]
        args = cut_data[2:]
        if (cmd_type == "GET"):
            if (cmd == "STATUS"):
                answer = self.create_status()
            else:
                answer = "Invalid cmd for GET: " + cmd
                self.plcPrint(answer)
        else:
            self.plcPrint("Ignored message " + data)

        if (len(answer) > 0):
            self.sendMessage(answer)
        return (0)

    def loop(self):
        i = 0

        if (self.ended == 0):
            self.plcPrint("Not entry for telescope found in config file : " + self.config_file)
            return (0)
        while (True):
            self.updateStatus(i)
            if (self.isConnected()):
                self.handleConnection()
                self.close()
            i += 1
            if (i > self.ended):
                return (0)
        return (1)

    def run(self):
        print("PLC simulator running")
        self.parse()
        self.configSocket()
        self.loop()
        return (0)

if __name__ == "__main__":
    sim = PLCSimulator(sys.argv)
    sim.run()