X-Git-Url: https://git.ralfj.de/mass-build.git/blobdiff_plain/c0813ab32e70aacc90adb18457b57aef8bc9f920..6b134177027b31d6507fc5904595a2847116c1bf:/vcs.py diff --git a/vcs.py b/vcs.py index 3e04f46..1494478 100644 --- a/vcs.py +++ b/vcs.py @@ -1,5 +1,5 @@ # mass-build - Easily Build Software Involving a Large Amount of Source Repositories -# Copyright (C) 2012 Ralf Jung +# Copyright (C) 2012-2013 Ralf Jung # # 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 @@ -15,7 +15,7 @@ # along with this program; if not, write to the Free Software # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. -import os, git, subprocess, re +import os, subprocess, re '''A VCS must have an "update" method with an optional "mode" parameter taking one of the three values below, a "version" method returning a version name (or None), @@ -27,17 +27,28 @@ MODE_RESET = 2 def natural_sort_key(val): return [ (int(c) if c.isdigit() else c) for c in re.split('([0-9]+)', val) ] +def get_non_digit_prefix(val): + return re.match('[^0-9]*', val).group(0) + +class GitCommand: + def __getattr__(self, name): + def call(*args, suppress_stderr = False, split = True): + cmd = ["git", name.replace('_', '-')] + list(args) + with subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE if suppress_stderr else None) as p: + (stdout, stderr) = p.communicate() + if p.returncode != 0: + raise Exception("Running %s returned non-zero exit code %d" % (str(cmd), p.returncode)) + stdout = stdout.decode('utf-8').strip('\n') + return stdout.split('\n') if split else stdout + return call +git = GitCommand() + # Fetch updates from git class Git: - - def __init__(self, folder, url, commit): + def __init__(self, folder, config): self.folder = os.path.abspath(folder) - self.url = url - self.commit = commit - - class _ProgressPrinter(git.remote.RemoteProgress): - def update(self, op_code, cur_count, max_count=None, message=''): - print self._cur_line+(" "*30)+"\r", + self.url = config['url'] + self.commit = config['version'] def update(self, mode = MODE_REBASE): isBranch = (self.commit.startswith('origin/')) @@ -48,61 +59,51 @@ class Git: # get us a git repository, and the "origin" remote if os.path.exists(self.folder): # load existing repo - repo = git.Repo(self.folder) - origin = repo.remotes.origin - origin.config_writer.set_value("url", self.url) # make sure we use the current URL + os.chdir(self.folder) + git.remote("set-url", "origin", self.url) # make sure we use the current URL else: # create a new one os.makedirs(self.folder) - repo = git.Repo.init(self.folder) - origin = repo.create_remote('origin', self.url) - origin.fetch(progress=Git._ProgressPrinter()) # download new data - print " "*80+"\r", # clean the line we are in + os.chdir(self.folder) + git.init() + git.remote("add", "origin", self.url) + git.fetch("origin") if mode == MODE_FETCH: return # create/find correct branch - if branchname in repo.heads: - branch = repo.heads[branchname] - else: - branch = repo.create_head(branchname, self.commit) - if isBranch: # track remote branch - branch.set_tracking_branch(origin.refs[branchname]) + if not git.branch("--list", branchname): # the branch does not yet exit + git.branch(branchname, self.commit) + if isBranch: # make sure we track the correct remote branch + git.branch("-u", self.commit, branchname) # update it to the latest remote commit - branch.checkout() + git.checkout(branchname, suppress_stderr=True) if mode == MODE_RESET: - repo.head.reset(self.commit, working_tree=True) + git.reset("--hard", self.commit) else: - repo.git.rebase(self.commit) + git.rebase(self.commit) # update submodules - repo.git.submodule("update", "--init", "--recursive", "--rebase") + git.submodule("update", "--init", "--recursive", "--rebase") # done - print "...done", - if repo.head.reference.commit != repo.commit(self.commit): - print "(keeping local patches around)", - print + print("...done", end=' ') + if git.rev_parse("HEAD", split=False) != git.rev_parse(self.commit, split=False): + print("(keeping local patches around)", end=' ') + print() def version(self): repo = git.Repo(self.folder) v = repo.git.describe() - if v.startswith('v'): v = v[1:] - return v + return v[len(get_non_digit_prefix(v)):] # remove the non-digit prefix from v (so that it starts with a number) def checkVersions(self): - repo = git.Repo(self.folder) - # get tag for current commit, if any - commit = repo.commit(self.commit) - commitTag = filter(lambda t: t.commit == commit, repo.tags) - if not commitTag: - print "Version is not a tag" - return - currentVersion = str(commitTag[0]) - # get sorted list of tag names - tags = map(str, repo.tags) - tags = filter(lambda t: natural_sort_key(t) > natural_sort_key(currentVersion), tags) + self.update(mode = MODE_FETCH) + currentVersion = git.describe(split=False) + # get sorted list of tag names with the same non-digit prefix and higher version number + tags = git.tag() + tags = [t for t in tags if get_non_digit_prefix(t) == get_non_digit_prefix(currentVersion) and natural_sort_key(t) > natural_sort_key(currentVersion)] if not tags: return tags.sort(key = natural_sort_key) - print "Versions newer than "+self.commit+" available:" - print tags + print("Versions newer than "+currentVersion+" available:") + print(tags) # Fetch updates via SVN class SVN: @@ -124,4 +125,4 @@ class SVN: return None def checkVersions(self): - print "Version checking not supporting with SVN" + print("Version checking not supporting with SVN")