From 31365f19416e2a275f05395e99f9376dcfc8b33a Mon Sep 17 00:00:00 2001 From: Joshua Spence Date: Sun, 5 Nov 2023 10:58:15 +1100 Subject: [PATCH] Don't import `http.server` unless it is needed --- multicast-relay.py | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) diff --git a/multicast-relay.py b/multicast-relay.py index f6067e1..a90898c 100755 --- a/multicast-relay.py +++ b/multicast-relay.py @@ -4,7 +4,6 @@ import binascii import errno import json -import http.server import os import re import select @@ -877,13 +876,6 @@ def broadcastIpToMac(addr): def cidrToNetmask(bits): return socket.inet_ntoa(struct.pack('!I', (1 << 32) - (1 << (32 - bits)))) -class K8sCheck(http.server.BaseHTTPRequestHandler): - def do_GET(self): - self.send_response(200) - self.send_header('Content-type', 'text/html') - self.end_headers() - self.wfile.write(bytes('OK', 'utf-8')) - def main(): parser = argparse.ArgumentParser() parser.add_argument('--interfaces', nargs='+', required=True, @@ -1037,6 +1029,15 @@ def main(): packetRelay.addListener(addr, port, relay[1]) if args.k8sport: + import http.server + + class K8sCheck(http.server.BaseHTTPRequestHandler): + def do_GET(self): + self.send_response(200) + self.send_header('Content-type', 'text/html') + self.end_headers() + self.wfile.write(bytes('OK', 'utf-8')) + webServer = http.server.HTTPServer(('0.0.0.0', args.k8sport), K8sCheck) threading.Thread(target=webServer.serve_forever, daemon=True).start()