##// END OF EJS Templates
feat: system-upgrade improved message on new version check page
feat: system-upgrade improved message on new version check page

File last commit:

r5095:aa627a5f default
r5429:e153339b default
Show More
update.py
83 lines | 3.0 KiB | text/x-python | PythonLexer
# Copyright (C) 2013-2023 RhodeCode GmbH
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License, version 3
# (only), as published by the Free Software Foundation.
#
# This program 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 Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
# This program is dual-licensed. If you wish to learn more about the
# RhodeCode Enterprise Edition, including its added features, Support services,
# and proprietary license terms, please see https://rhodecode.com/licenses/
import logging
import urllib.request
import urllib.error
import urllib.parse
from packaging.version import Version
import rhodecode
from rhodecode.lib.ext_json import json
from rhodecode.model import BaseModel
from rhodecode.model.meta import Session
from rhodecode.model.settings import SettingsModel
log = logging.getLogger(__name__)
class UpdateModel(BaseModel):
UPDATE_SETTINGS_KEY = 'update_version'
UPDATE_URL_SETTINGS_KEY = 'rhodecode_update_url'
@staticmethod
def get_update_data(update_url):
"""Return the JSON update data."""
ver = rhodecode.__version__
log.debug('Checking for upgrade on `%s` server', update_url)
opener = urllib.request.build_opener()
opener.addheaders = [('User-agent', 'RhodeCode-SCM/%s' % ver)]
response = opener.open(update_url)
response_data = response.read()
data = json.loads(response_data)
log.debug('update server returned data')
return data
def get_update_url(self):
settings = SettingsModel().get_all_settings()
return settings.get(self.UPDATE_URL_SETTINGS_KEY)
def store_version(self, version):
log.debug('Storing version %s into settings', version)
setting = SettingsModel().create_or_update_setting(
self.UPDATE_SETTINGS_KEY, version)
Session().add(setting)
Session().commit()
def get_stored_version(self, fallback=None):
obj = SettingsModel().get_setting_by_name(self.UPDATE_SETTINGS_KEY)
if obj:
return obj.app_settings_value
return fallback or '0.0.0'
def _sanitize_version(self, version):
"""
Cleanup our custom ver.
e.g 4.11.0_20171204_204825_CE_default_EE_default to 4.11.0
"""
return version.split('_')[0]
def is_outdated(self, cur_version, latest_version=None):
latest_version = latest_version or self.get_stored_version()
try:
cur_version = self._sanitize_version(cur_version)
return Version(latest_version) > Version(cur_version)
except Exception:
# could be invalid version, etc
return False