Limnoria/plugins/ChannelStats.py

349 lines
13 KiB
Python
Raw Normal View History

2003-03-12 07:26:59 +01:00
###
# Copyright (c) 2002-2004, Jeremiah Fincher
2003-03-12 07:26:59 +01:00
# 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.
###
"""
Silently listens to every message received on a channel and keeps statistics
concerning joins, parts, and various other commands in addition to tracking
statistics about smileys, actions, characters, and words.
"""
2003-11-25 09:23:47 +01:00
__revision__ = "$Id$"
2004-07-24 07:18:26 +02:00
import supybot.plugins as plugins
2003-03-12 07:26:59 +01:00
import os
2003-03-12 07:26:59 +01:00
import re
2003-08-12 21:38:23 +02:00
import sets
2003-03-12 07:26:59 +01:00
import time
import getopt
import string
from itertools import imap, ifilter
2003-03-12 07:26:59 +01:00
2004-07-24 07:18:26 +02:00
import supybot.log as log
import supybot.conf as conf
import supybot.utils as utils
import supybot.world as world
import supybot.ircdb as ircdb
2004-12-04 19:48:57 +01:00
from supybot.commands import *
2004-07-24 07:18:26 +02:00
import supybot.irclib as irclib
import supybot.ircmsgs as ircmsgs
import supybot.plugins as plugins
import supybot.ircutils as ircutils
import supybot.privmsgs as privmsgs
import supybot.registry as registry
import supybot.callbacks as callbacks
2003-03-12 07:26:59 +01:00
class Smileys(registry.Value):
def set(self, s):
L = s.split()
self.setValue(L)
2004-01-08 01:48:30 +01:00
def setValue(self, v):
self.s = ' '.join(v)
self.value = re.compile('|'.join(imap(re.escape, v)))
2004-01-08 01:48:30 +01:00
def __str__(self):
return self.s
conf.registerPlugin('ChannelStats')
conf.registerChannelValue(conf.supybot.plugins.ChannelStats, 'selfStats',
registry.Boolean(True, """Determines whether the bot will keep channel
statistics on itself, possibly skewing the channel stats (especially in
cases where the bot is relaying between channels on a network)."""))
conf.registerChannelValue(conf.supybot.plugins.ChannelStats, 'smileys',
Smileys(':) ;) ;] :-) :-D :D :P :p (= =)'.split(), """Determines what
words (i.e., pieces of text with no spaces in them) are considered
'smileys' for the purposes of stats-keeping."""))
conf.registerChannelValue(conf.supybot.plugins.ChannelStats, 'frowns',
Smileys(':| :-/ :-\\ :\\ :/ :( :-( :\'('.split(), """Determines what words
(i.e., pieces of text with no spaces in them ) are considered 'frowns' for
the purposes of stats-keeping."""))
2004-02-03 06:29:40 +01:00
class ChannelStat(irclib.IrcCommandDispatcher):
def __init__(self, actions=0, chars=0, frowns=0, joins=0, kicks=0, modes=0,
msgs=0, parts=0, quits=0, smileys=0, topics=0, words=0):
self.actions = actions
self.chars = chars
self.frowns = frowns
self.joins = joins
self.kicks = kicks
self.modes = modes
self.msgs = msgs
self.parts = parts
self.quits = quits
self.smileys = smileys
self.topics = topics
self.words = words
self._values = ['actions', 'chars', 'frowns', 'joins', 'kicks','modes',
'msgs', 'parts', 'quits', 'smileys', 'topics', 'words']
def values(self):
2004-04-20 11:44:58 +02:00
return map(utils.curry(getattr, self), self._values)
2004-02-03 06:29:40 +01:00
def addMsg(self, msg):
self.msgs += 1
method = self.dispatchCommand(msg.command)
if method is not None:
method(msg)
def doPayload(self, channel, payload):
self.chars += len(payload)
self.words += len(payload.split())
fRe = conf.supybot.plugins.ChannelStats.get('frowns').get(channel)()
sRe =conf.supybot.plugins.ChannelStats.get('smileys').get(channel)()
self.frowns += len(fRe.findall(payload))
self.smileys += len(sRe.findall(payload))
def doPrivmsg(self, msg):
self.doPayload(*msg.args)
if ircmsgs.isAction(msg):
self.actions += 1
def doTopic(self, msg):
self.doPayload(*msg.args)
self.topics += 1
def doKick(self, msg):
self.kicks += 1
def doPart(self, msg):
2004-02-03 23:59:09 +01:00
if len(msg.args) == 2:
self.doPayload(*msg.args)
2004-02-03 06:29:40 +01:00
self.parts += 1
def doJoin(self, msg):
if len(msg.args) == 2:
self.doPayload(*msg.args)
self.joins += 1
2004-02-03 06:29:40 +01:00
def doMode(self, msg):
self.modes += 1
# doQuit is handled by the plugin.
2004-02-03 06:29:40 +01:00
class UserStat(ChannelStat):
2004-02-08 11:03:27 +01:00
def __init__(self, kicked=0, *args):
2004-02-03 06:29:40 +01:00
ChannelStat.__init__(self, *args)
self.kicked = kicked
self._values.insert(0, 'kicked')
def doKick(self, msg):
self.doPayload(msg.args[0], msg.args[2])
self.kicks += 1
2004-02-08 07:48:07 +01:00
class StatsDB(plugins.ChannelUserDB):
def __init__(self, *args, **kwargs):
plugins.ChannelUserDB.__init__(self, *args, **kwargs)
2004-07-21 21:36:35 +02:00
2004-02-08 07:48:07 +01:00
def serialize(self, v):
return v.values()
def deserialize(self, channel, id, L):
L = map(int, L)
if id == 'channelStats':
2004-02-08 11:03:27 +01:00
return ChannelStat(*L)
2004-02-08 07:48:07 +01:00
else:
2004-02-08 11:03:27 +01:00
return UserStat(*L)
2004-02-08 07:48:07 +01:00
2004-02-03 06:29:40 +01:00
def addMsg(self, msg, id=None):
channel = msg.args[0]
if ircutils.isChannel(channel):
2004-02-08 11:03:27 +01:00
if (channel, 'channelStats') not in self:
self[channel, 'channelStats'] = ChannelStat()
self[channel, 'channelStats'].addMsg(msg)
2004-02-03 06:29:40 +01:00
try:
if id is None:
id = ircdb.users.getUserId(msg.prefix)
except KeyError:
return
2004-02-08 11:03:27 +01:00
if (channel, id) not in self:
self[channel, id] = UserStat()
self[channel, id].addMsg(msg)
2004-02-03 06:29:40 +01:00
2004-02-08 07:48:07 +01:00
def getChannelStats(self, channel):
2004-02-08 11:03:27 +01:00
return self[channel, 'channelStats']
2004-07-21 21:36:35 +02:00
2004-02-08 07:48:07 +01:00
def getUserStats(self, channel, id):
return self[channel, id]
2004-02-03 06:29:40 +01:00
class ChannelStats(callbacks.Privmsg):
2003-11-11 00:39:44 +01:00
noIgnore = True
2003-03-12 07:26:59 +01:00
def __init__(self):
2003-11-04 09:42:11 +01:00
callbacks.Privmsg.__init__(self)
2003-10-16 15:34:02 +02:00
self.lastmsg = None
self.laststate = None
2003-10-25 00:29:30 +02:00
self.outFiltering = False
2004-02-03 06:29:40 +01:00
self.db = StatsDB(os.path.join(conf.supybot.directories.data(),
'ChannelStats.db'))
world.flushers.append(self.db.flush)
2003-03-12 07:26:59 +01:00
2003-11-11 16:58:20 +01:00
def die(self):
if self.db.flush in world.flushers:
world.flushers.remove(self.db.flush)
else:
self.log.debug('Odd, no flush in flushers: %r', world.flushers)
2004-02-03 06:29:40 +01:00
self.db.close()
2003-11-11 16:58:20 +01:00
callbacks.Privmsg.die(self)
2003-08-20 18:26:23 +02:00
2003-10-16 15:34:02 +02:00
def __call__(self, irc, msg):
try:
if self.lastmsg:
self.laststate.addMsg(irc, self.lastmsg)
else:
self.laststate = irc.state.copy()
finally:
self.lastmsg = msg
2004-02-03 06:29:40 +01:00
self.db.addMsg(msg)
super(ChannelStats, self).__call__(irc, msg)
2004-07-21 21:36:35 +02:00
2003-10-25 00:29:30 +02:00
def outFilter(self, irc, msg):
if msg.command == 'PRIVMSG':
if ircutils.isChannel(msg.args[0]):
if self.registryValue('selfStats', msg.args[0]):
try:
self.outFiltering = True
2004-02-03 06:29:40 +01:00
self.db.addMsg(msg, 0)
finally:
self.outFiltering = False
2003-10-25 00:29:30 +02:00
return msg
2003-03-12 07:26:59 +01:00
2004-02-03 06:29:40 +01:00
def doQuit(self, irc, msg):
try:
id = ircdb.users.getUserId(msg.prefix)
except KeyError:
2004-02-03 06:29:40 +01:00
id = None
for (channel, c) in self.laststate.channels.iteritems():
if msg.nick in c.users:
2004-02-08 11:03:27 +01:00
if (channel, 'channelStats') not in self.db:
self.db[channel, 'channelStats'] = ChannelStat()
self.db[channel, 'channelStats'].quits += 1
2004-02-03 06:29:40 +01:00
if id is not None:
2004-02-08 11:03:27 +01:00
if (channel, id) not in self.db:
self.db[channel, id] = UserStat()
self.db[channel, id].quits += 1
2003-03-26 11:04:26 +01:00
def doKick(self, irc, msg):
2004-02-03 06:29:40 +01:00
(channel, nick, _) = msg.args
hostmask = irc.state.nickToHostmask(nick)
2003-03-26 11:04:26 +01:00
try:
2004-02-03 06:29:40 +01:00
id = ircdb.users.getUserId(hostmask)
2003-03-26 11:04:26 +01:00
except KeyError:
2004-02-03 06:29:40 +01:00
return
if channel not in self.db.channels:
self.db.channels[channel] = {}
if id not in self.db.channels[channel]:
2004-02-08 11:03:27 +01:00
self.db.channels[channel][id] = UserStat()
2004-02-03 06:29:40 +01:00
self.db.channels[channel][id].kicked += 1
2003-10-16 15:34:02 +02:00
2004-12-04 19:48:57 +01:00
def stats(self, irc, msg, args, channel, name):
"""[<channel>] [<name>]
Returns the statistics for <name> on <channel>. <channel> is only
necessary if the message isn't sent on the channel itself. If <name>
isn't given, it defaults to the user sending the command.
"""
2004-12-04 19:48:57 +01:00
if name and ircutils.strEqual(name, irc.nick):
2004-01-31 22:03:31 +01:00
id = 0
elif not name:
try:
id = ircdb.users.getUserId(msg.prefix)
2003-10-25 00:29:30 +02:00
name = ircdb.users.getUser(id).name
except KeyError:
irc.error('I couldn\'t find you in my user database.')
return
elif not ircdb.users.hasUser(name):
2003-03-26 11:04:26 +01:00
try:
2003-10-02 02:43:38 +02:00
hostmask = irc.state.nickToHostmask(name)
id = ircdb.users.getUserId(hostmask)
2003-03-26 11:04:26 +01:00
except KeyError:
2004-01-08 16:47:38 +01:00
irc.errorNoUser()
2003-03-26 11:04:26 +01:00
return
else:
id = ircdb.users.getUserId(name)
2004-02-03 06:29:40 +01:00
try:
2004-02-08 07:48:07 +01:00
stats = self.db.getUserStats(channel, id)
2004-02-03 06:29:40 +01:00
s = '%s has sent %s; a total of %s, %s, ' \
'%s, and %s; %s of those messages %s' \
'%s has joined %s, parted %s, quit %s, kicked someone %s, ' \
'been kicked %s, changed the topic %s, ' \
'and changed the mode %s.' % \
(name, utils.nItems('message', stats.msgs),
utils.nItems('character', stats.chars),
utils.nItems('word', stats.words),
utils.nItems('smiley', stats.smileys),
utils.nItems('frown', stats.frowns),
stats.actions, stats.actions == 1 and 'was an ACTION. '
or 'were ACTIONs. ',
name,
utils.nItems('time', stats.joins),
utils.nItems('time', stats.parts),
utils.nItems('time', stats.quits),
utils.nItems('time', stats.kicks),
utils.nItems('time', stats.kicked),
utils.nItems('time', stats.topics),
utils.nItems('time', stats.modes))
irc.reply(s)
except KeyError:
irc.error('I have no stats for that %s in %s' % (name, channel))
2004-12-04 19:48:57 +01:00
stats = wrap(stats, ['channeldb', additional('something')])
2003-03-12 07:26:59 +01:00
2004-12-04 19:48:57 +01:00
def channelstats(self, irc, msg, args, channel):
"""[<channel>]
Returns the statistics for <channel>. <channel> is only necessary if
the message isn't sent on the channel itself.
"""
2004-02-03 06:29:40 +01:00
try:
2004-02-08 07:48:07 +01:00
stats = self.db.getChannelStats(channel)
2004-02-03 06:29:40 +01:00
s = 'On %s there have been %s messages, containing %s ' \
'characters, %s, %s, and %s; ' \
'%s of those messages %s. There have been ' \
'%s, %s, %s, %s, %s, and %s.' % \
(channel, stats.msgs, stats.chars,
utils.nItems('word', stats.words),
utils.nItems('smiley', stats.smileys),
utils.nItems('frown', stats.frowns),
stats.actions, stats.actions == 1 and 'was an ACTION'
or 'were ACTIONs',
utils.nItems('join', stats.joins),
utils.nItems('part', stats.parts),
utils.nItems('quit', stats.quits),
utils.nItems('kick', stats.kicks),
utils.nItems('change', stats.modes, between='mode'),
utils.nItems('change', stats.topics, between='topic'))
irc.reply(s)
except KeyError:
irc.error('I\'ve never been on %s.' % channel)
2004-12-04 19:48:57 +01:00
channelstats = wrap(channelstats, ['channeldb'])
2004-07-21 21:36:35 +02:00
Class = ChannelStats
2003-03-26 11:04:26 +01:00
# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78: