From 6204d44c8b473a1c30e56555a3919a8dd185036d Mon Sep 17 00:00:00 2001 From: facat Date: Tue, 18 Jun 2019 18:18:53 +0800 Subject: [PATCH] =?UTF-8?q?1.=E6=B7=BB=E5=8A=A0=E6=95=B0=E6=8D=AE=E6=94=AF?= =?UTF-8?q?=E6=8C=81=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/main.py b/main.py index cc786eb..3cb1ff9 100644 --- a/main.py +++ b/main.py @@ -1,17 +1,22 @@ 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) + host = self.request.recv(1024) print("Host is {host}".format(host=host)) if host: - if host=='bwg'.encode('ascii'): - self.request.sendall("50%\n".encode('ascii')) - print('send to bwg') - if host=='hosteons'.encode('ascii'): - self.request.sendall("80%\n".encode('ascii')) + 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()