dns_hub: Fix flake8 warnings
authorTim Beale <timbeale@catalyst.net.nz>
Wed, 30 Jan 2019 00:24:45 +0000 (13:24 +1300)
committerAndrew Bartlett <abartlet@samba.org>
Fri, 1 Feb 2019 02:36:15 +0000 (03:36 +0100)
Signed-off-by: Tim Beale <timbeale@catalyst.net.nz>
Reviewed-by: Volker Lendecke <vl@samba.org>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
python/samba/tests/dns_forwarder_helpers/dns_hub.py

index 2ac675361e0a435bc7cba2969783fa3f48d46375..97b7432294c5129117253a586a9f07f0ab020d9e 100755 (executable)
 
 import threading
 import sys
-import os
 import select
 import socket
 from samba.dcerpc import dns
-from samba.tests.dns_base import DNSTest
 import samba.ndr as ndr
 
 if sys.version_info[0] < 3:
@@ -34,6 +32,7 @@ else:
     import socketserver
     sserver = socketserver
 
+
 class DnsHandler(sserver.BaseRequestHandler):
     def dns_transaction_udp(self, packet, host):
         "send a DNS query and read the reply"
@@ -90,7 +89,7 @@ class DnsHandler(sserver.BaseRequestHandler):
 
     def handle(self):
         data, sock = self.request
-        query = ndr.ndr_unpack(dns.name_packet, data);
+        query = ndr.ndr_unpack(dns.name_packet, data)
         name = query.questions[0].name
         forwarder = self.forwarder(name)
         response = None
@@ -134,9 +133,10 @@ class server_thread(threading.Thread):
         self.server.serve_forever()
         print("dns_hub: after serve_forever()")
 
+
 def main():
-    timeout = int(sys.argv[1])*1000
-    timeout = min(timeout, 2**31-1) # poll with 32-bit int can't take more
+    timeout = int(sys.argv[1]) * 1000
+    timeout = min(timeout, 2**31 - 1)  # poll with 32-bit int can't take more
     host = sys.argv[2]
     server = sserver.UDPServer((host, int(53)), DnsHandler)
     t = server_thread(server)