Remove old hotfix from the Great PkgDB2 Migration of 2014.
This commit is contained in:
parent
8861007d2a
commit
676ea12ff3
|
@ -1,797 +0,0 @@
|
|||
# -*- coding: utf-8 -*-
|
||||
###
|
||||
# Copyright (c) 2007, Mike McGrath
|
||||
# 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.
|
||||
###
|
||||
|
||||
import arrow
|
||||
import sgmllib
|
||||
import htmlentitydefs
|
||||
import requests
|
||||
|
||||
import supybot.utils as utils
|
||||
import supybot.conf as conf
|
||||
import time
|
||||
from supybot.commands import *
|
||||
import supybot.plugins as plugins
|
||||
import supybot.ircutils as ircutils
|
||||
import supybot.callbacks as callbacks
|
||||
|
||||
from fedora.client import AppError
|
||||
from fedora.client import AuthError
|
||||
from fedora.client import ServerError
|
||||
from fedora.client.fas2 import AccountSystem
|
||||
from fedora.client.fas2 import FASError
|
||||
from pkgdb2client import PkgDB
|
||||
|
||||
from kitchen.text.converters import to_unicode
|
||||
|
||||
import fedmsg.config
|
||||
import fedmsg.meta
|
||||
|
||||
import simplejson
|
||||
import urllib
|
||||
import commands
|
||||
import urllib2
|
||||
import socket
|
||||
import pytz
|
||||
import datetime
|
||||
import threading
|
||||
|
||||
SPARKLINE_RESOLUTION = 50
|
||||
|
||||
datagrepper_url = 'https://apps.fedoraproject.org/datagrepper/raw'
|
||||
|
||||
|
||||
def datagrepper_query(kwargs):
|
||||
""" Return the count of msgs filtered by kwargs for a given time.
|
||||
|
||||
The arguments for this are a little clumsy; this is imposed on us by
|
||||
multiprocessing.Pool.
|
||||
"""
|
||||
start, end = kwargs.pop('start'), kwargs.pop('end')
|
||||
params = {
|
||||
'start': time.mktime(start.timetuple()),
|
||||
'end': time.mktime(end.timetuple()),
|
||||
}
|
||||
params.update(kwargs)
|
||||
|
||||
req = requests.get(datagrepper_url, params=params)
|
||||
json_out = simplejson.loads(req.text)
|
||||
result = int(json_out['total'])
|
||||
return result
|
||||
|
||||
|
||||
class WorkerThread(threading.Thread):
|
||||
""" A simple worker thread for our threadpool. """
|
||||
|
||||
def __init__(self, fn, item, *args, **kwargs):
|
||||
self.fn = fn
|
||||
self.item = item
|
||||
super(WorkerThread, self).__init__(*args, **kwargs)
|
||||
|
||||
def run(self):
|
||||
self.result = self.fn(self.item)
|
||||
|
||||
|
||||
class ThreadPool(object):
|
||||
""" Our very own threadpool implementation.
|
||||
|
||||
We make our own thing because multiprocessing is too heavy.
|
||||
"""
|
||||
|
||||
def map(self, fn, items):
|
||||
threads = []
|
||||
|
||||
for item in items:
|
||||
threads.append(WorkerThread(fn=fn, item=item))
|
||||
|
||||
for thread in threads:
|
||||
thread.start()
|
||||
|
||||
for thread in threads:
|
||||
thread.join()
|
||||
|
||||
return [thread.result for thread in threads]
|
||||
|
||||
|
||||
class Title(sgmllib.SGMLParser):
|
||||
entitydefs = htmlentitydefs.entitydefs.copy()
|
||||
entitydefs['nbsp'] = ' '
|
||||
|
||||
def __init__(self):
|
||||
self.inTitle = False
|
||||
self.title = ''
|
||||
sgmllib.SGMLParser.__init__(self)
|
||||
|
||||
def start_title(self, attrs):
|
||||
self.inTitle = True
|
||||
|
||||
def end_title(self):
|
||||
self.inTitle = False
|
||||
|
||||
def unknown_entityref(self, name):
|
||||
if self.inTitle:
|
||||
self.title += ' '
|
||||
|
||||
def unknown_charref(self, name):
|
||||
if self.inTitle:
|
||||
self.title += ' '
|
||||
|
||||
def handle_data(self, data):
|
||||
if self.inTitle:
|
||||
self.title += data
|
||||
|
||||
|
||||
class Fedora(callbacks.Plugin):
|
||||
"""Use this plugin to retrieve Fedora-related information."""
|
||||
threaded = True
|
||||
|
||||
def __init__(self, irc):
|
||||
super(Fedora, self).__init__(irc)
|
||||
|
||||
# caches, automatically downloaded on __init__, manually refreshed on
|
||||
# .refresh
|
||||
self.userlist = None
|
||||
self.bugzacl = None
|
||||
|
||||
# To get the information, we need a username and password to FAS.
|
||||
# DO NOT COMMIT YOUR USERNAME AND PASSWORD TO THE PUBLIC REPOSITORY!
|
||||
self.fasurl = self.registryValue('fas.url')
|
||||
self.username = self.registryValue('fas.username')
|
||||
self.password = self.registryValue('fas.password')
|
||||
|
||||
self.fasclient = AccountSystem(self.fasurl, username=self.username,
|
||||
password=self.password)
|
||||
self.pkgdb = PkgDB()
|
||||
# URLs
|
||||
#self.url = {}
|
||||
|
||||
# fetch necessary caches
|
||||
self._refresh()
|
||||
|
||||
# Pull in /etc/fedmsg.d/ so we can build the fedmsg.meta processors.
|
||||
fm_config = fedmsg.config.load_config()
|
||||
fedmsg.meta.make_processors(**fm_config)
|
||||
|
||||
def _refresh(self):
|
||||
timeout = socket.getdefaulttimeout()
|
||||
socket.setdefaulttimeout(None)
|
||||
self.log.info("Downloading user data")
|
||||
request = self.fasclient.send_request('/user/list',
|
||||
req_params={'search': '*'},
|
||||
auth=True,
|
||||
timeout=240)
|
||||
users = request['people'] + request['unapproved_people']
|
||||
del request
|
||||
self.log.info("Caching necessary user data")
|
||||
self.users = {}
|
||||
self.faslist = {}
|
||||
for user in users:
|
||||
name = user['username']
|
||||
self.users[name] = {}
|
||||
self.users[name]['id'] = user['id']
|
||||
key = ' '.join([user['username'], user['email'] or '',
|
||||
user['human_name'] or '', user['ircnick'] or ''])
|
||||
key = key.lower()
|
||||
value = "%s '%s' <%s>" % (user['username'], user['human_name'] or
|
||||
'', user['email'] or '')
|
||||
self.faslist[key] = value
|
||||
self.log.info("Downloading package owners cache")
|
||||
data = requests.get(
|
||||
'https://admin.fedoraproject.org/pkgdb/api/bugzilla?format=json',
|
||||
verify=True).json()
|
||||
self.bugzacl = data['bugzillaAcls']
|
||||
socket.setdefaulttimeout(timeout)
|
||||
|
||||
def refresh(self, irc, msg, args):
|
||||
"""takes no arguments
|
||||
|
||||
Refresh the necessary caches."""
|
||||
self._refresh()
|
||||
irc.replySuccess()
|
||||
refresh = wrap(refresh)
|
||||
|
||||
def _load_json(self, url):
|
||||
timeout = socket.getdefaulttimeout()
|
||||
socket.setdefaulttimeout(45)
|
||||
json = simplejson.loads(utils.web.getUrl(url))
|
||||
socket.setdefaulttimeout(timeout)
|
||||
return json
|
||||
|
||||
def whoowns(self, irc, msg, args, package):
|
||||
"""<package>
|
||||
|
||||
Retrieve the owner of a given package
|
||||
"""
|
||||
try:
|
||||
mainowner = self.bugzacl['Fedora'][package]['owner']
|
||||
except KeyError:
|
||||
irc.reply("No such package exists.")
|
||||
return
|
||||
others = []
|
||||
for key in self.bugzacl:
|
||||
if key == 'Fedora':
|
||||
continue
|
||||
try:
|
||||
owner = self.bugzacl[key][package]['owner']
|
||||
if owner == mainowner:
|
||||
continue
|
||||
except KeyError:
|
||||
continue
|
||||
others.append("%s in %s" % (owner, key))
|
||||
if others == []:
|
||||
irc.reply(mainowner)
|
||||
else:
|
||||
irc.reply("%s (%s)" % (mainowner, ', '.join(others)))
|
||||
whoowns = wrap(whoowns, ['text'])
|
||||
|
||||
def branches(self, irc, msg, args, package):
|
||||
"""<package>
|
||||
|
||||
Return the branches a package is in."""
|
||||
try:
|
||||
pkginfo = self.pkgdb.get_package(package)
|
||||
except AppError:
|
||||
irc.reply("No such package exists.")
|
||||
return
|
||||
branch_list = []
|
||||
for listing in pkginfo['packages']:
|
||||
branch_list.append(listing['collection']['branchname'])
|
||||
branch_list.sort()
|
||||
irc.reply(' '.join(branch_list))
|
||||
return
|
||||
branches = wrap(branches, ['text'])
|
||||
|
||||
def what(self, irc, msg, args, package):
|
||||
"""<package>
|
||||
|
||||
Returns a description of a given package.
|
||||
"""
|
||||
try:
|
||||
summary = self.bugzacl['Fedora'][package]['summary']
|
||||
irc.reply("%s: %s" % (package, summary))
|
||||
except KeyError:
|
||||
irc.reply("No such package exists.")
|
||||
return
|
||||
what = wrap(what, ['text'])
|
||||
|
||||
def fas(self, irc, msg, args, find_name):
|
||||
"""<query>
|
||||
|
||||
Search the Fedora Account System usernames, full names, and email
|
||||
addresses for a match."""
|
||||
find_name = to_unicode(find_name)
|
||||
matches = []
|
||||
for entry in self.faslist.keys():
|
||||
if entry.find(find_name.lower()) != -1:
|
||||
matches.append(entry)
|
||||
if len(matches) == 0:
|
||||
irc.reply("'%s' Not Found!" % find_name)
|
||||
else:
|
||||
output = []
|
||||
for match in matches:
|
||||
output.append(self.faslist[match])
|
||||
irc.reply(' - '.join(output).encode('utf-8'))
|
||||
fas = wrap(fas, ['text'])
|
||||
|
||||
def hellomynameis(self, irc, msg, args, name):
|
||||
"""<username>
|
||||
|
||||
Return brief information about a Fedora Account System username. Useful
|
||||
for things like meeting roll call and calling attention to yourself."""
|
||||
try:
|
||||
person = self.fasclient.person_by_username(name)
|
||||
except:
|
||||
irc.reply('Something blew up, please try again')
|
||||
return
|
||||
if not person:
|
||||
irc.reply('Sorry, but you don\'t exist')
|
||||
return
|
||||
irc.reply(('%(username)s \'%(human_name)s\' <%(email)s>' %
|
||||
person).encode('utf-8'))
|
||||
hellomynameis = wrap(hellomynameis, ['text'])
|
||||
|
||||
def himynameis(self, irc, msg, args, name):
|
||||
"""<username>
|
||||
|
||||
Will the real Slim Shady please stand up?"""
|
||||
try:
|
||||
person = self.fasclient.person_by_username(name)
|
||||
except:
|
||||
irc.reply('Something blew up, please try again')
|
||||
return
|
||||
if not person:
|
||||
irc.reply('Sorry, but you don\'t exist')
|
||||
return
|
||||
irc.reply(('%(username)s \'Slim Shady\' <%(email)s>' %
|
||||
person).encode('utf-8'))
|
||||
himynameis = wrap(himynameis, ['text'])
|
||||
|
||||
def localtime(self, irc, msg, args, name):
|
||||
"""<username>
|
||||
|
||||
Returns the current time of the user.
|
||||
The timezone is queried from FAS."""
|
||||
try:
|
||||
person = self.fasclient.person_by_username(name)
|
||||
except:
|
||||
irc.reply('Error getting info user user: "%s"' % name)
|
||||
return
|
||||
if not person:
|
||||
irc.reply('User "%s" doesn\'t exist' % name)
|
||||
return
|
||||
timezone_name = person['timezone']
|
||||
if timezone_name is None:
|
||||
irc.reply('User "%s" doesn\'t share his timezone' % name)
|
||||
return
|
||||
try:
|
||||
time = datetime.datetime.now(pytz.timezone(timezone_name))
|
||||
except:
|
||||
irc.reply('The timezone of "%s" was unknown: "%s"' % (name,
|
||||
timezone))
|
||||
return
|
||||
irc.reply('The current local time of "%s" is: "%s" (timezone: %s)' %
|
||||
(name, time.strftime('%H:%M'), timezone_name))
|
||||
localtime = wrap(localtime, ['text'])
|
||||
|
||||
def fasinfo(self, irc, msg, args, name):
|
||||
"""<username>
|
||||
|
||||
Return information on a Fedora Account System username."""
|
||||
try:
|
||||
person = self.fasclient.person_by_username(name)
|
||||
except:
|
||||
irc.reply('Error getting info for user: "%s"' % name)
|
||||
return
|
||||
if not person:
|
||||
irc.reply('User "%s" doesn\'t exist' % name)
|
||||
return
|
||||
person['creation'] = person['creation'].split(' ')[0]
|
||||
string = ("User: %(username)s, Name: %(human_name)s"
|
||||
", email: %(email)s, Creation: %(creation)s"
|
||||
", IRC Nick: %(ircnick)s, Timezone: %(timezone)s"
|
||||
", Locale: %(locale)s"
|
||||
", GPG key ID: %(gpg_keyid)s, Status: %(status)s") % person
|
||||
irc.reply(string.encode('utf-8'))
|
||||
|
||||
# List of unapproved groups is easy
|
||||
unapproved = ''
|
||||
for group in person['unapproved_memberships']:
|
||||
unapproved = unapproved + "%s " % group['name']
|
||||
if unapproved != '':
|
||||
irc.reply('Unapproved Groups: %s' % unapproved)
|
||||
|
||||
# List of approved groups requires a separate query to extract roles
|
||||
constraints = {'username': name, 'group': '%',
|
||||
'role_status': 'approved'}
|
||||
columns = ['username', 'group', 'role_type']
|
||||
roles = []
|
||||
try:
|
||||
roles = self.fasclient.people_query(constraints=constraints,
|
||||
columns=columns)
|
||||
except:
|
||||
irc.reply('Error getting group memberships.')
|
||||
return
|
||||
|
||||
approved = ''
|
||||
for role in roles:
|
||||
if role['role_type'] == 'sponsor':
|
||||
approved += '+' + role['group'] + ' '
|
||||
elif role['role_type'] == 'administrator':
|
||||
approved += '@' + role['group'] + ' '
|
||||
else:
|
||||
approved += role['group'] + ' '
|
||||
if approved == '':
|
||||
approved = "None"
|
||||
|
||||
irc.reply('Approved Groups: %s' % approved)
|
||||
fasinfo = wrap(fasinfo, ['text'])
|
||||
|
||||
def group(self, irc, msg, args, name):
|
||||
"""<group short name>
|
||||
|
||||
Return information about a Fedora Account System group."""
|
||||
try:
|
||||
group = self.fasclient.group_by_name(name)
|
||||
irc.reply('%s: %s' %
|
||||
(name, group['display_name']))
|
||||
except AppError:
|
||||
irc.reply('There is no group "%s".' % name)
|
||||
group = wrap(group, ['text'])
|
||||
|
||||
def admins(self, irc, msg, args, name):
|
||||
"""<group short name>
|
||||
|
||||
Return the administrators list for the selected group"""
|
||||
|
||||
try:
|
||||
group = self.fasclient.group_members(name)
|
||||
sponsors = ''
|
||||
for person in group:
|
||||
if person['role_type'] == 'administrator':
|
||||
sponsors += person['username'] + ' '
|
||||
irc.reply('Administrators for %s: %s' % (name, sponsors))
|
||||
except AppError:
|
||||
irc.reply('There is no group %s.' % name)
|
||||
|
||||
admins = wrap(admins, ['text'])
|
||||
|
||||
def sponsors(self, irc, msg, args, name):
|
||||
"""<group short name>
|
||||
|
||||
Return the sponsors list for the selected group"""
|
||||
|
||||
try:
|
||||
group = self.fasclient.group_members(name)
|
||||
sponsors = ''
|
||||
for person in group:
|
||||
if person['role_type'] == 'sponsor':
|
||||
sponsors += person['username'] + ' '
|
||||
elif person['role_type'] == 'administrator':
|
||||
sponsors += '@' + person['username'] + ' '
|
||||
irc.reply('Sponsors for %s: %s' % (name, sponsors))
|
||||
except AppError:
|
||||
irc.reply('There is no group %s.' % name)
|
||||
|
||||
sponsors = wrap(sponsors, ['text'])
|
||||
|
||||
def members(self, irc, msg, args, name):
|
||||
"""<group short name>
|
||||
|
||||
Return a list of members of the specified group"""
|
||||
try:
|
||||
group = self.fasclient.group_members(name)
|
||||
members = ''
|
||||
for person in group:
|
||||
if person['role_type'] == 'administrator':
|
||||
members += '@' + person['username'] + ' '
|
||||
elif person['role_type'] == 'sponsor':
|
||||
members += '+' + person['username'] + ' '
|
||||
else:
|
||||
members += person['username'] + ' '
|
||||
irc.reply('Members of %s: %s' % (name, members))
|
||||
except AppError:
|
||||
irc.reply('There is no group %s.' % name)
|
||||
|
||||
members = wrap(members, ['text'])
|
||||
|
||||
def showticket(self, irc, msg, args, baseurl, number):
|
||||
"""<baseurl> <number>
|
||||
|
||||
Return the name and URL of a trac ticket or bugzilla bug.
|
||||
"""
|
||||
url = format(baseurl, str(number))
|
||||
size = conf.supybot.protocols.http.peekSize()
|
||||
text = utils.web.getUrl(url, size=size)
|
||||
parser = Title()
|
||||
try:
|
||||
parser.feed(text)
|
||||
except sgmllib.SGMLParseError:
|
||||
irc.reply(format('Encountered a problem parsing %u', url))
|
||||
if parser.title:
|
||||
irc.reply(utils.web.htmlToText(parser.title.strip()) + ' - ' + url)
|
||||
else:
|
||||
irc.reply(format('That URL appears to have no HTML title ' +
|
||||
'within the first %i bytes.', size))
|
||||
showticket = wrap(showticket, ['httpUrl', 'int'])
|
||||
|
||||
def swedish(self, irc, msg, args):
|
||||
"""takes no arguments
|
||||
|
||||
Humor mmcgrath."""
|
||||
|
||||
# Import this here to avoid a circular import problem.
|
||||
from __init__ import __version__
|
||||
|
||||
irc.reply(str('kwack kwack'))
|
||||
irc.reply(str('bork bork bork'))
|
||||
irc.reply(str('(supybot-fedora version %s)' % __version__))
|
||||
swedish = wrap(swedish)
|
||||
|
||||
def wikilink(self, irc, msg, args, name):
|
||||
"""<username>
|
||||
|
||||
Return MediaWiki link syntax for a FAS user's page on the wiki."""
|
||||
try:
|
||||
person = self.fasclient.person_by_username(name)
|
||||
except:
|
||||
irc.reply('Error getting info for user: "%s"' % name)
|
||||
return
|
||||
if not person:
|
||||
irc.reply('User "%s" doesn\'t exist' % name)
|
||||
return
|
||||
string = "[[User:%s|%s]]" % (person["username"],
|
||||
person["human_name"] or '')
|
||||
irc.reply(string.encode('utf-8'))
|
||||
wikilink = wrap(wikilink, ['text'])
|
||||
|
||||
def mirroradmins(self, irc, msg, args, hostname):
|
||||
"""<hostname>
|
||||
|
||||
Return MirrorManager list of FAS usernames which administer <hostname>.
|
||||
<hostname> must be the FQDN of the host."""
|
||||
url = ("https://admin.fedoraproject.org/mirrormanager/mirroradmins?"
|
||||
"tg_format=json&host=" + hostname)
|
||||
result = self._load_json(url)['values']
|
||||
if len(result) == 0:
|
||||
irc.reply('Hostname "%s" not found' % hostname)
|
||||
return
|
||||
string = 'Mirror Admins of %s: ' % hostname
|
||||
string += ' '.join(result)
|
||||
irc.reply(string.encode('utf-8'))
|
||||
mirroradmins = wrap(mirroradmins, ['text'])
|
||||
|
||||
def nextmeeting(self, irc, msg, args, channel):
|
||||
"""<channel>
|
||||
|
||||
Return the next meeting scheduled for a particular channel.
|
||||
"""
|
||||
|
||||
channel = channel.strip('#').split('@')[0]
|
||||
meetings = list(self._future_meetings(channel))
|
||||
if not meetings:
|
||||
response = "There are no meetings scheduled for #%s." % channel
|
||||
irc.reply(response.encode('utf-8'))
|
||||
return
|
||||
|
||||
date, meeting = meetings[0]
|
||||
response = "The next meeting in #%s is %s (starting %s)" % (
|
||||
channel,
|
||||
meeting['meeting_name'],
|
||||
arrow.get(date).humanize(),
|
||||
)
|
||||
irc.reply(response.encode('utf-8'))
|
||||
base = "https://apps.fedoraproject.org/calendar/location/"
|
||||
url = base + urllib.quote("%s@irc.freenode.net/" % channel)
|
||||
irc.reply("- " + url.encode('utf-8'))
|
||||
nextmeeting = wrap(nextmeeting, ['text'])
|
||||
|
||||
@staticmethod
|
||||
def _future_meetings(channel):
|
||||
response = requests.get(
|
||||
'https://apps.fedoraproject.org/calendar/api/meetings',
|
||||
params=dict(
|
||||
location='%s@irc.freenode.net' % channel,
|
||||
)
|
||||
)
|
||||
|
||||
data = response.json()
|
||||
now = datetime.datetime.utcnow()
|
||||
|
||||
for meeting in data['meetings']:
|
||||
string = meeting['meeting_date'] + " " + meeting['meeting_time_start']
|
||||
dt = datetime.datetime.strptime(string, "%Y-%m-%d %H:%M:%S")
|
||||
|
||||
if now < dt:
|
||||
yield dt, meeting
|
||||
|
||||
def badges(self, irc, msg, args, name):
|
||||
"""<username>
|
||||
|
||||
Return badges statistics about a user.
|
||||
"""
|
||||
url = "https://badges.fedoraproject.org/user/" + name
|
||||
d = requests.get(url + "/json").json()
|
||||
|
||||
if 'error' in d:
|
||||
response = d['error']
|
||||
else:
|
||||
template = "{name} has unlocked {n} Fedora Badges: {url}"
|
||||
n = len(d['assertions'])
|
||||
response = template.format(name=name, url=url, n=n)
|
||||
|
||||
irc.reply(response.encode('utf-8'))
|
||||
badges = wrap(badges, ['text'])
|
||||
|
||||
def quote(self, irc, msg, args, arguments):
|
||||
"""<SYMBOL> [daily, weekly, monthly, quarterly]
|
||||
|
||||
Return some datagrepper statistics on fedmsg categories.
|
||||
"""
|
||||
|
||||
# First, some argument parsing. Supybot should be able to do this for
|
||||
# us, but I couldn't figure it out. The supybot.plugins.additional
|
||||
# object is the thing to use... except its weird.
|
||||
tokens = arguments.split(None, 1)
|
||||
if len(tokens) == 1:
|
||||
symbol, frame = tokens[0], 'daily'
|
||||
else:
|
||||
symbol, frame = tokens
|
||||
|
||||
# Second, build a lookup table for symbols. By default, we'll use the
|
||||
# fedmsg category names, take their first 3 characters and uppercase
|
||||
# them. That will take things like "wiki" and turn them into "WIK" and
|
||||
# "bodhi" and turn them into "BOD". This handles a lot for us. We'll
|
||||
# then override those that don't make sense manually here. For
|
||||
# instance "fedoratagger" by default would be "FED", but that's no
|
||||
# good. We want "TAG".
|
||||
# Why all this trouble? Well, as new things get added to the fedmsg
|
||||
# bus, we don't want to have keep coming back here and modifying this
|
||||
# code. Hopefully this dance will at least partially future-proof us.
|
||||
symbols = dict([
|
||||
(processor.__name__.lower(), processor.__name__[:3].upper())
|
||||
for processor in fedmsg.meta.processors
|
||||
])
|
||||
symbols.update({
|
||||
'fedoratagger': 'TAG',
|
||||
'fedbadges': 'BDG',
|
||||
'buildsys': 'KOJ',
|
||||
'pkgdb': 'PKG',
|
||||
'meetbot': 'MTB',
|
||||
'planet': 'PLN',
|
||||
'trac': 'TRC',
|
||||
'mailman': 'MM3',
|
||||
})
|
||||
|
||||
# Now invert the dict so we can lookup the argued symbol.
|
||||
# Yes, this is vulnerable to collisions.
|
||||
symbols = dict([(sym, name) for name, sym in symbols.items()])
|
||||
|
||||
# These aren't user-facing topics, so drop 'em.
|
||||
del symbols['LOG']
|
||||
del symbols['UNH']
|
||||
del symbols['ANN'] # And this one is unused...
|
||||
|
||||
key_fmt = lambda d: ', '.join(sorted(d.keys()))
|
||||
|
||||
if not symbol in symbols:
|
||||
response = "No such symbol %r. Try one of %s"
|
||||
irc.reply((response % (symbol, key_fmt(symbols))).encode('utf-8'))
|
||||
return
|
||||
|
||||
# Now, build another lookup of our various timeframes.
|
||||
frames = dict(
|
||||
daily=datetime.timedelta(days=1),
|
||||
weekly=datetime.timedelta(days=7),
|
||||
monthly=datetime.timedelta(days=30),
|
||||
quarterly=datetime.timedelta(days=91),
|
||||
)
|
||||
|
||||
if not frame in frames:
|
||||
response = "No such timeframe %r. Try one of %s"
|
||||
irc.reply((response % (frame, key_fmt(frames))).encode('utf-8'))
|
||||
return
|
||||
|
||||
category = [symbols[symbol]]
|
||||
|
||||
t2 = datetime.datetime.now()
|
||||
t1 = t2 - frames[frame]
|
||||
t0 = t1 - frames[frame]
|
||||
|
||||
# Count the number of messages between t0 and t1, and between t1 and t2
|
||||
query1 = dict(start=t0, end=t1, category=category)
|
||||
query2 = dict(start=t1, end=t2, category=category)
|
||||
|
||||
# Do this async for superfast datagrepper queries.
|
||||
tpool = ThreadPool()
|
||||
batched_values = tpool.map(datagrepper_query, [
|
||||
dict(start=x, end=y, category=category)
|
||||
for x, y in Utils.daterange(t1, t2, SPARKLINE_RESOLUTION)
|
||||
] + [query1, query2])
|
||||
|
||||
count2 = batched_values.pop()
|
||||
count1 = batched_values.pop()
|
||||
|
||||
# Just rename the results. We'll use the rest for the sparkline.
|
||||
sparkline_values = batched_values
|
||||
|
||||
yester_phrases = dict(
|
||||
daily="yesterday",
|
||||
weekly="the week preceding this one",
|
||||
monthly="the month preceding this one",
|
||||
quarterly="the 3 months preceding these past three months",
|
||||
)
|
||||
phrases = dict(
|
||||
daily="24 hours",
|
||||
weekly="week",
|
||||
monthly="month",
|
||||
quarterly="3 months",
|
||||
)
|
||||
|
||||
if count1 and count2:
|
||||
percent = ((float(count2) / count1) - 1) * 100
|
||||
elif not count1 and count2:
|
||||
# If the older of the two time periods had zero messages, but there
|
||||
# are some in the more current period.. well, that's an infinite
|
||||
# percent increase.
|
||||
percent = float('inf')
|
||||
elif not count1 and not count2:
|
||||
# If counts are zero for both periods, then the change is 0%.
|
||||
percent = 0
|
||||
else:
|
||||
# Else, if there were some messages in the old time period, but
|
||||
# none in the current... then that's a 100% drop off.
|
||||
percent = -100
|
||||
|
||||
sign = lambda value: value >= 0 and '+' or '-'
|
||||
|
||||
template = u"{sym}, {name} {sign}{percent:.2f}% over {phrase}"
|
||||
response = template.format(
|
||||
sym=symbol,
|
||||
name=symbols[symbol],
|
||||
sign=sign(percent),
|
||||
percent=abs(percent),
|
||||
phrase=yester_phrases[frame],
|
||||
)
|
||||
irc.reply(response.encode('utf-8'))
|
||||
|
||||
# Now, make a graph out of it.
|
||||
sparkline = Utils.sparkline(sparkline_values)
|
||||
|
||||
template = u" {sparkline} ⤆ over {phrase}"
|
||||
response = template.format(
|
||||
sym=symbol,
|
||||
sparkline=sparkline,
|
||||
phrase=phrases[frame]
|
||||
)
|
||||
irc.reply(response.encode('utf-8'))
|
||||
|
||||
to_utc = lambda t: time.gmtime(time.mktime(t.timetuple()))
|
||||
# And a final line for "x-axis tics"
|
||||
t1_fmt = time.strftime("%H:%M UTC %m/%d", to_utc(t1))
|
||||
t2_fmt = time.strftime("%H:%M UTC %m/%d", to_utc(t2))
|
||||
padding = u" " * (SPARKLINE_RESOLUTION - len(t1_fmt) - 3)
|
||||
template = u" ↑ {t1}{padding}↑ {t2}"
|
||||
response = template.format(t1=t1_fmt, t2=t2_fmt, padding=padding)
|
||||
irc.reply(response.encode('utf-8'))
|
||||
quote = wrap(quote, ['text'])
|
||||
|
||||
|
||||
class Utils(object):
|
||||
""" Some handy utils for datagrepper visualization. """
|
||||
|
||||
@classmethod
|
||||
def sparkline(cls, values):
|
||||
bar = u'▁▂▃▄▅▆▇█'
|
||||
barcount = len(bar) - 1
|
||||
values = map(float, values)
|
||||
mn, mx = min(values), max(values)
|
||||
extent = mx - mn
|
||||
|
||||
if extent == 0:
|
||||
indices = [0 for n in values]
|
||||
else:
|
||||
indices = [int((n - mn) / extent * barcount) for n in values]
|
||||
|
||||
unicode_sparkline = u''.join([bar[i] for i in indices])
|
||||
return unicode_sparkline
|
||||
|
||||
@classmethod
|
||||
def daterange(cls, start, stop, steps):
|
||||
""" A generator for stepping through time. """
|
||||
delta = (stop - start) / steps
|
||||
current = start
|
||||
while current + delta <= stop:
|
||||
yield current, current + delta
|
||||
current += delta
|
||||
|
||||
|
||||
Class = Fedora
|
||||
|
||||
|
||||
# vim:set shiftwidth=4 tabstop=4 expandtab textwidth=79:
|
|
@ -38,12 +38,6 @@
|
|||
- name: teams cron job
|
||||
cron: name=meetings-by-team hour="23" minute="0" user=daemon job="/usr/local/bin/meetings_by_team.sh"
|
||||
|
||||
- name: hotfix - packagedb-cli which is a new dep but is not there in the rpm
|
||||
yum: pkg=packagedb-cli state=present
|
||||
|
||||
- name: hotfix - supybot plugin
|
||||
copy: src=plugin.py dest=/usr/lib/python2.6/site-packages/supybot/plugins/Fedora/plugin.py mode=755 owner=root
|
||||
|
||||
- name: setup meetbot.conf apache config
|
||||
copy: src=meetbot.conf dest=/etc/httpd/conf.d/meetbot.conf mode=644
|
||||
|
||||
|
|
Loading…
Reference in New Issue