class Server:
def __init__(self, TCP_PORT=8080, BUFFER_SIZE=32768):
self.BUFFER_SIZE = BUFFER_SIZE
self.server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.server_socket.bind(('', TCP_PORT))
self.server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR,
1)
self.server_socket.listen(5)
self.vf = Videofeed("server")
self.client_socket = None
self.ALARM_ON = False
self.PREV_ALARM_ON = False
print("\n[INFO] Host name: %s" % (socket.gethostname()))
print("[INFO] Server is listening on port: %s" % (TCP_PORT))
def send_alarm_status(self):
if self.PREV_ALARM_ON != self.ALARM_ON:
self.PREV_ALARM_ON = self.ALARM_ON
self.client_socket.send(str(self.ALARM_ON).encode('utf-8'))
def receive_data(self):
data = b''
while True:
part = self.client_socket.recv(self.BUFFER_SIZE)
data += part
if len(part) < self.BUFFER_SIZE: break
return data
def start(self):
self.client_socket, addr = self.server_socket.accept()
print("[INFO] connection received from: %s\n" % (addr[0])),
data = self.receive_data()
frame = self.vf.convert_to_frame(data)
self.detector = Detector(frame,
EYE_AR_THRESH=0.24,
ROLL_THRESH=15,
TIME_THRESH=3)
while self.client_socket:
try:
data = self.receive_data()
frame = self.vf.convert_to_frame(data)
if sys.getsizeof(data) < 40:
print("[INFO] Client stopped sending messages")
break
self.ALARM_ON = self.detector.detect_drowsiness(frame)
self.send_alarm_status()
if self.vf.show_frame(frame): break
except Exception as e:
print("[EXCEPTION] ", e)
break
self.stop()
def stop(self):
self.client_socket.send(b'eof')
self.client_socket.shutdown(socket.SHUT_RDWR)
self.client_socket.close()
self.server_socket.close()
