77 lines
2.9 KiB
Python
77 lines
2.9 KiB
Python
#!/usr/bin/env python
|
|
|
|
import socketserver
|
|
import redis
|
|
|
|
|
|
class MyTCPHandler(socketserver.BaseRequestHandler):
|
|
"""
|
|
The RequestHandler class for our server.
|
|
|
|
It is instantiated once per connection to the server, and must
|
|
override the handle() method to implement communication to the
|
|
client.
|
|
"""
|
|
|
|
def _get_config(self):
|
|
self.max_height = int(self.r.hget('config', 'height'))
|
|
self.max_width = int(self.r.hget('config', 'width'))
|
|
self.max_framerate = int(self.r.hget('config', 'max_framerate'))
|
|
self.cur_framerate = int(self.r.hget('config', 'cur_framerate'))
|
|
self.imgsize = int(self.r.hget('config', 'imgsize'))
|
|
|
|
def _set_config(self, framerate, height, width):
|
|
self.r.hset('config', 'cur_framerate', framerate)
|
|
self.imgsize = height * width * 24
|
|
|
|
def _send_config_to_client(self):
|
|
self.request.sendall(self.max_height.to_bytes(4, byteorder='little'))
|
|
self.request.sendall(self.max_width.to_bytes(4, byteorder='little'))
|
|
self.request.sendall(self.max_framerate.to_bytes(4, byteorder='little'))
|
|
|
|
def _receive_client_config(self):
|
|
height = int.from_bytes(self.request.recv(4), byteorder='little')
|
|
width = int.from_bytes(self.request.recv(4), byteorder='little')
|
|
framerate = int.from_bytes(self.request.recv(4), byteorder='little')
|
|
print(height, width, framerate)
|
|
good, reason = self._check_config(height, width, framerate)
|
|
if good:
|
|
self._set_config(framerate, height, width)
|
|
return good, reason
|
|
|
|
def _check_config(self, height, width, framerate):
|
|
if height > 0 and height > self.max_height:
|
|
return False, "height has to be between 0 and {}. Current: {}".format(self.max_height, height)
|
|
if width > 0 and width > self.max_width:
|
|
return False, "width has to be between 0 and {}. Current: {}".format(self.max_width, width)
|
|
if framerate > 0 and framerate > self.max_framerate:
|
|
return False, "framerate has to be between 0 and {}. Current: {}".format(self.max_framerate, framerate)
|
|
return True, None
|
|
|
|
def handle(self):
|
|
self.r = redis.Redis()
|
|
self._get_config()
|
|
self._send_config_to_client()
|
|
good, reason = self._receive_client_config()
|
|
if not good:
|
|
print(reason)
|
|
return None
|
|
print('Start receiving from {}...'.format(self.client_address[0]))
|
|
while True:
|
|
data = self.request.recv(self.imgsize)
|
|
if len(data) == 0:
|
|
break
|
|
self.r.lpush('new', data)
|
|
print('... Done with {}.'.format(self.client_address[0]))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
HOST, PORT = "localhost", 9999
|
|
|
|
# Create the server, binding to localhost on port 9999
|
|
server = socketserver.TCPServer((HOST, PORT), MyTCPHandler)
|
|
|
|
# Activate the server; this will keep running until you
|
|
# interrupt the program with Ctrl-C
|
|
server.serve_forever()
|