e6ecac7ab0
Changed BaseException to CommandException for python34 test compatibility Change-Id: Id5db4950a06a0875dc6485758318f0bc9090f6f2
124 lines
3.8 KiB
Python
124 lines
3.8 KiB
Python
# Copyright (c) 2015 Rackspace
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
|
# implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
# Tools for interacting with Github.
|
|
|
|
import base64
|
|
import getpass
|
|
import json
|
|
import random
|
|
import re
|
|
import string
|
|
|
|
import httplib2
|
|
|
|
|
|
class GitHubAuth(object):
|
|
username = None
|
|
password = None
|
|
full_repo_name = None
|
|
git_url = None
|
|
_token = None
|
|
|
|
_auth_url = 'https://api.github.com/authorizations'
|
|
|
|
def __init__(self, git_url):
|
|
self.git_url = git_url
|
|
|
|
user_org_name, repo = '', ''
|
|
|
|
github_regex = r'github\.com[:/](.+?)/(.+?)($|/$|\.git$|\.git/$)'
|
|
repo_pat = re.compile(github_regex)
|
|
match = repo_pat.search(self.git_url)
|
|
if match:
|
|
user_org_name, repo = match.group(1), match.group(2)
|
|
else:
|
|
raise ValueError("Failed to parse %s." % git_url)
|
|
self.full_repo_name = '/'.join([user_org_name, repo])
|
|
prompt = ("Username for repo '%s' [%s]:" %
|
|
(self.full_repo_name, user_org_name))
|
|
username = raw_input(prompt) or user_org_name
|
|
password = getpass.getpass("Password: ")
|
|
self.username = username
|
|
self.password = password
|
|
|
|
@property
|
|
def _auth_header(self):
|
|
authstring = '%s:%s' % (self.username, self.password)
|
|
auth = base64.encodestring(authstring)
|
|
return {
|
|
'Authorization': 'Basic %s' % auth,
|
|
'Content-Type': 'application/json',
|
|
}
|
|
|
|
@property
|
|
def token(self):
|
|
if self._token is None:
|
|
self._get_repo_token()
|
|
return self._token
|
|
|
|
def _get_repo_token(self):
|
|
note = ''.join(random.sample(string.lowercase, 5))
|
|
auth_info = {
|
|
'scopes': 'repo',
|
|
'note': 'Solum-status-%s' % note,
|
|
}
|
|
resp, content = httplib2.Http().request(
|
|
self._auth_url,
|
|
'POST',
|
|
headers=self._auth_header,
|
|
body=json.dumps(auth_info))
|
|
if resp.get('status') in ['200', '201']:
|
|
response_body = json.loads(content)
|
|
self._token = response_body.get('token')
|
|
|
|
def create_webhook(self, trigger_uri):
|
|
hook_url = ('https://api.github.com/repos/%s/hooks' %
|
|
self.full_repo_name)
|
|
hook_info = {
|
|
'name': 'web',
|
|
'events': ['pull_request', 'commit_comment'],
|
|
'config': {
|
|
'content_type': 'json',
|
|
'url': trigger_uri,
|
|
}
|
|
}
|
|
resp, content = httplib2.Http().request(
|
|
hook_url,
|
|
'POST',
|
|
headers=self._auth_header,
|
|
body=json.dumps(hook_info))
|
|
if resp.get('status') not in ['200', '201']:
|
|
pass
|
|
|
|
def add_ssh_key(self, public_key=None, is_private=False):
|
|
if not public_key:
|
|
return
|
|
api_url = ('https://api.github.com/repos/%s/keys' %
|
|
self.full_repo_name)
|
|
if is_private:
|
|
api_url = 'https://api.github.com/user/keys'
|
|
key_info = {
|
|
'title': 'devops@Solum',
|
|
'key': public_key,
|
|
}
|
|
resp, content = httplib2.Http().request(
|
|
api_url,
|
|
'POST',
|
|
headers=self._auth_header,
|
|
body=json.dumps(key_info))
|
|
if resp.get('status') not in ['200', '201']:
|
|
pass
|