mirror of
https://github.com/SickGear/SickGear.git
synced 2024-11-25 14:25:05 +00:00
837 lines
32 KiB
Python
837 lines
32 KiB
Python
|
#
|
||
|
# This file is part of SickGear.
|
||
|
#
|
||
|
# SickGear 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.
|
||
|
#
|
||
|
# SickGear 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 SickGear. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
||
|
import os
|
||
|
import platform
|
||
|
import re
|
||
|
import shutil
|
||
|
import stat
|
||
|
import sys
|
||
|
import tarfile
|
||
|
import time
|
||
|
import traceback
|
||
|
from . import gh_api as github
|
||
|
|
||
|
# noinspection PyPep8Naming
|
||
|
import encodingKludge as ek
|
||
|
from exceptions_helper import ex
|
||
|
|
||
|
import sickgear
|
||
|
from . import logger, notifiers, ui
|
||
|
from .piper import check_pip_outdated
|
||
|
from sg_helpers import cmdline_runner, get_url
|
||
|
|
||
|
# noinspection PyUnresolvedReferences
|
||
|
from six.moves import urllib
|
||
|
from six import string_types
|
||
|
from _23 import list_keys
|
||
|
|
||
|
# noinspection PyUnreachableCode
|
||
|
if False:
|
||
|
from typing import Tuple
|
||
|
|
||
|
|
||
|
class PackagesUpdater(object):
|
||
|
|
||
|
def __init__(self):
|
||
|
self.install_type = 'Python package updates'
|
||
|
|
||
|
def run(self, force=False):
|
||
|
if not sickgear.EXT_UPDATES \
|
||
|
and self.check_for_new_version(force) \
|
||
|
and sickgear.UPDATE_PACKAGES_AUTO:
|
||
|
msg = 'Automatic %s enabled, restarting to update...' % self.install_type
|
||
|
logger.log(msg)
|
||
|
ui.notifications.message(msg)
|
||
|
time.sleep(3)
|
||
|
sickgear.restart(soft=False)
|
||
|
|
||
|
def check_for_new_version(self, force=False):
|
||
|
"""
|
||
|
Checks for available Python package installs/updates
|
||
|
:param force: ignore the UPDATE_PACKAGES_NOTIFY setting
|
||
|
:type force: bool
|
||
|
:returns: True when package install/updates are available
|
||
|
:rtype: bool
|
||
|
"""
|
||
|
if force and not sickgear.UPDATE_PACKAGES_MENU:
|
||
|
logger.log('Checking not enabled from menu action for %s' % self.install_type)
|
||
|
return False
|
||
|
|
||
|
if not sickgear.UPDATE_PACKAGES_NOTIFY and not sickgear.UPDATE_PACKAGES_AUTO and not force:
|
||
|
logger.log('Checking not enabled for %s' % self.install_type)
|
||
|
return False
|
||
|
|
||
|
logger.log('Checking for %s%s' % (self.install_type, ('', ' (from menu)')[force]))
|
||
|
sickgear.UPDATES_TODO = check_pip_outdated(force)
|
||
|
if not sickgear.UPDATES_TODO:
|
||
|
msg = 'No %s needed' % self.install_type
|
||
|
logger.log(msg)
|
||
|
|
||
|
if force:
|
||
|
ui.notifications.message(msg)
|
||
|
return False
|
||
|
|
||
|
logger.log('Update(s) for %s found %s' % (self.install_type, list_keys(sickgear.UPDATES_TODO)))
|
||
|
|
||
|
# save updates_todo to config to be loaded after restart
|
||
|
sickgear.save_config()
|
||
|
|
||
|
if not sickgear.UPDATE_PACKAGES_AUTO:
|
||
|
msg = '%s available — <a href="%s">Update Now</a>' % (
|
||
|
self.install_type, '%s/home/restart/?update_pkg=1&pid=%s' % (sickgear.WEB_ROOT, sickgear.PID))
|
||
|
if None is sickgear.NEWEST_VERSION_STRING:
|
||
|
sickgear.NEWEST_VERSION_STRING = ''
|
||
|
if msg not in sickgear.NEWEST_VERSION_STRING:
|
||
|
if sickgear.NEWEST_VERSION_STRING:
|
||
|
sickgear.NEWEST_VERSION_STRING += '<br>Also, '
|
||
|
sickgear.NEWEST_VERSION_STRING += msg
|
||
|
|
||
|
return True
|
||
|
|
||
|
|
||
|
class SoftwareUpdater(object):
|
||
|
"""
|
||
|
Version check class meant to run as a thread object with the sg scheduler.
|
||
|
"""
|
||
|
|
||
|
def __init__(self):
|
||
|
self._min_python = (100, 0) # set default to absurdly high to prevent update
|
||
|
self.install_type = self.find_install_type()
|
||
|
|
||
|
if 'git' == self.install_type:
|
||
|
self.updater = GitUpdateManager()
|
||
|
elif 'source' == self.install_type:
|
||
|
self.updater = SourceUpdateManager()
|
||
|
else:
|
||
|
self.updater = None
|
||
|
|
||
|
@property
|
||
|
def is_updatable(self):
|
||
|
# type: (...) -> bool
|
||
|
"""
|
||
|
test if the used Python version is greater or equal the required version and therefore, is update capable
|
||
|
"""
|
||
|
self.get_python_requirement()
|
||
|
return sys.version_info[:3] >= self.required_python
|
||
|
|
||
|
@property
|
||
|
def required_python(self):
|
||
|
# type: (...) -> Tuple[int, int]
|
||
|
"""
|
||
|
the required Python version that is update capable
|
||
|
|
||
|
value is cached after first calling get_python_requirement()
|
||
|
"""
|
||
|
return self._min_python
|
||
|
|
||
|
def get_python_requirement(self):
|
||
|
"""
|
||
|
get the required Python version from remote source
|
||
|
"""
|
||
|
branch = self.get_branch()
|
||
|
branch = ('dev', branch)[branch in ('main', 'dev')]
|
||
|
data = get_url(r'https://github.com/SickGear/SickGear/raw/%s/sickgear/py_requirement.data' % branch,
|
||
|
failure_monitor=False)
|
||
|
try:
|
||
|
self._min_python = (100, 0) # set default to absurdly high to prevent update
|
||
|
if isinstance(data, string_types):
|
||
|
self._min_python = tuple(int(_v) for _v in data.splitlines()[0].split('.'))
|
||
|
except (BaseException, Exception):
|
||
|
pass
|
||
|
|
||
|
def run(self, force=False):
|
||
|
# set current branch version
|
||
|
sickgear.BRANCH = self.get_branch()
|
||
|
|
||
|
if not sickgear.EXT_UPDATES \
|
||
|
and self.check_for_new_version(force) \
|
||
|
and sickgear.UPDATE_AUTO \
|
||
|
and sickgear.update_software_scheduler.action.update():
|
||
|
msg = 'Automatic software updates enabled, restarting with updated...'
|
||
|
logger.log(msg)
|
||
|
ui.notifications.message(msg)
|
||
|
time.sleep(3)
|
||
|
sickgear.restart(soft=False)
|
||
|
|
||
|
@staticmethod
|
||
|
def find_install_type():
|
||
|
"""
|
||
|
Determines how this copy of sg was installed.
|
||
|
|
||
|
returns: type of installation. Possible values are:
|
||
|
'git': running from source using git
|
||
|
'source': running from source without git
|
||
|
"""
|
||
|
return ('source', 'git')[os.path.isdir(ek.ek(os.path.join, sickgear.PROG_DIR, '.git'))]
|
||
|
|
||
|
def check_for_new_version(self, force=False):
|
||
|
"""
|
||
|
Checks for a new software release
|
||
|
:param force: ignore the UPDATE_NOTIFY setting
|
||
|
:type force: bool
|
||
|
:returns: True when a new software version is available
|
||
|
:rtype: bool
|
||
|
"""
|
||
|
|
||
|
if not sickgear.UPDATE_NOTIFY and not sickgear.UPDATE_AUTO and not force:
|
||
|
logger.log('Checking not enabled for software updates')
|
||
|
return False
|
||
|
|
||
|
if not self.is_updatable:
|
||
|
self._log_cannot_update()
|
||
|
return False
|
||
|
|
||
|
logger.log('Checking for "%s" software update%s' % (self.install_type, ('', ' (from menu)')[force]))
|
||
|
if not self.updater.need_update():
|
||
|
sickgear.NEWEST_VERSION_STRING = None
|
||
|
msg = 'No "%s" software update needed' % self.install_type
|
||
|
logger.log(msg)
|
||
|
|
||
|
if force:
|
||
|
ui.notifications.message(msg)
|
||
|
return False
|
||
|
|
||
|
if not sickgear.UPDATE_AUTO:
|
||
|
self.updater.set_newest_text()
|
||
|
|
||
|
return True
|
||
|
|
||
|
def _log_cannot_update(self):
|
||
|
if self._min_python >= (100, 0):
|
||
|
logger.warning('Cannot update SickGear because minimum required Python version is not available')
|
||
|
else:
|
||
|
logger.error('Currently used Python version %s does not meet the minimum required %s' %
|
||
|
('.'.join(['%s' % _v for _v in sys.version_info[:3]]),
|
||
|
'.'.join(['%s' % _v for _v in self.required_python])))
|
||
|
|
||
|
def update(self):
|
||
|
# update branch with current config branch value
|
||
|
self.updater.branch = sickgear.BRANCH
|
||
|
|
||
|
if not self.is_updatable:
|
||
|
self._log_cannot_update()
|
||
|
return False
|
||
|
|
||
|
# check for updates
|
||
|
if self.updater.need_update():
|
||
|
return self.updater.update()
|
||
|
|
||
|
def fetch(self, pull_request):
|
||
|
return self.updater.fetch(pull_request)
|
||
|
|
||
|
def list_remote_branches(self):
|
||
|
return self.updater.list_remote_branches()
|
||
|
|
||
|
def list_remote_pulls(self):
|
||
|
return self.updater.list_remote_pulls()
|
||
|
|
||
|
def get_branch(self):
|
||
|
return self.updater.branch
|
||
|
|
||
|
|
||
|
class UpdateManager(object):
|
||
|
@staticmethod
|
||
|
def get_github_repo_user():
|
||
|
return 'SickGear'
|
||
|
|
||
|
@staticmethod
|
||
|
def get_github_repo():
|
||
|
return 'SickGear'
|
||
|
|
||
|
@staticmethod
|
||
|
def get_update_url():
|
||
|
return '%s/home/update/?pid=%s' % (sickgear.WEB_ROOT, sickgear.PID)
|
||
|
|
||
|
|
||
|
class GitUpdateManager(UpdateManager):
|
||
|
def __init__(self):
|
||
|
self.unsafe = False
|
||
|
self._git_path = self._find_working_git()
|
||
|
self.github_repo_user = self.get_github_repo_user()
|
||
|
self.github_repo = self.get_github_repo()
|
||
|
|
||
|
self.branch = self._find_installed_branch()
|
||
|
if '' == self.branch:
|
||
|
self.branch = sickgear.BRANCH
|
||
|
if self.branch and self.branch != sickgear.BRANCH:
|
||
|
sickgear.BRANCH = self.branch
|
||
|
|
||
|
self._cur_commit_hash = None
|
||
|
self._newest_commit_hash = None
|
||
|
self._old_commit_hash = None
|
||
|
self._old_branch = None
|
||
|
self._num_commits_behind = 0
|
||
|
self._num_commits_ahead = 0
|
||
|
self._cur_pr_number = self.get_cur_pr_number()
|
||
|
|
||
|
def _find_working_git(self):
|
||
|
|
||
|
logger.debug(u'Checking if git commands are available')
|
||
|
|
||
|
main_git = (sickgear.GIT_PATH, 'git')[not sickgear.GIT_PATH]
|
||
|
|
||
|
_, _, exit_status = self._git_version(main_git)
|
||
|
|
||
|
if 0 == exit_status:
|
||
|
logger.debug(u'Using: %s' % main_git)
|
||
|
return main_git
|
||
|
|
||
|
logger.debug(u'Git not found: %s' % main_git)
|
||
|
|
||
|
# trying alternatives
|
||
|
|
||
|
alt_git_paths = []
|
||
|
|
||
|
# osx users who start sg from launchd have a broken path, so try a possible location
|
||
|
if 'darwin' == platform.system().lower():
|
||
|
alt_git_paths.append('/usr/local/git/bin/git')
|
||
|
|
||
|
if 'windows' == platform.system().lower():
|
||
|
if main_git != main_git.lower():
|
||
|
alt_git_paths.append(main_git.lower())
|
||
|
if sickgear.GIT_PATH:
|
||
|
logger.debug(u'git.exe is missing, remove `git_path` from config.ini: %s' % main_git)
|
||
|
if re.search(r' \(x86\)', main_git):
|
||
|
alt_git_paths.append(re.sub(r' \(x86\)', '', main_git))
|
||
|
else:
|
||
|
alt_git_paths.append(re.sub('Program Files', 'Program Files (x86)', main_git))
|
||
|
logger.debug(u'Until `git_path` is removed by a config.ini edit, trying: %s' % alt_git_paths[-1])
|
||
|
|
||
|
if alt_git_paths:
|
||
|
logger.debug('Trying known alternative git locations')
|
||
|
|
||
|
for cur_git_path in alt_git_paths:
|
||
|
_, _, exit_status = self._git_version(cur_git_path)
|
||
|
|
||
|
if 0 == exit_status:
|
||
|
logger.debug(u'Using: %s' % cur_git_path)
|
||
|
return cur_git_path
|
||
|
logger.debug(u'Not using: %s' % cur_git_path)
|
||
|
|
||
|
# Still haven't found a working git
|
||
|
error_message = 'Unable to find your git executable - Shutdown SickGear and EITHER set git_path' \
|
||
|
' in your config.ini OR delete your .git folder and run from source to enable updates.'
|
||
|
sickgear.NEWEST_VERSION_STRING = error_message
|
||
|
|
||
|
def _git_version(self, git_path):
|
||
|
|
||
|
return self._run_git(['version'], git_path)
|
||
|
|
||
|
def _run_git(self, arg_list, git_path=None, repeat=False):
|
||
|
|
||
|
output = err = None
|
||
|
exit_status = 1
|
||
|
|
||
|
if None is git_path:
|
||
|
git_path = self._git_path
|
||
|
|
||
|
if not git_path:
|
||
|
logger.error(u'No git specified, cannot use git commands')
|
||
|
return output, err, exit_status
|
||
|
|
||
|
cmd = ' '.join([git_path] + arg_list)
|
||
|
|
||
|
try:
|
||
|
logger.debug(u'Executing %s with your shell in %s' % (cmd, sickgear.PROG_DIR))
|
||
|
output, err, exit_status = cmdline_runner([git_path] + arg_list, env={'LANG': 'en_US.UTF-8'})
|
||
|
logger.debug(u'git output: %s' % output)
|
||
|
|
||
|
except OSError:
|
||
|
logger.log('Failed command: %s' % cmd)
|
||
|
|
||
|
except (BaseException, Exception) as e:
|
||
|
logger.log('Failed command: %s, %s' % (cmd, ex(e)))
|
||
|
|
||
|
if 0 == exit_status:
|
||
|
logger.debug(u'Successful return: %s' % cmd)
|
||
|
exit_status = 0
|
||
|
self.unsafe = False
|
||
|
|
||
|
elif 1 == exit_status:
|
||
|
logger.error(u'Failed: %s returned: %s' % (cmd, output))
|
||
|
|
||
|
elif 128 == exit_status or 'fatal:' in output or err:
|
||
|
if 'unsafe repository' not in output and 'fatal:' in output:
|
||
|
try:
|
||
|
outp, err, exit_status = cmdline_runner([git_path] + ['rev-parse', 'HEAD'],
|
||
|
env={'LANG': 'en_US.UTF-8'})
|
||
|
if 'unsafe repository' in outp:
|
||
|
self.unsafe = True
|
||
|
except (BaseException, Exception):
|
||
|
pass
|
||
|
if 'unsafe repository' in output:
|
||
|
self.unsafe = True
|
||
|
if self.unsafe and not repeat:
|
||
|
try:
|
||
|
outp, err, exit_status = cmdline_runner(
|
||
|
[git_path] + ['config', '--global', '--add', 'safe.directory',
|
||
|
sickgear.PROG_DIR.replace('\\', '/')], env={'LANG': 'en_US.UTF-8'})
|
||
|
if 0 == exit_status:
|
||
|
return self._run_git(arg_list, git_path, repeat=True)
|
||
|
except (BaseException, Exception):
|
||
|
pass
|
||
|
exit_status = 128
|
||
|
msg = u'Fatal: %s returned: %s' % (cmd, output)
|
||
|
if 'develop' in output.lower() or 'main' in output.lower():
|
||
|
logger.error(msg)
|
||
|
else:
|
||
|
logger.debug(msg)
|
||
|
|
||
|
else:
|
||
|
logger.error(u'Treat as error for now, command: %s returned: %s' % (cmd, output))
|
||
|
|
||
|
return output, err, exit_status
|
||
|
|
||
|
def _find_installed_version(self):
|
||
|
"""
|
||
|
Attempts to find the currently installed version of SickGear.
|
||
|
Uses git show to get commit version.
|
||
|
Returns: True for success or False for failure
|
||
|
"""
|
||
|
|
||
|
output, _, exit_status = self._run_git(['rev-parse', 'HEAD'])
|
||
|
|
||
|
if 0 == exit_status and output:
|
||
|
cur_commit_hash = output.strip()
|
||
|
if not re.match(r'^[a-z0-9]+$', cur_commit_hash):
|
||
|
logger.error(u'Output doesn\'t look like a hash, not using it')
|
||
|
return False
|
||
|
self._cur_commit_hash = cur_commit_hash
|
||
|
sickgear.CUR_COMMIT_HASH = str(cur_commit_hash)
|
||
|
return True
|
||
|
return False
|
||
|
|
||
|
def _find_installed_branch(self):
|
||
|
output, _, exit_status = self._run_git(['symbolic-ref', '-q', 'HEAD'])
|
||
|
if 0 == exit_status and output:
|
||
|
branch = output.strip().replace('refs/heads/', '', 1)
|
||
|
if branch:
|
||
|
return branch
|
||
|
|
||
|
return ''
|
||
|
|
||
|
def _check_github_for_update(self):
|
||
|
"""
|
||
|
Uses git commands to check if there is a newer version that the provided
|
||
|
commit hash. If there is a newer version it sets _num_commits_behind.
|
||
|
"""
|
||
|
|
||
|
self._num_commits_behind = 0
|
||
|
self._num_commits_ahead = 0
|
||
|
|
||
|
# get all new info from GitHub
|
||
|
_, _, exit_status = self._run_git(['fetch', '%s' % sickgear.GIT_REMOTE])
|
||
|
|
||
|
if 0 != exit_status:
|
||
|
logger.error(u'Unable to contact github, can\'t check for update')
|
||
|
return
|
||
|
|
||
|
if not self._cur_pr_number:
|
||
|
|
||
|
# get the latest commit_hash from remote
|
||
|
output, _, exit_status = self._run_git(['rev-parse', '--verify', '--quiet', '@{upstream}'])
|
||
|
|
||
|
if 0 == exit_status and output:
|
||
|
cur_commit_hash = output.strip()
|
||
|
|
||
|
if not re.match('^[a-z0-9]+$', cur_commit_hash):
|
||
|
logger.debug(u'Output doesn\'t look like a hash, not using it')
|
||
|
return
|
||
|
|
||
|
self._newest_commit_hash = cur_commit_hash
|
||
|
self._old_commit_hash = cur_commit_hash
|
||
|
self._old_branch = self._find_installed_branch()
|
||
|
else:
|
||
|
logger.debug(u'git didn\'t return newest commit hash')
|
||
|
return
|
||
|
|
||
|
# get number of commits behind and ahead (option --count not supported git < 1.7.2)
|
||
|
output, _, exit_status = self._run_git(['rev-list', '--left-right', '@{upstream}...HEAD'])
|
||
|
|
||
|
if 0 == exit_status and output:
|
||
|
|
||
|
try:
|
||
|
self._num_commits_behind = int(output.count('<'))
|
||
|
self._num_commits_ahead = int(output.count('>'))
|
||
|
|
||
|
except (BaseException, Exception):
|
||
|
logger.debug(u'git didn\'t return numbers for behind and ahead, not using it')
|
||
|
return
|
||
|
|
||
|
logger.debug(u'cur_commit = %s, newest_commit = %s, num_commits_behind = %s, num_commits_ahead = %s' % (
|
||
|
self._cur_commit_hash, self._newest_commit_hash, self._num_commits_behind, self._num_commits_ahead))
|
||
|
else:
|
||
|
# we need to treat pull requests specially as it doesn't seem possible to set their "@{upstream}" tag
|
||
|
output, _, _ = self._run_git(['ls-remote', '%s' % sickgear.GIT_REMOTE,
|
||
|
'refs/pull/%s/head' % self._cur_pr_number])
|
||
|
self._newest_commit_hash = re.findall('(.*)\t', output)[0]
|
||
|
self._old_commit_hash = None
|
||
|
self._old_branch = None
|
||
|
|
||
|
def set_newest_text(self):
|
||
|
# if we're up-to-date then don't set this
|
||
|
newest_text = None
|
||
|
url = 'https://github.com/%s/%s' % (self.github_repo_user, self.github_repo)
|
||
|
|
||
|
if self._num_commits_ahead:
|
||
|
newest_text = 'Local branch is ahead of %s. Automatic update not possible.' % self.branch
|
||
|
logger.error(newest_text)
|
||
|
|
||
|
elif 0 < self._num_commits_behind:
|
||
|
|
||
|
if self._newest_commit_hash:
|
||
|
url += '/compare/%s...%s' % (self._cur_commit_hash, self._newest_commit_hash)
|
||
|
else:
|
||
|
url += '/commits/'
|
||
|
|
||
|
newest_text = 'There is a <a href="%s" onclick="window.open(this.href); return !1;">newer' \
|
||
|
' version available</a> (you\'re %s commit%s behind) — <a href="%s">Update Now</a>' % \
|
||
|
(url, self._num_commits_behind,
|
||
|
('', 's')[1 < self._num_commits_behind], self.get_update_url())
|
||
|
|
||
|
elif self._cur_pr_number and (self._cur_commit_hash != self._newest_commit_hash):
|
||
|
url += '/commit/%s' % self._newest_commit_hash
|
||
|
|
||
|
newest_text = 'There is a <a href="%s" onclick="window.open(this.href); return !1;">newer' \
|
||
|
' version available</a> — <a href="%s">Update Now</a>' % (url, self.get_update_url())
|
||
|
|
||
|
sickgear.NEWEST_VERSION_STRING = newest_text
|
||
|
|
||
|
def need_update(self):
|
||
|
|
||
|
installed_branch = self._find_installed_branch()
|
||
|
if self.branch != installed_branch:
|
||
|
logger.debug(u'Branch checkout: %s->%s' % (installed_branch, self.branch))
|
||
|
return True
|
||
|
|
||
|
self._find_installed_version()
|
||
|
if not self._cur_commit_hash:
|
||
|
return True
|
||
|
|
||
|
self._old_branch = None
|
||
|
self._old_commit_hash = None
|
||
|
try:
|
||
|
self._check_github_for_update()
|
||
|
except (BaseException, Exception) as e:
|
||
|
logger.error(u'Unable to contact github, can\'t check for update: %r' % e)
|
||
|
return False
|
||
|
|
||
|
if 0 < self._num_commits_behind:
|
||
|
return True
|
||
|
|
||
|
if self._cur_pr_number and self._cur_commit_hash != self._newest_commit_hash:
|
||
|
return True
|
||
|
|
||
|
return False
|
||
|
|
||
|
@staticmethod
|
||
|
def _is_python_supported():
|
||
|
"""
|
||
|
check for required python version on current install
|
||
|
"""
|
||
|
try:
|
||
|
with open(os.path.join(sickgear.PROG_DIR, 'sickgear', 'py_requirement.data'), 'rt') as f:
|
||
|
data = f.read()
|
||
|
except (BaseException, Exception):
|
||
|
return False
|
||
|
if isinstance(data, string_types):
|
||
|
try:
|
||
|
min_python = tuple(int(_v) for _v in data.splitlines()[0].split('.'))
|
||
|
except (BaseException, Exception):
|
||
|
min_python = (100, 0) # set default to absurdly high to prevent update
|
||
|
return min_python <= sys.version_info[:3]
|
||
|
return False
|
||
|
|
||
|
def revert_update(self):
|
||
|
if self._old_commit_hash:
|
||
|
if self._old_branch:
|
||
|
cur_branch = self._find_installed_branch()
|
||
|
if cur_branch and self._old_branch != cur_branch:
|
||
|
# in case of a branch switch first switch back to old branch
|
||
|
_, _, exit_status = self._run_git(['checkout', '-f', '-B', '%s' % self._old_branch,
|
||
|
'%s/%s' % (sickgear.GIT_REMOTE, self._old_branch)])
|
||
|
if 0 != exit_status:
|
||
|
return exit_status
|
||
|
_, _, exit_status = self._run_git(['reset', '--hard', '"%s"' % self._old_commit_hash])
|
||
|
if 0 == exit_status:
|
||
|
self._old_commit_hash = None
|
||
|
self._old_branch = None
|
||
|
return exit_status
|
||
|
return 1
|
||
|
|
||
|
def update(self):
|
||
|
"""
|
||
|
Calls git pull origin <branch> in order to update SickGear. Returns a bool depending
|
||
|
on the call's success.
|
||
|
"""
|
||
|
|
||
|
if self.branch == self._find_installed_branch():
|
||
|
if not self._cur_pr_number:
|
||
|
_, _, exit_status = self._run_git(['pull', '-f', '%s' % sickgear.GIT_REMOTE, '%s' % self.branch])
|
||
|
else:
|
||
|
_, _, exit_status = self._run_git(['pull', '-f', '%s' % sickgear.GIT_REMOTE,
|
||
|
'pull/%s/head:%s' % (self._cur_pr_number, self.branch)])
|
||
|
|
||
|
else:
|
||
|
self._run_git(['fetch', '%s' % sickgear.GIT_REMOTE])
|
||
|
_, _, exit_status = self._run_git(['checkout', '-f', '-B', '%s' % self.branch,
|
||
|
'%s/%s' % (sickgear.GIT_REMOTE, self.branch)])
|
||
|
|
||
|
try:
|
||
|
if 0 == exit_status:
|
||
|
# recheck that the checked out version supports this python version and revert if not
|
||
|
if self._old_commit_hash and not self._is_python_supported():
|
||
|
if 0 != self.revert_update():
|
||
|
logger.error('Failed to revert update')
|
||
|
self._find_installed_version()
|
||
|
return False
|
||
|
|
||
|
self._find_installed_version()
|
||
|
|
||
|
# Notify update successful
|
||
|
notifiers.notify_git_update(sickgear.CUR_COMMIT_HASH if sickgear.CUR_COMMIT_HASH else '')
|
||
|
return True
|
||
|
|
||
|
return False
|
||
|
finally:
|
||
|
self._old_branch = None
|
||
|
self._old_commit_hash = None
|
||
|
|
||
|
def list_remote_branches(self):
|
||
|
output, _, exit_status = self._run_git(['ls-remote', '--heads', '%s' % sickgear.GIT_REMOTE])
|
||
|
if 0 == exit_status and output:
|
||
|
return re.findall(r'\S+\Wrefs/heads/(.*)', output)
|
||
|
return []
|
||
|
|
||
|
def list_remote_pulls(self):
|
||
|
gh = github.GitHub(self.github_repo_user, self.github_repo, self.branch)
|
||
|
return gh.pull_requests()
|
||
|
|
||
|
def fetch(self, pull_request):
|
||
|
_, _, exit_status = self._run_git(['fetch', '-f', '%s' % sickgear.GIT_REMOTE, '%s' % pull_request])
|
||
|
return 0 == exit_status
|
||
|
|
||
|
def get_cur_pr_number(self):
|
||
|
try:
|
||
|
pull_number = int(self.branch.split('/')[1])
|
||
|
except (BaseException, Exception):
|
||
|
pull_number = None
|
||
|
|
||
|
return pull_number
|
||
|
|
||
|
|
||
|
class SourceUpdateManager(UpdateManager):
|
||
|
def __init__(self):
|
||
|
self.github_repo_user = self.get_github_repo_user()
|
||
|
self.github_repo = self.get_github_repo()
|
||
|
|
||
|
self.branch = sickgear.BRANCH
|
||
|
if '' == sickgear.BRANCH:
|
||
|
self.branch = self._find_installed_branch()
|
||
|
|
||
|
self._cur_commit_hash = sickgear.CUR_COMMIT_HASH
|
||
|
self._newest_commit_hash = None
|
||
|
self._num_commits_behind = 0
|
||
|
|
||
|
@staticmethod
|
||
|
def _find_installed_branch():
|
||
|
if '' == sickgear.CUR_COMMIT_BRANCH:
|
||
|
return 'main'
|
||
|
return sickgear.CUR_COMMIT_BRANCH
|
||
|
|
||
|
def need_update(self):
|
||
|
# need this to run first to set self._newest_commit_hash
|
||
|
try:
|
||
|
self._check_github_for_update()
|
||
|
except (BaseException, Exception) as e:
|
||
|
logger.error(u'Unable to contact github, can\'t check for update: %r' % e)
|
||
|
return False
|
||
|
|
||
|
installed_branch = self._find_installed_branch()
|
||
|
if self.branch != installed_branch:
|
||
|
logger.debug(u'Branch checkout: %s->%s' % (installed_branch, self.branch))
|
||
|
return True
|
||
|
|
||
|
if not self._cur_commit_hash or 0 < self._num_commits_behind:
|
||
|
return True
|
||
|
|
||
|
return False
|
||
|
|
||
|
def _check_github_for_update(self):
|
||
|
"""
|
||
|
Uses pygithub to ask GitHub if there is a newer version that the provided
|
||
|
commit hash. If there is a newer version it sets SickGear's version text.
|
||
|
|
||
|
commit_hash: hash that we're checking against
|
||
|
"""
|
||
|
|
||
|
self._num_commits_behind = 0
|
||
|
self._newest_commit_hash = None
|
||
|
|
||
|
gh = github.GitHub(self.github_repo_user, self.github_repo, self.branch)
|
||
|
|
||
|
# try to get the newest commit hash and commits behind directly by comparing branch and current commit
|
||
|
if self._cur_commit_hash:
|
||
|
branch_compared = gh.compare(base=self.branch, head=self._cur_commit_hash)
|
||
|
|
||
|
if 'base_commit' in branch_compared:
|
||
|
self._newest_commit_hash = branch_compared['base_commit']['sha']
|
||
|
|
||
|
if 'behind_by' in branch_compared:
|
||
|
self._num_commits_behind = int(branch_compared['behind_by'])
|
||
|
|
||
|
# fall back and iterate over last 100 (items per page in gh_api) commits
|
||
|
if not self._newest_commit_hash:
|
||
|
|
||
|
for curCommit in gh.commits():
|
||
|
if not self._newest_commit_hash:
|
||
|
self._newest_commit_hash = curCommit['sha']
|
||
|
if not self._cur_commit_hash:
|
||
|
break
|
||
|
|
||
|
if curCommit['sha'] == self._cur_commit_hash:
|
||
|
break
|
||
|
|
||
|
# when _cur_commit_hash doesn't match anything _num_commits_behind == 100
|
||
|
self._num_commits_behind += 1
|
||
|
|
||
|
logger.debug(u'cur_commit = %s, newest_commit = %s, num_commits_behind = %s'
|
||
|
% (self._cur_commit_hash, self._newest_commit_hash, self._num_commits_behind))
|
||
|
|
||
|
def set_newest_text(self):
|
||
|
|
||
|
# if we're up-to-date then don't set this
|
||
|
newest_text = None
|
||
|
|
||
|
if not self._cur_commit_hash:
|
||
|
logger.debug(u'Unknown current version number, don\'t know if we should update or not')
|
||
|
|
||
|
newest_text = 'Unknown current version number: If you\'ve never used the SickGear upgrade system' \
|
||
|
' before then current version is not set. — <a href="%s">Update Now</a>' \
|
||
|
% self.get_update_url()
|
||
|
|
||
|
elif 0 < self._num_commits_behind:
|
||
|
url = 'https://github.com/%s/%s' % (self.github_repo_user, self.github_repo)
|
||
|
if self._newest_commit_hash:
|
||
|
url += '/compare/' + self._cur_commit_hash + '...' + self._newest_commit_hash
|
||
|
else:
|
||
|
url += '/commits/'
|
||
|
|
||
|
newest_text = 'There is a <a href="%s" onclick="window.open(this.href); return false;">newer' \
|
||
|
' version available</a> (you\'re %s commit%s behind) — <a href="%s">Update Now</a>' \
|
||
|
% (url, self._num_commits_behind,
|
||
|
('', 's')[1 < self._num_commits_behind], self.get_update_url())
|
||
|
|
||
|
sickgear.NEWEST_VERSION_STRING = newest_text
|
||
|
|
||
|
def update(self):
|
||
|
"""
|
||
|
Downloads the latest source tarball from GitHub and installs it over the existing version.
|
||
|
"""
|
||
|
|
||
|
tar_download_url = 'https://github.com/%s/%s/tarball/%s' \
|
||
|
% (self.github_repo_user, self.github_repo, self.branch)
|
||
|
|
||
|
try:
|
||
|
# prepare the update dir
|
||
|
sg_update_dir = ek.ek(os.path.join, sickgear.PROG_DIR, u'sg-update')
|
||
|
|
||
|
if os.path.isdir(sg_update_dir):
|
||
|
logger.log(u'Clearing out update folder %s before extracting' % sg_update_dir)
|
||
|
shutil.rmtree(sg_update_dir)
|
||
|
|
||
|
logger.log(u'Creating update folder %s before extracting' % sg_update_dir)
|
||
|
os.makedirs(sg_update_dir)
|
||
|
|
||
|
# retrieve file
|
||
|
logger.log(u'Downloading update from %r' % tar_download_url)
|
||
|
tar_download_path = os.path.join(sg_update_dir, u'sg-update.tar')
|
||
|
urllib.request.urlretrieve(tar_download_url, tar_download_path)
|
||
|
|
||
|
if not ek.ek(os.path.isfile, tar_download_path):
|
||
|
logger.error(u'Unable to retrieve new version from %s, can\'t update' % tar_download_url)
|
||
|
return False
|
||
|
|
||
|
if not ek.ek(tarfile.is_tarfile, tar_download_path):
|
||
|
logger.error(u'Retrieved version from %s is corrupt, can\'t update' % tar_download_url)
|
||
|
return False
|
||
|
|
||
|
# extract to sg-update dir
|
||
|
logger.log(u'Extracting file %s' % tar_download_path)
|
||
|
tar = tarfile.open(tar_download_path)
|
||
|
tar.extractall(sg_update_dir)
|
||
|
tar.close()
|
||
|
|
||
|
# delete .tar.gz
|
||
|
logger.log(u'Deleting file %s' % tar_download_path)
|
||
|
os.remove(tar_download_path)
|
||
|
|
||
|
# find update dir name
|
||
|
update_dir_contents = [x for x in os.listdir(sg_update_dir) if
|
||
|
os.path.isdir(os.path.join(sg_update_dir, x))]
|
||
|
if 1 != len(update_dir_contents):
|
||
|
logger.error(u'Invalid update data, update failed: %s' % update_dir_contents)
|
||
|
return False
|
||
|
content_dir = os.path.join(sg_update_dir, update_dir_contents[0])
|
||
|
|
||
|
# walk temp folder and move files to main folder
|
||
|
logger.log(u'Moving files from %s to %s' % (content_dir, sickgear.PROG_DIR))
|
||
|
for dirname, dirnames, filenames in os.walk(content_dir):
|
||
|
dirname = dirname[len(content_dir) + 1:]
|
||
|
for curfile in filenames:
|
||
|
old_path = os.path.join(content_dir, dirname, curfile)
|
||
|
new_path = os.path.join(sickgear.PROG_DIR, dirname, curfile)
|
||
|
|
||
|
# Avoid DLL access problem on WIN32/64
|
||
|
# These files needing to be updated manually
|
||
|
# or find a way to kill the access from memory
|
||
|
if curfile in ('unrar.dll', 'unrar64.dll'):
|
||
|
try:
|
||
|
os.chmod(new_path, stat.S_IWRITE)
|
||
|
os.remove(new_path)
|
||
|
os.renames(old_path, new_path)
|
||
|
except (BaseException, Exception) as e:
|
||
|
logger.debug(u'Unable to update %s: %s' % (new_path, ex(e)))
|
||
|
os.remove(old_path) # Trash the updated file without moving in new path
|
||
|
continue
|
||
|
|
||
|
if os.path.isfile(new_path):
|
||
|
os.remove(new_path)
|
||
|
os.renames(old_path, new_path)
|
||
|
|
||
|
sickgear.CUR_COMMIT_HASH = self._newest_commit_hash
|
||
|
sickgear.CUR_COMMIT_BRANCH = self.branch
|
||
|
|
||
|
except (BaseException, Exception) as e:
|
||
|
logger.error(u'Error while trying to update: %s' % ex(e))
|
||
|
logger.debug(u'Traceback: %s' % traceback.format_exc())
|
||
|
return False
|
||
|
|
||
|
# Notify update successful
|
||
|
notifiers.notify_git_update(sickgear.NEWEST_VERSION_STRING)
|
||
|
|
||
|
return True
|
||
|
|
||
|
def list_remote_branches(self):
|
||
|
gh = github.GitHub(self.github_repo_user, self.github_repo, self.branch)
|
||
|
return [x['name'] for x in gh.branches() if x and 'name' in x]
|
||
|
|
||
|
@staticmethod
|
||
|
def list_remote_pulls():
|
||
|
# we don't care about testers that don't use git
|
||
|
return []
|