1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
# Copyright (C) 2016 Codethink Limited
#
# This program 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; version 2 of the License.
#
# 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 General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
from __future__ import absolute_import
import urlparse
import itertools
try:
import gitlab
except ImportError:
gitlab = None
class MissingGitlabModuleError(Exception):
pass
class Gitlab(object):
'''Run commands on a GitLab instance.
This uses the python wrapper around the GitLab API.
Use of the API requires the private token of a user with master access
to the targetted group.
'''
def __init__(self, host, token):
if gitlab:
url = "http://" + host
self.gl = gitlab.Gitlab(url, token)
else:
raise MissingGitlabModuleError('gitlab module missing\n'
'\tpython-gitlab is required with GitLab as the git server')
def first(self, predicate, iterable):
return next(itertools.ifilter(predicate, iterable))
def split_path(self, path):
return path.rsplit('/', 1)
def find_project(self, group, project):
predicate = lambda x: x.namespace.name == group and x.name == project
return self.first(predicate, self.gl.projects.search(project))
def has_project(self, repo_path):
group, project = self.split_path(repo_path)
try:
return bool(self.find_project(group, project))
except StopIteration:
return False
def create_project(self, repo_path):
group_name, project_name = self.split_path(repo_path)
group = None
try:
group = self.gl.groups.get(group_name)
except gitlab.GitlabGetError as e:
if e.error_message == '404 Not found':
group = self.gl.groups.create(
{'name': group_name, 'path': group_name})
else:
raise
project = {
'name': project_name,
'public': True,
'merge_requests_enabled': False,
'namespace_id': group.id
}
self.gl.projects.create(project)
def list_projects(self):
return [x.path_with_namespace for x in self.gl.projects.list()]
def get_project_url(self, protocol, project_path):
'''Return the clone url for a GitLab project.
Depending on the protocol specified, will return a suitable clone url.
If 'ssh', a url in the format 'git@host:group/project.git' will be
returned.
If 'http' or 'https', the http_url_to_repo from the GitLab API is split
with urlparse into its constituent parts: the protocol (http by
default), the host, and the path ('group/project.git'). This is then
rejoined, replacing the protocol with what is specified. The resulting
format matching 'http(s)://host/group/project.git'.
'''
group, project = self.split_path(project_path)
project = self.find_project(group, project)
if protocol == 'ssh':
return project.ssh_url_to_repo
elif protocol in ('http', 'https'):
split = urlparse.urlsplit(project.http_url_to_repo)
return urlparse.urlunsplit((
protocol, split.netloc, split.path, '', ''))
|
