76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
import random
|
|
import urllib.request
|
|
from util import log, curr_directory
|
|
import settings
|
|
from PyQt5 import QtNetwork, QtCore
|
|
import json
|
|
|
|
|
|
class Node:
|
|
|
|
def __init__(self, node):
|
|
self._ip, self._port, self._tox_key = node['ipv4'], node['port'], node['public_key']
|
|
self._priority = random.randint(1, 1000000) if node['status_tcp'] and node['status_udp'] else 0
|
|
|
|
def get_priority(self):
|
|
return self._priority
|
|
|
|
priority = property(get_priority)
|
|
|
|
def get_data(self):
|
|
return bytes(self._ip, 'utf-8'), self._port, self._tox_key
|
|
|
|
|
|
def generate_nodes():
|
|
with open(curr_directory() + '/nodes.json', 'rt') as fl:
|
|
json_nodes = json.loads(fl.read())['nodes']
|
|
nodes = map(lambda json_node: Node(json_node), json_nodes)
|
|
sorted_nodes = sorted(nodes, key=lambda x: x.priority)[-4:]
|
|
for node in sorted_nodes:
|
|
yield node.get_data()
|
|
|
|
|
|
def save_nodes(nodes):
|
|
if not nodes:
|
|
return
|
|
print('Saving nodes...')
|
|
with open(curr_directory() + '/nodes.json', 'wb') as fl:
|
|
fl.write(nodes)
|
|
|
|
|
|
def download_nodes_list():
|
|
url = 'https://nodes.tox.chat/json'
|
|
s = settings.Settings.get_instance()
|
|
if not s['download_nodes_list']:
|
|
return
|
|
|
|
if not s['proxy_type']: # no proxy
|
|
try:
|
|
req = urllib.request.Request(url)
|
|
req.add_header('Content-Type', 'application/json')
|
|
response = urllib.request.urlopen(req)
|
|
result = response.read()
|
|
save_nodes(result)
|
|
except Exception as ex:
|
|
log('TOX nodes loading error: ' + str(ex))
|
|
else: # proxy
|
|
netman = QtNetwork.QNetworkAccessManager()
|
|
proxy = QtNetwork.QNetworkProxy()
|
|
proxy.setType(
|
|
QtNetwork.QNetworkProxy.Socks5Proxy if s['proxy_type'] == 2 else QtNetwork.QNetworkProxy.HttpProxy)
|
|
proxy.setHostName(s['proxy_host'])
|
|
proxy.setPort(s['proxy_port'])
|
|
netman.setProxy(proxy)
|
|
try:
|
|
request = QtNetwork.QNetworkRequest()
|
|
request.setUrl(QtCore.QUrl(url))
|
|
reply = netman.get(request)
|
|
|
|
while not reply.isFinished():
|
|
QtCore.QThread.msleep(1)
|
|
QtCore.QCoreApplication.processEvents()
|
|
data = bytes(reply.readAll().data())
|
|
save_nodes(data)
|
|
except Exception as ex:
|
|
log('TOX nodes loading error: ' + str(ex))
|