SickGear/sickgear/version_checker.py
Prinz23 7f51b9ce4d Change prevent git update issue.
Add '--no-rebase' to git pull (update) command to prevent issues with git 2.37+
2024-02-13 11:07:07 +00:00

858 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
from exceptions_helper import ex
import sickgear
from . import logger, notifiers, ui
from .scheduler import (Scheduler, Job)
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
# noinspection PyUnreachableCode
if False:
from typing import Tuple
class PackagesUpdater(Job):
def __init__(self):
super(PackagesUpdater, self).__init__(self.job_run, kwargs={})
self.install_type = 'Python package updates'
def job_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
"""
response = Scheduler.blocking_jobs()
if response:
logger.log(f'Update skipped because {response}', logger.DEBUG)
return False
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(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 &mdash; <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(Job):
"""
Version check class meant to run as a thread object with the sg scheduler.
"""
def __init__(self):
super(SoftwareUpdater, self).__init__(self.job_run, kwargs={})
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 job_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(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
response = Scheduler.blocking_jobs()
if response:
logger.log(f'Update skipped because {response}', logger.DEBUG)
return False
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('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(f'Using: {main_git}')
return main_git
logger.debug(f'Git not found: {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(f'git.exe is missing, remove `git_path` from config.ini: {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(f'Until `git_path` is removed by a config.ini edit, trying: {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(f'Using: {cur_git_path}')
return cur_git_path
logger.debug(f'Not using: {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('No git specified, cannot use git commands')
return output, err, exit_status
cmd = ' '.join([git_path] + arg_list)
try:
logger.debug(f'Executing {cmd} with your shell in {sickgear.PROG_DIR}')
output, err, exit_status = cmdline_runner([git_path] + arg_list, env={'LANG': 'en_US.UTF-8'})
logger.debug(f'git output: {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(f'Successful return: {cmd}')
exit_status = 0
self.unsafe = False
elif 1 == exit_status:
logger.error(f'Failed: {cmd} returned: {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 = f'Fatal: {cmd} returned: {output}'
if 'develop' in output.lower() or 'main' in output.lower():
logger.error(msg)
else:
logger.debug(msg)
else:
logger.error(f'Treat as error for now, command: {cmd} returned: {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("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("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("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("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("git didn't return numbers for behind and ahead, not using it")
return
logger.debug(f'cur_commit = {self._cur_commit_hash}'
f', newest_commit = {self._newest_commit_hash}'
f', num_commits_behind = {self._num_commits_behind}'
f', num_commits_ahead = {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) &mdash; <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> &mdash; <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(f'Branch checkout: {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(f"Unable to contact github, can't check for update: {e!r}")
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.branch = self._old_branch
sickgear.BRANCH = self.branch
self._cur_commit_hash = self._old_commit_hash
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.
"""
self._old_branch = self._find_installed_branch()
self._old_commit_hash = self._cur_commit_hash
if self.branch == self._find_installed_branch():
if not self._cur_pr_number:
_, _, exit_status = self._run_git(['pull', '-f', '--no-rebase', '%s' % sickgear.GIT_REMOTE,
'%s' % self.branch])
else:
_, _, exit_status = self._run_git(['pull', '-f', '--no-rebase', '%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(f"Unable to contact github, can't check for update: {e!r}")
return False
installed_branch = self._find_installed_branch()
if self.branch != installed_branch:
logger.debug(f'Branch checkout: {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(f'cur_commit = {self._cur_commit_hash}'
f', newest_commit = {self._newest_commit_hash}'
f', num_commits_behind = {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("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. &mdash; <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) &mdash; <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 = os.path.join(sickgear.PROG_DIR, 'sg-update')
if os.path.isdir(sg_update_dir):
logger.log(f'Clearing out update folder {sg_update_dir} before extracting')
shutil.rmtree(sg_update_dir)
logger.log(f'Creating update folder {sg_update_dir} before extracting')
os.makedirs(sg_update_dir)
# retrieve file
logger.log(f'Downloading update from {tar_download_url!r}')
tar_download_path = os.path.join(sg_update_dir, 'sg-update.tar')
urllib.request.urlretrieve(tar_download_url, tar_download_path)
if not os.path.isfile(tar_download_path):
logger.error(f"Unable to retrieve new version from {tar_download_url}, can't update")
return False
if not tarfile.is_tarfile(tar_download_path):
logger.error(f"Retrieved version from {tar_download_url} is corrupt, can't update")
return False
# extract to sg-update dir
logger.log(f'Extracting file {tar_download_path}')
tar = tarfile.open(tar_download_path)
tar.extractall(sg_update_dir)
tar.close()
# delete .tar.gz
logger.log(f'Deleting file {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(f'Invalid update data, update failed: {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(f'Moving files from {content_dir} to {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(f'Unable to update {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(f'Error while trying to update: {ex(e)}')
logger.debug(f'Traceback: {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 []