27 lines
929 B
Python
27 lines
929 B
Python
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_name in latest_speed_dic:
|
|
server=latest_speed_dic[server_name]
|
|
if host == server_name.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()
|