caveControl.py

#!/usr/bin/env python

#GUI
import wx
from data.layout import maingui

#transmissions
from threading import Thread, active_count
import socket       #for christie
import telnetlib    #for marantz

class Transmit():
    def __init__(self):
        self.callback = None
        #projectors
        self.ips = ["172.22.1." + str(i) for i in range(101,111)]
        
        #audio
        self.audioIP = "172.22.1.50"

    #dispatch to all projectors
    def allProjectors(self, command):
        for i in range(len(self.ips)):
            thr = Thread(target=self._transmitThread, args=(i, command))
            thr.start()
    
    #dispatch to 1 projector
    def projector(self, id, command):        
        thr = Thread(target=self._transmitThread, args=(id, command))
        thr.start()
        
    def _transmitThread(self, id, command):    
        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        s.settimeout(5.0)  
        ip = self.ips[int(id) - 1]
        try:
            s.connect((ip, 3002))
            s.send(("(#" + command + ")").encode())
            response = str(s.recv(1024).decode())
            self.callback.gridUpdate(id, response)
        except socket.timeout:
            self.callback.gridUpdate(id, "Timeout")
        s.close()        
    
    def audio(self, command):
        thr = Thread(target=self._transmitAudio, args=(command,))
        thr.start()
        
    def _transmitAudio(self, command):
        #is the connection still open?
        try: 
            self.tn.sock_avail()
        except:
            self.tn = telnetlib.Telnet()
            try:
                self.tn.open(self.audioIP, 23, 3)
            except:
                return

        try:
            self.tn.write((command + "\r").encode())
            #print(self.tn.read_eager().decode())
            #print(response)
            self.callback.gridUpdate(10, command)
        except:
            self.callback.gridUpdate(10, "Message error.")

    
class gui(maingui): 
    def __init__(self,parent):
        #initialize widgets
        maingui.__init__(self,parent)
        
        #initialize grid statuses
        transmit.callback = self
        transmit.allProjectors("PWR?")
        transmit.audio("PW?")        

    def togglePower(self, event):
        if event.IsChecked():
            event.GetEventObject().SetLabel("ON")
            transmit.allProjectors("PWR1")
        else:
            #confirm power off, because the cooldown takes uninterruptible minutes
            dlg = wx.MessageDialog(None, "Are you sure you want to shut down the projectors?",'Confirm', wx.YES_NO | wx.ICON_QUESTION)
            result = dlg.ShowModal()
            if result == wx.ID_YES:
                event.GetEventObject().SetLabel("OFF")
                transmit.allProjectors("PWR0")
            else:
                event.GetEventObject().SetValue(True)

    def toggleShutters(self, event):
        if event.IsChecked():
            event.GetEventObject().SetLabel("OPEN") 
            transmit.allProjectors("SHU0")
        else:
            event.GetEventObject().SetLabel("CLOSED")
            transmit.allProjectors("SHU1")
            
    def toggleFloor(self, event):
        if event.IsChecked():
            event.GetEventObject().SetLabel("ON")
            transmit.projector(9,  "PWR1")
            transmit.projector(10, "PWR1")
        else:
            event.GetEventObject().SetLabel("OFF")
            transmit.projector(9,  "PWR0")
            transmit.projector(10, "PWR0")

    def toggleStereo(self, event):
        if event.IsChecked():
            event.GetEventObject().SetLabel("ON")
            transmit.allProjectors("TDM+MAIN5")            
        else:
            event.GetEventObject().SetLabel("OFF")
            transmit.allProjectors("TDM+MAIN0")

    def toggleAudio(self, event):
        if event.IsChecked():
            event.GetEventObject().SetLabel("ON")
            transmit.audio("PWON")
        else:
            event.GetEventObject().SetLabel("OFF")
            transmit.audio("PWSTANDBY")

    def updateVolume(self, event):
        volume = "MV" + str(event.GetEventObject().GetValue()).zfill(2)
        transmit.audio(volume)

    #the threads call this when they're done
    def gridUpdate(self, id, message):
        self.grid.SetCellValue(row=id, col=0, s=message)
        
    def quit(self, event):
        transmit.tn.close()
        wx.Exit()
        

if __name__ == '__main__':
    app = wx.App(False)
    transmit = Transmit()        
    frame = gui(None)
    frame.Show()
    app.MainLoop()