class SocketServerThread(QThread):
def __init__(self, parent=None, readerThread=None):
QThread.__init__(self, parent)
self.readerThread = readerThread;
self.connected = False
self.timer = QTimer()
self.server = QLocalServer()
QLocalServer.removeServer('doorbotgui')
if not self.server.listen('doorbotgui'):
print(self.server.errorString())
return
botlog.info('SocketServerThread Initialized.')
def __del__(self):
botlog.info('SocketServer Thread Deletion')
def onTimer(self):
if self.connected:
self.sendCurrentTime()
self.sendCurrentSchedule()
def sendPacket(self, pkt):
if self.connected:
try:
self.client.write(json.dumps(pkt) + '\n')
self.client.flush()
except:
print('could not sendPacket')
def sendReadStatus(self):
pkt = {'cmd':'readstatus', 'status':str(self.readerThread.status)}
self.sendPacket(pkt)
def sendAccess(self, allowed, memberData):
pkt = {'cmd':'access', 'result':allowed, 'member':memberData}
self.sendPacket(pkt)
def sendCurrentTime(self):
pkt = {'cmd':'time', 'time':datetime.now().strftime("%A, %d %B %Y %H:%M:%S")}
self.sendPacket(pkt)
def sendCurrentSchedule(self):
pkt = {'cmd':'schedule', 'description':qsetup.schedule.scheduleDesc()}
self.sendPacket(pkt)
def onStatusChange(self, status):
self.sendReadStatus()
def onAccess(self, allowed, memberData):
self.sendAccess(allowed, memberData)
def onConnect(self):
botlog.info('New connection to SocketServer')
try:
self.client = self.server.nextPendingConnection()
self.client.disconnected.connect(self.onDisconnect)
except:
botlog.error('exception creating client')
self.connected = True
self.sendReadStatus()
self.sendCurrentSchedule()
def onDisconnect(self):
self.connected = False
self.client.deleteLater()
botlog.info('SocketServer disconnected')
def run(self):
botlog.info('SocketServer Thread Running')
if self.readerThread:
self.readerThread.signalStatusChange.connect(self.onStatusChange)
self.readerThread.signalAccess.connect(self.onAccess)
self.server.newConnection.connect(self.onConnect)
self.timer.timeout.connect(self.onTimer)
self.timer.start(1000)
self.exec()
botlog.info('SocketServerThread Stopped.')
class GUIServer():
def __init__(self):
self.heartbeats = 0
self.connected = False
self.timer = QTimer()
self.timer.timeout.connect(self.tick)
self.timer.start(500)
self.server = QLocalServer()
QLocalServer.removeServer('doorbotgui')
if not self.server.listen('doorbotgui'):
print(self.server.errorString())
return
self.server.newConnection.connect(self.handleConnection)
super(GUIServer, self).__init__()
def handleDisconnect(self):
self.connected = False
self.client.deleteLater()
def handleConnection(self):
self.client = self.server.nextPendingConnection()
self.client.disconnected.connect(self.handleDisconnect)
self.heartbeats = 0
self.connected = True
self.sendPacket({'cmd':'schedule', 'description':'Hobbyist and Pro Allowed Until 10PM'})
self.sendPacket({'cmd':'bulletin', 'source':'testmsg1.html'})
self.sendPacket({'cmd':'readstatus', 'status':'Please scan your RFID below.'})
def sendPacket(self, pkt):
if self.connected:
try:
self.client.write(json.dumps(pkt) + '\n')
self.client.flush()
except:
print('could not sendPacket')
def tick(self):
if self.connected:
self.heartbeats = self.heartbeats + 1
if self.heartbeats % 10 == 0:
heartbeat = {'cmd':'heartbeat', 'count': self.heartbeats}
self.sendPacket(heartbeat)
if self.heartbeats % 20 == 0:
self.sendPacket({'cmd':'bulletin', 'source':'testmsg1.html'})
else:
self.sendPacket({'cmd':'bulletin', 'source':'testmsg2.html'})
if self.heartbeats % 22 == 0:
access = {'cmd':'access', 'result':'allowed',
'member':{"tagid": "12345", "member": "Johnathan.Smith", "warning": "", "plan": "pro", "allowed": "allowed", "nickname": "Johnathan Smith"}
}
self.sendPacket(access)
if self.heartbeats % 33 == 0:
access = {'cmd':'access', 'result':'allowed',
'member':{"tagid": "12345", "member": "Bill.Jones", "warning": "Your membership expired (2016-01-15T00:04:25Z) and you are in a grace period for 14 days. Please update your payment information or you will lose access.", "plan": "hobbyist", "allowed": "allowed", "nickname": "Bill Jones"}
}
self.sendPacket(access)
if self.heartbeats % 55 == 0:
access = {'cmd':'access', 'result':'denied',
'member':{"tagid": "abcde", "member": "Al.Johnson", "warning": "Your membership expired (2012-07-02T00:04:25Z) and the grace period for access has ended. Contact [email protected] with any questions.", "plan": "hobbyist", "allowed": "denied", "nickname": None}
}
self.sendPacket(access)
if self.heartbeats % 77 == 0:
access = {'cmd':'access', 'result':'denied',
'member':{"tagid": "123abc", "member": "Michael.Michaelson", "warning": "You do not have access to this resource. See the Wiki for training information and resource manager contact info.", "plan": "hobbyist", "allowed": "denied", "nickname": None}
}
self.sendPacket(access)
time = {'cmd':'time', 'time':datetime.now().strftime("%A, %d %B %Y %H:%M:%S")}
self.sendPacket(time)