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
|
""" The Git plugin provides a revision interface for Bcfg2 repos using
git. """
import sys
import Bcfg2.Server.Plugin
from subprocess import Popen, PIPE
try:
import git
HAS_GITPYTHON = True
except ImportError:
HAS_GITPYTHON = False
class Git(Bcfg2.Server.Plugin.Version):
""" The Git plugin provides a revision interface for Bcfg2 repos
using git. """
__author__ = 'bcfg-dev@mcs.anl.gov'
__vcs_metadata_path__ = ".git"
if HAS_GITPYTHON:
__rmi__ = Bcfg2.Server.Plugin.Version.__rmi__ + ['Update']
def __init__(self, core, datastore):
Bcfg2.Server.Plugin.Version.__init__(self, core, datastore)
if HAS_GITPYTHON:
self.repo = git.Repo(self.vcs_root)
else:
self.logger.debug("Git: GitPython not found, using CLI interface "
"to Git")
self.repo = None
self.logger.debug("Initialized git plugin with git directory %s" %
self.vcs_path)
def get_revision(self):
"""Read git revision information for the Bcfg2 repository."""
try:
if HAS_GITPYTHON:
return self.repo.head.commit.hexsha
else:
cmd = ["git", "--git-dir", self.vcs_path,
"--work-tree", self.vcs_root, "rev-parse", "HEAD"]
self.debug_log("Git: Running cmd")
proc = Popen(cmd, stdout=PIPE, stderr=PIPE)
rv, err = proc.communicate()
if proc.wait():
raise Exception(err)
return rv
except:
err = sys.exc_info()[1]
msg = "Git: Error getting revision from %s: %s" % (self.vcs_root,
err)
self.logger.error(msg)
raise Bcfg2.Server.Plugin.PluginExecutionError(msg)
def Update(self):
""" Git.Update() => True|False
Update the working copy against the upstream repository
"""
try:
self.repo.git.pull("--rebase")
self.logger.info("Git repo at %s updated to %s" %
(self.vcs_root, self.get_revision()))
return True
except: # pylint: disable=W0702
err = sys.exc_info()[1]
msg = "Failed to pull from git repository: %s" % err
self.logger.error(msg)
raise Bcfg2.Server.Plugin.PluginExecutionError(msg)
|
