130 lines
4.2 KiB
Python
130 lines
4.2 KiB
Python
# -*- mode: python; indent-tabs-mode: nil; py-indent-offset: 4; coding: utf-8 -*-
|
|
import utils.util as util
|
|
import utils.ui as util_ui
|
|
import os
|
|
import platform
|
|
import urllib
|
|
from PyQt5 import QtNetwork, QtCore
|
|
import subprocess
|
|
|
|
global LOG
|
|
import logging
|
|
LOG = logging.getLogger('app.'+__name__)
|
|
log = lambda x: LOG.info(x)
|
|
|
|
TIMEOUT=10
|
|
|
|
def connection_available():
|
|
try:
|
|
urllib.request.urlopen('http://216.58.192.142', timeout=TIMEOUT) # google.com
|
|
return True
|
|
except:
|
|
return False
|
|
|
|
def updater_available():
|
|
if is_from_sources():
|
|
return os.path.exists(util.curr_directory() + '/toxygen_updater.py')
|
|
elif platform.system() == 'Windows':
|
|
return os.path.exists(util.curr_directory() + '/toxygen_updater.exe')
|
|
else:
|
|
return os.path.exists(util.curr_directory() + '/toxygen_updater')
|
|
|
|
|
|
def check_for_updates(current_version, settings):
|
|
major, minor, patch = list(map(lambda x: int(x), current_version.split('.')))
|
|
versions = generate_versions(major, minor, patch)
|
|
for version in versions:
|
|
if send_request(version, settings):
|
|
return version
|
|
return None # no new version was found
|
|
|
|
|
|
def is_from_sources():
|
|
return __file__.endswith('.py')
|
|
|
|
|
|
def test_url(version):
|
|
return 'https://github.com/toxygen-project/toxygen/releases/tag/v' + version
|
|
|
|
|
|
def get_url(version):
|
|
if is_from_sources():
|
|
return 'https://github.com/toxygen-project/toxygen/archive/v' + version + '.zip'
|
|
else:
|
|
if platform.system() == 'Windows':
|
|
name = 'toxygen_windows.zip'
|
|
elif util.is_64_bit():
|
|
name = 'toxygen_linux_64.tar.gz'
|
|
else:
|
|
name = 'toxygen_linux.tar.gz'
|
|
return 'https://github.com/toxygen-project/toxygen/releases/download/v{}/{}'.format(version, name)
|
|
|
|
|
|
def get_params(url, version):
|
|
if is_from_sources():
|
|
if platform.system() == 'Windows':
|
|
return ['python', 'toxygen_updater.py', url, version]
|
|
else:
|
|
return ['python3', 'toxygen_updater.py', url, version]
|
|
elif platform.system() == 'Windows':
|
|
return [util.curr_directory() + '/toxygen_updater.exe', url, version]
|
|
else:
|
|
return ['./toxygen_updater', url, version]
|
|
|
|
|
|
def download(version):
|
|
os.chdir(util.curr_directory())
|
|
url = get_url(version)
|
|
params = get_params(url, version)
|
|
LOG.info('Updating Toxygen')
|
|
try:
|
|
subprocess.Popen(params)
|
|
except Exception as ex:
|
|
LOG.error('running updater failed with ' + str(ex))
|
|
|
|
|
|
def send_request(version, settings):
|
|
netman = QtNetwork.QNetworkAccessManager()
|
|
proxy = QtNetwork.QNetworkProxy()
|
|
if settings['proxy_type']:
|
|
proxy.setType(QtNetwork.QNetworkProxy.Socks5Proxy if settings['proxy_type'] == 2 else QtNetwork.QNetworkProxy.HttpProxy)
|
|
proxy.setHostName(settings['proxy_host'])
|
|
proxy.setPort(settings['proxy_port'])
|
|
netman.setProxy(proxy)
|
|
url = test_url(version)
|
|
try:
|
|
request = QtNetwork.QNetworkRequest()
|
|
request.setUrl(QtCore.QUrl(url))
|
|
reply = netman.get(request)
|
|
while not reply.isFinished():
|
|
QtCore.QThread.msleep(1)
|
|
QtCore.QCoreApplication.processEvents()
|
|
attr = reply.attribute(QtNetwork.QNetworkRequest.HttpStatusCodeAttribute)
|
|
return attr is not None and 200 <= attr < 300
|
|
except Exception as ex:
|
|
LOG.error('TOXYGEN UPDATER ' + str(ex))
|
|
return False
|
|
|
|
|
|
def generate_versions(major, minor, patch):
|
|
new_major = '.'.join([str(major + 1), '0', '0'])
|
|
new_minor = '.'.join([str(major), str(minor + 1), '0'])
|
|
new_patch = '.'.join([str(major), str(minor), str(patch + 1)])
|
|
return new_major, new_minor, new_patch
|
|
|
|
|
|
def start_update_if_needed(version, settings):
|
|
updating = False
|
|
if settings['update'] and updater_available() and connection_available(): # auto update
|
|
version = check_for_updates(version, settings)
|
|
if version is not None:
|
|
if settings['update'] == 2:
|
|
download(version)
|
|
updating = True
|
|
else:
|
|
reply = util_ui.question(util_ui.tr('Update for Toxygen was found. Download and install it?'))
|
|
if reply:
|
|
download(version)
|
|
updating = True
|
|
return updating
|