from socketserver import BaseRequestHandler, TCPServer
import db


class EchoHandler(BaseRequestHandler):
    def handle(self):
        print("Got connection from", self.client_address)
        host = self.request.recv(1024)
        print("Host is {host}".format(host=host))
        if host:
            latest_speed_dic = db.get_latest_speed()
            for server in latest_speed_dic:
                if host == server.encode("ascii"):
                    speed = int(server["speed"])
                    self.request.sendall(
                        "{weight}%\n".format(weight=speed).encode("ascii")
                    )
                    print("send to bwg with wight {weight}".format(weight=speed))
                    break
        self.request.close()


if __name__ == "__main__":
    serv = TCPServer(("", 20000), EchoHandler)
    serv.serve_forever()