summaryrefslogtreecommitdiffstats
path: root/src/lib/Server/Plugins/Bzr.py
blob: 6498b006ec4ea6fce94304f37c2eea6ab20be0d4 (plain)
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
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