mirror of
https://github.com/Mikaela/Limnoria.git
synced 2024-11-23 19:19:32 +01:00
337 lines
13 KiB
Python
337 lines
13 KiB
Python
#!/usr/bin/env python
|
|
|
|
###
|
|
# Copyright (c) 2003, Daniel Berlin
|
|
# Based on code from kibot
|
|
# All rights reserved.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without
|
|
# modification, are permitted provided that the following conditions are met:
|
|
#
|
|
# * Redistributions of source code must retain the above copyright notice,
|
|
# this list of conditions, and the following disclaimer.
|
|
# * Redistributions in binary form must reproduce the above copyright notice,
|
|
# this list of conditions, and the following disclaimer in the
|
|
# documentation and/or other materials provided with the distribution.
|
|
# * Neither the name of the author of this software nor the name of
|
|
# contributors to this software may be used to endorse or promote products
|
|
# derived from this software without specific prior written consent.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
|
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
# POSSIBILITY OF SUCH DAMAGE.
|
|
###
|
|
|
|
"""
|
|
Bugzilla bug retriever
|
|
"""
|
|
|
|
import os
|
|
import re
|
|
import string
|
|
import urllib
|
|
import base64
|
|
import xml.dom.minidom as minidom
|
|
from htmlentitydefs import entitydefs as entities
|
|
|
|
import sqlite
|
|
|
|
import conf
|
|
import utils
|
|
import plugins
|
|
import privmsgs
|
|
import callbacks
|
|
import ircutils
|
|
|
|
dbfilename = os.path.join(conf.dataDir, 'Bugzilla.db')
|
|
def makeDb(filename):
|
|
if os.path.exists(filename):
|
|
return sqlite.connect(filename)
|
|
db = sqlite.connect(filename)
|
|
cursor = db.cursor()
|
|
cursor.execute("""CREATE TABLE bugzillas (
|
|
shorthand TEXT UNIQUE ON CONFLICT REPLACE,
|
|
url TEXT,
|
|
description TEXT
|
|
)""")
|
|
cursor = db.cursor();
|
|
cursor.execute("""INSERT INTO bugzillas VALUES (%s, %s, %s)""",
|
|
'rh', 'http://bugzilla.redhat.com/bugzilla', 'Red Hat')
|
|
cursor.execute("""INSERT INTO bugzillas VALUES (%s, %s, %s)""",
|
|
'gnome', 'http://bugzilla.gnome.org', 'Gnome')
|
|
cursor.execute("""INSERT INTO bugzillas VALUES (%s, %s, %s)""",
|
|
'xim', 'http://bugzilla.ximian.com', 'Ximian')
|
|
cursor.execute("""INSERT INTO bugzillas VALUES (%s, %s, %s)""",
|
|
'moz', 'http://bugzilla.mozilla.org', 'Mozilla')
|
|
cursor.execute("""INSERT INTO bugzillas VALUES (%s, %s, %s)""",
|
|
'gcc', 'http://gcc.gnu.org/bugzilla', 'GCC')
|
|
db.commit()
|
|
return db
|
|
|
|
class BugError(Exception):
|
|
"""A bugzilla error"""
|
|
def __init__(self, args = None):
|
|
Exception.__init__(self)
|
|
self.args = args
|
|
|
|
def configure(onStart, afterConnect, advanced):
|
|
from questions import expect, anything, yn
|
|
onStart.append('load Bugzilla')
|
|
if advanced:
|
|
print 'The Bugzilla plugin has the functionality to watch for URLs'
|
|
print 'that match a specific pattern (we call this a snarfer). When'
|
|
print 'supybot sees such a URL, he will parse the web page for'
|
|
print 'information and reply with the results.\n'
|
|
if yn('Do you want the Bugzilla snarfer enabled by default?') == 'n':
|
|
onStart.append('Bugzilla toggle bug off')
|
|
|
|
class Bugzilla(callbacks.PrivmsgCommandAndRegexp, plugins.Toggleable):
|
|
"""Show a link to a bug report with a brief description"""
|
|
threaded = True
|
|
regexps = ['bzSnarfer']
|
|
toggles = plugins.ToggleDictionary({'bug' : True})
|
|
|
|
def __init__(self):
|
|
callbacks.PrivmsgCommandAndRegexp.__init__(self)
|
|
plugins.Toggleable.__init__(self)
|
|
self.entre = re.compile('&(\S*?);')
|
|
self.db = makeDb(dbfilename)
|
|
|
|
def die(self):
|
|
self.db.close()
|
|
del self.db
|
|
|
|
def add(self, irc, msg, args):
|
|
"""<abbreviation> <url> <description>
|
|
|
|
Add a bugzilla <url> to the list of defined bugzillae. <abbreviation>
|
|
is the name that will be used to reference the zilla in all other
|
|
commands. <description> is the common name for the bugzilla and will
|
|
be listed with the bugzilla query.
|
|
E.g.: add rh http://bugzilla.redhat.com/bugzilla Red Hat Zilla"""
|
|
(shorthand, url, description) = privmsgs.getArgs(args, needed=3)
|
|
cursor = self.db.cursor()
|
|
cursor.execute("""INSERT INTO bugzillas VALUES (%s, %s, %s)""",
|
|
shorthand, url, description)
|
|
self.db.commit()
|
|
irc.reply(msg, 'Added bugzilla entry for "%s" with shorthand "%s"' % (
|
|
description, shorthand))
|
|
return
|
|
|
|
def remove(self, irc, msg, args):
|
|
"""<abbreviation>
|
|
|
|
Remove the bugzilla associated with <abbreviation> from the list of
|
|
defined bugzillae.
|
|
E.g.: remove rh"""
|
|
shorthand = privmsgs.getArgs(args)
|
|
cursor = self.db.cursor()
|
|
cursor.execute("""SELECT * from bugzillas where shorthand = %s""",
|
|
shorthand)
|
|
if cursor.rowcount == 0:
|
|
irc.reply(msg, 'Bugzilla "%s" not defined. Try list.' %
|
|
shorthand)
|
|
return
|
|
cursor.execute("""DELETE FROM bugzillas where shorthand = %s""",
|
|
shorthand)
|
|
self.db.commit()
|
|
irc.reply(msg, 'Deleted bugzilla "%s"' % shorthand)
|
|
return
|
|
|
|
def list(self, irc, msg, args):
|
|
"""[<abbreviation>]
|
|
|
|
List defined bugzillae. If <abbreviation> is specified, list the
|
|
information for that bugzilla.
|
|
E.g.: list rh; or just list"""
|
|
shorthand = privmsgs.getArgs(args, needed=0, optional=1)
|
|
if shorthand:
|
|
cursor = self.db.cursor()
|
|
cursor.execute("""SELECT url,description from bugzillas where
|
|
shorthand = %s""", shorthand)
|
|
if cursor.rowcount == 0:
|
|
irc.reply(msg, 'No such bugzilla defined: "%s".' % shorthand)
|
|
return
|
|
url, description = cursor.fetchone()
|
|
irc.reply(msg, '%s: %s, %s' % (shorthand, description, url))
|
|
return
|
|
else:
|
|
cursor = self.db.cursor()
|
|
cursor.execute("""SELECT shorthand from bugzillas""")
|
|
if cursor.rowcount == 0:
|
|
irc.reply(msg, 'No bugzillae defined. Add some with "add"!')
|
|
return
|
|
results = ['%s' % (item[0]) for item in cursor.fetchall()]
|
|
irc.reply(msg, 'Defined bugzillae: %s' % ' '.join(results))
|
|
return
|
|
|
|
def bzSnarfer(self, irc, msg, match):
|
|
r"(http://\S+)/show_bug.cgi\?id=([0-9]+)"
|
|
if not self.toggles.get('bug', channel=msg.args[0]):
|
|
return
|
|
queryurl = '%s/xml.cgi?id=%s' % (match.group(1), match.group(2))
|
|
try:
|
|
summary = self._get_short_bug_summary(queryurl, 'Snarfed '\
|
|
'Bugzilla URL', match.group(2))
|
|
except BugError, e:
|
|
irc.reply(msg, str(e))
|
|
return
|
|
except IOError, e:
|
|
msgtouser = '%s. Try yourself: %s' % (e, queryurl)
|
|
irc.reply(msg, msgtouser)
|
|
return
|
|
report = {}
|
|
report['id'] = match.group(2)
|
|
report['url'] = str('%s/show_bug.cgi?id=%s' % (match.group(1),
|
|
match.group(2)))
|
|
report['title'] = str(summary['title'])
|
|
report['summary'] = str(self._mk_summary_string(summary))
|
|
report['product'] = str(summary['product'])
|
|
irc.reply(msg, '%(product)s bug #%(id)s: %(title)s %(summary)s'
|
|
% report, prefixName = False)
|
|
|
|
def bug(self, irc, msg, args):
|
|
"""<abbreviation> <number>
|
|
|
|
Look up bug <number> in the bugzilla associated with <abbreviation>.
|
|
E.g.: bug rh 10301"""
|
|
(shorthand, num) = privmsgs.getArgs(args, needed=2)
|
|
cursor = self.db.cursor()
|
|
cursor.execute("""SELECT url,description from bugzillas where
|
|
shorthand = %s""", shorthand)
|
|
if cursor.rowcount == 0:
|
|
irc.reply(msg, 'Bugzilla "%s" is not defined.' % shorthand)
|
|
return
|
|
if not self._is_bug_number(num):
|
|
irc.reply(msg, '"%s" does not seem to be a number' % num)
|
|
return
|
|
url, desc = cursor.fetchone()
|
|
queryurl = '%s/xml.cgi?id=%s' % (url, num)
|
|
try:
|
|
summary = self._get_short_bug_summary(queryurl, desc, num)
|
|
except BugError, e:
|
|
irc.reply(msg, str(e))
|
|
return
|
|
except IOError, e:
|
|
msgtouser = '%s. Try yourself: %s' % (e, queryurl)
|
|
irc.reply(msg, msgtouser)
|
|
return
|
|
|
|
report = {}
|
|
report['zilla'] = str(desc)
|
|
report['id'] = str(num)
|
|
report['url'] = str('%s/show_bug.cgi?id=%s' % (url, num))
|
|
report['title'] = str(summary['title'])
|
|
report['summary'] = str(self._mk_summary_string(summary))
|
|
irc.reply(msg, '%(zilla)s bug #%(id)s: %(title)s %(summary)s %(url)s'
|
|
% report)
|
|
return
|
|
|
|
def _mk_summary_string(self, summary):
|
|
ary = []
|
|
if 'product' in summary:
|
|
ary.append(ircutils.bold('Product: ') + summary['product'])
|
|
if 'component' in summary:
|
|
ary.append(ircutils.bold('Component: ') + summary['component'])
|
|
if 'severity' in summary:
|
|
ary.append(ircutils.bold('Severity: ') + summary['severity'])
|
|
if 'assigned to' in summary:
|
|
ary.append(ircutils.bold('Assigned to: ') + summary['assigned to'])
|
|
if 'status' in summary:
|
|
ary.append(ircutils.bold('Status: ') + summary['status'])
|
|
if 'resolution' in summary:
|
|
ary.append(ircutils.bold('Resolution: ') + summary['resolution'])
|
|
out = string.join(ary, ', ')
|
|
return out
|
|
|
|
def _is_bug_number(self, bug):
|
|
try: int(bug)
|
|
except: return 0
|
|
else: return 1
|
|
|
|
def _get_short_bug_summary(self, url, desc, num):
|
|
bugxml = self._getbugxml(url, desc)
|
|
try: zilladom = minidom.parseString(bugxml)
|
|
except Exception, e:
|
|
msg = 'Could not parse XML returned by %s bugzilla: %s'
|
|
raise BugError(str(msg % (desc, e)))
|
|
bug_n = zilladom.getElementsByTagName('bug')[0]
|
|
if bug_n.hasAttribute('error'):
|
|
errtxt = bug_n.getAttribute('error')
|
|
zilladom.unlink()
|
|
msg = 'Error getting %s bug #%s: %s' % (desc, num, errtxt)
|
|
raise BugError(str(msg))
|
|
summary = {}
|
|
try:
|
|
node = bug_n.getElementsByTagName('short_desc')[0]
|
|
summary['title'] = self._getnodetxt(node)
|
|
node = bug_n.getElementsByTagName('bug_status')[0]
|
|
summary['status'] = self._getnodetxt(node)
|
|
try:
|
|
node = bug_n.getElementsByTagName('resolution')[0]
|
|
summary['resolution'] = self._getnodetxt(node)
|
|
except:
|
|
pass
|
|
node = bug_n.getElementsByTagName('assigned_to')[0]
|
|
summary['assigned to'] = self._getnodetxt(node)
|
|
node = bug_n.getElementsByTagName('product')[0]
|
|
summary['product'] = self._getnodetxt(node)
|
|
node = bug_n.getElementsByTagName('component')[0]
|
|
summary['component'] = self._getnodetxt(node)
|
|
node = bug_n.getElementsByTagName('bug_severity')[0]
|
|
summary['severity'] = self._getnodetxt(node)
|
|
except Exception, e:
|
|
zilladom.unlink()
|
|
msg = 'Could not parse XML returned by %s bugzilla: %s'
|
|
raise BugError(str(msg % (desc, e)))
|
|
zilladom.unlink()
|
|
return summary
|
|
|
|
def _getbugxml(self, url, desc):
|
|
try: fh = urllib.urlopen(url)
|
|
except: raise IOError('Connection to %s bugzilla failed' % desc)
|
|
bugxml = ''
|
|
while 1:
|
|
chunk = fh.read(8192)
|
|
if chunk == '':
|
|
break
|
|
bugxml = bugxml + chunk
|
|
fh.close()
|
|
if not len(bugxml):
|
|
msg = 'Error getting bug content from %s' % desc
|
|
raise IOError(msg)
|
|
return bugxml
|
|
|
|
def _getnodetxt(self, node):
|
|
val = ''
|
|
for childnode in node.childNodes:
|
|
if childnode.nodeType == childnode.TEXT_NODE:
|
|
val = val + childnode.data
|
|
if node.hasAttribute('encoding'):
|
|
encoding = node.getAttribute('encoding')
|
|
if encoding == 'base64':
|
|
try:
|
|
val = base64.decodestring(val)
|
|
except:
|
|
val = 'Cannot convert bug data from base64!'
|
|
entre = self.entre
|
|
while entre.search(val):
|
|
entity = entre.search(val).group(1)
|
|
if entities.has_key(entity):
|
|
val = re.sub(entre, entities[entity], val)
|
|
else:
|
|
val = re.sub(entre, '_', val)
|
|
return val
|
|
|
|
Class = Bugzilla
|
|
|
|
# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78:
|