Switch to bencode_open

This commit is contained in:
Ivanq 2019-12-09 22:13:38 +03:00
parent 28fcf3c1ea
commit 3178b69172
4 changed files with 183 additions and 5 deletions

View file

@ -3,7 +3,7 @@ import urllib.request
import struct
import socket
import bencode
import bencode_open
from lib.subtl.subtl import UdpTrackerClient
import socks
import sockshandler
@ -133,9 +133,7 @@ class SiteAnnouncerPlugin(object):
# Decode peers
try:
peer_data = bencode.decode(response)["peers"]
if type(peer_data) is not bytes:
peer_data = peer_data.encode()
peer_data = bencode_open.loads(response)[b"peers"]
response = None
peer_count = int(len(peer_data) / 6)
peers = []