sid/bts.py

   1 # -*- coding: utf-8 -*-
   2
   3 # Copyright (C) 2010, 2011 Anaël Verrier <elghinn@free.fr>
   4 # Copyright (C) 2015, 2020 kaliko <kaliko@azylum.org>
   5
   6 # This program is free software: you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation, version 3 only.
   9
  10 # This program is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14
  15 # You should have received a copy of the GNU General Public License
  16 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17
  18
  19 from re import compile as re_compile
  20
  21 import debianbts
  22
  23 from .plugin import Plugin, botcmd
  24
  25
  26 class Bugs(Plugin):
  27     """Gets bugs info from the BTS
  28     """
  29     re_bugs = re_compile(r'(?<=#)(\d{6,7})')
  30     re_pkg = re_compile(r'(?P<package>[0-9a-z.+-]+)$')
  31
  32     def __init__(self, bot):
  33         Plugin.__init__(self, bot)
  34         bot.add_event_handler("muc::%s::message" %
  35                               self.bot.room, self.muc_message)
  36
  37     def muc_message(self, msg):
  38         """Handler method dealing with MUC incoming messages"""
  39         # Does not reply to myself
  40         if msg['mucnick'] == self.bot.nick:
  41             return
  42         if '#' not in msg['body']:
  43             return
  44         bugs = list()
  45         for bug_id in set(Bugs.re_bugs.findall(msg['body'].strip())):
  46             self.log.debug('got bug id: %s', bug_id)
  47             query = debianbts.get_status(bug_id)
  48             if len(query) == 1:
  49                 bug = query[0]
  50                 url = debianbts.BTS_URL + bug_id
  51                 bugs.append({'id': bug_id,
  52                              'package': bug.package,
  53                              'summary': bug.subject,
  54                              'url': url})
  55             else:
  56                 self.log.warning('Wrong bug number "%s"?', bug_id)
  57                 bugs.append({'id': bug_id})
  58         for bug in bugs:
  59             if len(bug) == 1:
  60                 message = 'Invalid bug id: {id}'.format(**bug)
  61                 self.reply(msg, message)
  62             else:
  63                 message = {'mhtml': '<a href="%(url)s">#%(id)s</a>: %(package)s “ %(summary)s ”' % bug,
  64                            'mbody': '#%(id)s: %(package)s “ %(summary)s ” %(url)s' % bug}
  65                 self.reply(msg, message)
  66
  67     @botcmd
  68     def bugs(self, rcv, args):
  69         """Intercepts bugs number in messages (as #629234), reply a bug summary.
  70         !bugs pkg-name : Returns latest bug reports if any
  71         """
  72         if not args:
  73             return
  74         if len(args) > 1:
  75             self.log.info('more than one packages provided')
  76         pkg = Bugs.re_pkg.match(args[0])
  77         if not pkg:
  78             msg = 'Wrong package name format re: "{}"'.format(Bugs.re_pkg.pattern)
  79             self.reply(rcv, msg)
  80             return
  81         reports_ids = debianbts.get_bugs(status='open', **pkg.groupdict())
  82         if not reports_ids:
  83             self.reply(rcv, 'No open bugs for "{}"'.format(pkg.string))
  84             return
  85         reports = debianbts.get_status(reports_ids)
  86         reports = sorted(reports, key=lambda r: r.date)
  87         msg = ['Latest reports for {1} (total {0})'.format(len(reports), pkg.string)]
  88         # Reverse and take last reports
  89         for rep in reports[::-1][:4]:
  90             msg.append('{r.bug_num}: {r.date:%Y-%m-%d} {r.subject}'.format(r=rep))
  91         message = {'mbody': '\n'.join(msg)}
  92         self.reply(rcv, message)