buildstatusimage.py

# From https://github.com/agroszer/buildbot-status-image/
#
# this is a snippet ready to be pasted into master.cfg
# then you can get the image by:
# http://localhost:8010/buildstatusimage?builder=runtests

import os
from buildbot.status import html  # NOQA
try:
    # buildbot 0.8.7p1
    from buildbot.status.results import SUCCESS, WARNINGS, FAILURE, SKIPPED, EXCEPTION  # NOQA
    from buildbot.status.results import Results
except ImportError:
    # buildbot 0.8.0
    from buildbot.status.builder import SUCCESS, WARNINGS, FAILURE, EXCEPTION, RETRY  # NOQA
    from buildbot.status.builder import Results
from buildbot.status.web.base import HtmlResource

class BuildStatusImageResource(HtmlResource):
    contentType = "image/svg+xml"

    def __init__(self, categories=None):
        HtmlResource.__init__(self)

    def content(self, request, ctx):
        """Display a build status image like Travis does."""

        status = self.getStatus(request)
        request.setHeader('Cache-Control', 'no-cache')

        # Get the parameters.
        name = request.args.get("builder", [None])[0]

        # Check if the builder in parameter exists.
        try:
            builder = status.getBuilder(name)
        except:
            # unknown builder
            return self._image('unknown')

        # Check if the build in parameter exists.
        build = builder.getLastFinishedBuild()
        if not build:
            # unknown build
            return self._image('unknown')

        # SUCCESS, WARNINGS, FAILURE, SKIPPED or EXCEPTION
        res = build.getResults()
        resname = Results[res]
        return self._image(resname)

    def _image(self, resname):
        img = 'public_html/status_%s.svg' % resname
        here = os.path.dirname(__file__)
        imgfile = os.path.join(here, img)

        return open(imgfile, 'rb').read()

# class WebStatus(html.WebStatus):
#    def setupUsualPages(self, numbuilds, num_events, num_events_max):
#        html.WebStatus.setupUsualPages(self, numbuilds, num_events, num_events_max)
#        self.putChild("buildstatusimage", BuildStatusImageResource())


# and use the WebStatus defined above instead of buildbot's
# c['status'].append(WebStatus(http_port=8010))