Allow to run directly from a git checkout (#1037)

Signed-off-by: Sol Jerome <sol.jerome@gmail.com>

1 files changed, 35 insertions, 0 deletions

diff --git a/src/lib/Bcfg2/Server/Plugins/Bzr.py b/src/lib/Bcfg2/Server/Plugins/Bzr.py
new file mode 100644
index 000000000..a71021cb5
--- /dev/null
+++ b/src/lib/Bcfg2/Server/Plugins/Bzr.py
@@ -0,0 +1,35 @@
+import Bcfg2.Server.Plugin
+from bzrlib.workingtree import WorkingTree
+from bzrlib import errors
+
+# for debugging output only
+import logging
+logger = logging.getLogger('Bcfg2.Plugins.Bzr')
+
+class Bzr(Bcfg2.Server.Plugin.Plugin,
+          Bcfg2.Server.Plugin.Version):
+    """Bzr is a version plugin for dealing with Bcfg2 repos."""
+    name = 'Bzr'
+    __author__ = 'bcfg-dev@mcs.anl.gov'
+
+    def __init__(self, core, datastore):
+        Bcfg2.Server.Plugin.Plugin.__init__(self, core, datastore)
+        self.core = core
+        self.datastore = datastore
+
+        # Read revision from bcfg2 repo
+        revision = self.get_revision()
+
+        logger.debug("Initialized Bazaar plugin with directory = %(dir)s at revision = %(rev)s" % {'dir': datastore, 'rev': revision})
+
+    def get_revision(self):
+        """Read Bazaar revision information for the Bcfg2 repository."""
+        try:
+            working_tree = WorkingTree.open(self.datastore)
+            revision = str(working_tree.branch.revno())
+            if working_tree.has_changes(working_tree.basis_tree()) or working_tree.unknowns():
+                revision += "+"
+        except errors.NotBranchError:
+            logger.error("Failed to read Bazaar branch; disabling Bazaar support")
+            raise Bcfg2.Server.Plugin.PluginInitError
+        return revision