60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
# coding=utf-8
|
|
from __future__ import absolute_import, division, print_function
|
|
|
|
__license__ = 'GNU Affero General Public License http://www.gnu.org/licenses/agpl.html'
|
|
__copyright__ = "Copyright (C) 2017 The OctoPrint Project - Released under terms of the AGPLv3 License"
|
|
|
|
import requests
|
|
import logging
|
|
import base64
|
|
|
|
BRANCH_HEAD_URL = "https://api.bitbucket.org/2.0/repositories/{user}/{repo}/commit/{branch}"
|
|
|
|
logger = logging.getLogger("octoprint.plugins.softwareupdate.version_checks.bitbucket_commit")
|
|
|
|
|
|
def _get_latest_commit(user, repo, branch, api_user=None, api_password=None):
|
|
url = BRANCH_HEAD_URL.format(user=user, repo=repo, branch=branch)
|
|
headers = {}
|
|
if api_user is not None and api_password is not None:
|
|
auth_value = base64.b64encode(b"{user}:{pw}".format(user=api_user, pw=api_password))
|
|
headers["authorization"] = "Basic {}".format(auth_value)
|
|
r = requests.get(url, headers=headers)
|
|
|
|
if not r.status_code == requests.codes.ok:
|
|
return None
|
|
|
|
reference = r.json()
|
|
if not "hash" in reference:
|
|
return None
|
|
|
|
return reference["hash"]
|
|
|
|
|
|
def get_latest(target, check):
|
|
from ..exceptions import ConfigurationInvalid
|
|
|
|
if "user" not in check or "repo" not in check:
|
|
raise ConfigurationInvalid("Update configuration for %s of type bitbucket_commit needs all of user and repo" % target)
|
|
|
|
branch = "master"
|
|
if "branch" in check and check["branch"] is not None:
|
|
branch = check["branch"]
|
|
|
|
api_user = check.get("api_user")
|
|
api_password = check.get("api_password")
|
|
|
|
current = check.get("current")
|
|
|
|
remote_commit = _get_latest_commit(check["user"], check["repo"], branch, api_user, api_password)
|
|
|
|
information = dict(
|
|
local=dict(name="Commit {commit}".format(commit=current if current is not None else "unknown"), value=current),
|
|
remote=dict(name="Commit {commit}".format(commit=remote_commit if remote_commit is not None else "unknown"), value=remote_commit)
|
|
)
|
|
is_current = (current is not None and current == remote_commit) or remote_commit is None
|
|
|
|
logger.debug("Target: %s, local: %s, remote: %s" % (target, current, remote_commit))
|
|
|
|
return information, is_current
|
|
|