From 43c8a687653ccf2ae38310afab3d128d149ff869 Mon Sep 17 00:00:00 2001
From: David Strauss <david@fourkitchens.com>
Date: Wed, 23 Sep 2009 06:15:56 +0000
Subject: Initial Bazaar VCS support.

git-svn-id: https://svn.mcs.anl.gov/repos/bcfg/trunk/bcfg2@5451 ce84e21b-d406-0410-9b95-82705330c041
---
 src/lib/Server/Plugins/Bzr.py      | 45 ++++++++++++++++++++++++++++++++++++++
 src/lib/Server/Plugins/__init__.py |  1 +
 2 files changed, 46 insertions(+)
 create mode 100644 src/lib/Server/Plugins/Bzr.py

(limited to 'src')

diff --git a/src/lib/Server/Plugins/Bzr.py b/src/lib/Server/Plugins/Bzr.py
new file mode 100644
index 000000000..6498b006e
--- /dev/null
+++ b/src/lib/Server/Plugins/Bzr.py
@@ -0,0 +1,45 @@
+import os
+from subprocess import Popen, PIPE
+import Bcfg2.Server.Plugin
+
+# for debugging output only
+import logging
+logger = logging.getLogger('Bcfg2.Plugins.Bzr')
+
+class Bzr(Bcfg2.Server.Plugin.Plugin,
+          Bcfg2.Server.Plugin.Version):
+    name = 'Bzr'
+    __version__ = '$Id$'
+    __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
+
+        # path to bzr directory for bcfg2 repo
+        bzr_dir = "%s/.bzr" % datastore
+
+        # Read revision from bcfg2 repo
+        if os.path.isdir(bzr_dir):
+            revision = self.get_revision()
+        else:
+            logger.error("%s is not a directory" % bzr_dir)
+            raise Bcfg2.Server.Plugin.PluginInitError
+
+        logger.debug("Initialized Bazaar plugin with directory = %(dir)s at revision = %(rev)s" % {'dir': bzr_dir, 'rev': revision})
+
+    def get_revision(self):
+        '''Read Bazaar revision information for the bcfg2 repository'''
+        try:
+            data = Popen(("env LC_ALL=C bzr revno %s" %
+                         (self.datastore)), shell=True,
+                         stdout=PIPE).stdout.readlines()
+            revision = data
+        except IndexError:
+            logger.error("Failed to read bzr revno; disabling Bazaar support")
+            logger.error('''Ran command "bzr revno %s"''' % \
+                            (self.datastore))
+            logger.error("Got output: %s" % data)
+            raise Bcfg2.Server.Plugin.PluginInitError
+        return revision
diff --git a/src/lib/Server/Plugins/__init__.py b/src/lib/Server/Plugins/__init__.py
index 127035919..abcee623e 100644
--- a/src/lib/Server/Plugins/__init__.py
+++ b/src/lib/Server/Plugins/__init__.py
@@ -5,6 +5,7 @@ __all__ = [
            'Account',
            'Base',
            'Bundler',
+           'Bzr',
            'Cfg',
            'Decisions',
            'GBundler',
-- 
cgit v1.2.3-1-g7c22