__author__ = 'nrees'
import threading
import socket

class PlugPoller(threading.Thread):

    def __init__(self, ip, port):
        threading.Thread.__init__(self)
        self.ip = ip
        self.port = port
        self.connect(ip, port)
        self.status = '0000'
        self.command = '!getstat\r'
        self.loop = True
        self.statusChangeCallback = None
    def connect(self, ip, port):
        try:
            self.s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            self.s.settimeout(1)
            self.s.connect((ip, port))
        except:
            pass

    def run(self):
        while self.loop:
            try:
                sleep(.1)
                self.s.send(self.command)
                self.status = str(self.s.recv(4))
                # print "data:{"+self.command,self.status+"}"
            except:
                # print 'plug connection error'
                sleep(5.0)
                self.connect(self.ip, self.port)
                pass
            if (self.command is not '!getstat\r'):
                # print self.command
                self.command = '!getstat\r'

    def getstatus(self):
        return self.status

    def toggleAll(self):
        self.sendcmd('!toggle1\r')
        self.sendcmd('!toggle2\r')
        if self.statusChangeCallback:
            self.statusChangeCallback('{"Lamps":"'+self.getstatus()+'"}')
    def sendcmd(self, c):
        try:
            self.s.send(c)
        except:
            pass
        return self.s.recv(4)