summaryrefslogtreecommitdiffstats
path: root/src/lib/Bcfg2/Server/Plugins/Fossil.py
blob: 1b1627688f1d63a51ad17f0076ed95b51d521254 (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
46
47
48
49
50
51
import os
from subprocess import Popen, PIPE
import Bcfg2.Server.Plugin

# for debugging output only
import logging
logger = logging.getLogger('Bcfg2.Plugins.Fossil')

class Fossil(Bcfg2.Server.Plugin.Plugin,
             Bcfg2.Server.Plugin.Version):
    """Fossil is a version plugin for dealing with Bcfg2 repos."""
    name = 'Fossil'
    __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 fossil file for bcfg2 repo
        fossil_file = "%s/_FOSSIL_" % datastore

        # Read revision from bcfg2 repo
        if os.path.isfile(fossil_file):
            revision = self.get_revision()
        elif not os.path.isdir(datastore):
            logger.error("%s is not a directory" % datastore)
            raise Bcfg2.Server.Plugin.PluginInitError
        else:
            logger.error("%s is not a file" % fossil_file)
            raise Bcfg2.Server.Plugin.PluginInitError

        logger.debug("Initialized Fossil.py plugin with %(ffile)s at revision %(frev)s" \
                    % {'ffile': fossil_file, 'frev': revision})

    def get_revision(self):
        """Read fossil revision information for the Bcfg2 repository."""
        try:
            data = Popen("env LC_ALL=C fossil info",
                        shell=True,
                        cwd=self.datastore,
                        stdout=PIPE).stdout.readlines()
            revline = [line.split(': ')[1].strip() for line in data if \
                       line.split(': ')[0].strip() == 'checkout'][-1]
            revision = revline.split(' ')[0]
        except IndexError:
            logger.error("Failed to read fossil info; disabling fossil support")
            logger.error('''Ran command "fossil info" from directory "%s"''' % (self.datastore))
            logger.error("Got output: %s" % data)
            raise Bcfg2.Server.Plugin.PluginInitError
        return revision