from common.models import *
from .Device import DeviceController

'''
    Device controller for PLC.
    This class must implement set, get and do functions (DeviceController is an abstract)
'''
class PLCController(DeviceController):

    def __init__(self):
        super().__init__("PLC")

    def get(self, command: str, *args):
        message = "GET " + command + " " + " ".join([str(arg) for arg in args])
        self.sendMessage(message)
        return (self.blockAndReadMessage())

    def set(self, command: str, *args):
        message = "SET " + command + " " + " ".join([str(arg) for arg in args])
        return (self.sendMessage(message))

    def do(self, command: str, *args):
        message = "DO " + command + " " + " ".join([str(arg) for arg in args])
        return (self.sendMessage(message))