fix bug

Author: gaoyb7

fetchMetadata.py

@@ -106,7 +106,6 @@ def recv_piece(s, timeout=5):
 
 def fetch_metadata(nid, infohash, address, timeout=5):
     try:
-        print("in fetch_metadata")
         s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
         s.settimeout(timeout)
         s.connect(address)
@@ -117,15 +116,14 @@ def fetch_metadata(nid, infohash, address, timeout=5):
 
         # verification
         if not check_handshake_response(msg, infohash):
-            print("error in check handshake")
             return
 
         # advertize to support ut_metadata(BEP-09)
         ext_handshake(s)
         msg = s.recv(4096)
 
         ut_metadata, metadata_size = decode_ext_handshake_msg(msg)
-        print("ut_metadata", ut_metadata, "metadata_size", metadata_size)
+        #print("ut_metadata", ut_metadata, "metadata_size", metadata_size)
 
         metadata = []
         piece_tot = int(ceil(metadata_size / (1024 * 16)))
@@ -138,8 +136,9 @@ def fetch_metadata(nid, infohash, address, timeout=5):
         print(bencodepy.decode(metadata)[b"name"].decode(), "size", len(metadata))
 
     except socket.timeout:
-        print("timeout")
+        pass
     except Exception as e:
-        print(e)
+        #print(e)
+        pass
     finally:
         s.close()

main.py

@@ -21,7 +21,7 @@ def run(self):
     def fetch(self):
         for i in range(100):
             if self.que.qsize() == 0:
-                time.sleep(1)
+                sleep(1)
                 continue
             r = self.que.get()
             t = Thread(target=fetch_metadata, args=(r[0], r[1], r[2]))
@@ -31,7 +31,7 @@ def fetch(self):
     def log(self, nid, infohash, name, address):
         #print("%s %s" % (codecs.encode(infohash, "hex_codec").decode(), name.decode()))
         #fetch_metadata(nid, infohash, address)
-        Queue.put([nid, infohash, address])
+        self.que.put([nid, infohash, address])
 
 
 if __name__ == "__main__":