summaryrefslogtreecommitdiffstats
path: root/src/lib/Bcfg2/Server/Plugins/Bzr.py
blob: a71021cb52063ee021eb8327ec35c3e1de0a0ada (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
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