Merge lp:~james-w/tarmac/check-command into lp:~ubuntuone-hackers/tarmac/trunk

[James Westby]

Hi,

This adds the "check" command that we were using with the jenkins
scripttrigger plugin to only run the job if there were branches to
merge.

Thanks,

James

[Jonathan Lange (jml)]

Good to land.

_any_merges should probably have a docstring, or at least get rid of that blank line after the function declaration.

review: Approve

=== modified file 'tarmac/bin/commands.py'
--- tarmac/bin/commands.py	2012-12-18 21:25:38 +0000
+++ tarmac/bin/commands.py	2012-12-18 21:25:38 +0000
@@ -400,3 +400,50 @@
         if len(filter(lambda x: x is False, statuses)) > 0:
             return 2
         return 0
+
+
+class cmd_check(TarmacCommand):
+    '''Check whether there are any merge proposals ready to land.'''
+
+    takes_args = ['branch_url']
+    takes_options = [
+        options.http_debug_option,
+        options.debug_option]
+
+    def _any_merges(self, branch_url):
+
+        lp_branch = self.launchpad.branches.getByUrl(url=branch_url)
+        if lp_branch is None:
+            self.logger.info('Not a valid branch: {0}'.format(branch_url))
+            return
+
+        proposals = _get_mergable_proposals_for_branch(lp_branch, self.logger)
+
+        if not proposals:
+            self.logger.info(
+                'No approved proposals found for %(branch_url)s' % {
+                    'branch_url': branch_url})
+            return False
+        return True
+
+    def run(self, branch_url, launchpad=None, **kwargs):
+        for key, value in kwargs.iteritems():
+            self.config.set('Tarmac', key, value)
+
+        setup(self.logger, debug=self.config.debug,
+                http_debug=self.config.http_debug)
+
+        self.launchpad = launchpad
+        if self.launchpad is None:
+            self.logger.debug('Loading launchpad object')
+            self.launchpad = self.get_launchpad_object()
+            self.logger.debug('launchpad object loaded')
+
+        self.logger.debug('%(branch_url)s specified as branch_url' % {
+            'branch_url': branch_url})
+        if not branch_url.startswith('lp:'):
+            raise TarmacCommandError('Branch urls must start with lp:')
+        ret = self._any_merges(branch_url)
+        if ret:
+            return 0
+        return 1