update.py

# =============================================================================
# Copyright (C) 2014 Ryan Holmes
#
# This file is part of pyfa.
#
# pyfa is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# pyfa is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with pyfa.  If not, see <http://www.gnu.org/licenses/>.
# =============================================================================

import calendar
import threading

# noinspection PyPackageRequirements
import dateutil.parser
# noinspection PyPackageRequirements
import wx
from logbook import Logger
from packaging.version import Version

import config
from service.network import Network
from service.settings import UpdateSettings

pyfalog = Logger(__name__)


class CheckUpdateThread(threading.Thread):
    def __init__(self, callback):
        threading.Thread.__init__(self)
        self.name = "CheckUpdate"
        self.callback = callback
        self.settings = UpdateSettings.getInstance()
        self.network = Network.getInstance()
        self.running = True

    def run(self):
        network = Network.getInstance()

        try:
            try:
                response = network.get(
                    url='https://www.pyfa.io/update_check?pyfa_version={}&client_hash={}'.format(config.version, config.getClientSecret()),
                    type=network.UPDATE, timeout=5)
            except (KeyboardInterrupt, SystemExit):
                raise
            except Exception as e:
                response = network.get(
                    url='https://api.github.com/repos/pyfa-org/Pyfa/releases',
                    type=network.UPDATE, timeout=5)

            jsonResponse = response.json()
            jsonResponse.sort(
                key=lambda x: calendar.timegm(dateutil.parser.parse(x['published_at']).utctimetuple()),
                reverse=True
            )

            for release in jsonResponse[:5]:
                rVersion = Version(release['tag_name'])
                cVersion = Version(config.version)

                # Suppress pre releases if we're not already on a pre-release (if we are, we want to know about new ones)
                if not cVersion.is_prerelease and rVersion.is_prerelease and self.settings.get('prerelease'):
                    continue

                # Handle use-case of updating to suppressed version
                if self.settings.get('version') == 'v' + config.version:
                    self.settings.set('version', None)

                # Suppress version
                if release['tag_name'] == self.settings.get('version'):
                    break

                if rVersion > cVersion:
                    wx.CallAfter(self.callback, release, rVersion)
                    break

        except (KeyboardInterrupt, SystemExit):
            raise
        except Exception as e:
            pyfalog.error("Caught exception in run")
            pyfalog.error(e)
            pass

    @staticmethod
    def versiontuple(v):
        return tuple(map(int, (v.split("."))))

    def stop(self):
        self.running = False


class Update:
    instance = None

    @staticmethod
    def CheckUpdate(callback):
        thread = CheckUpdateThread(callback)
        pyfalog.debug("Starting Check Update Thread.")
        thread.start()

    @classmethod
    def getInstance(cls):
        if cls.instance is None:
            cls.instance = Update()
        return cls.instance