Limnoria/plugins/Status.py

251 lines
9.0 KiB
Python
Raw Normal View History

#!/usr/bin/env python
2003-09-30 12:47:05 +02:00
###
# Copyright (c) 2002, Jeremiah Fincher
# 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.
###
"""
A simple module to handle various informational commands querying the bot's
current status and statistics.
"""
2003-11-25 09:23:47 +01:00
__revision__ = "$Id$"
import plugins
2003-09-30 12:47:05 +02:00
import os
import sys
import sets
import time
import threading
from itertools import islice, ifilter, imap
2003-09-30 12:47:05 +02:00
import conf
2003-09-30 12:47:05 +02:00
import utils
import world
import privmsgs
import callbacks
def configure(onStart, afterConnect, advanced):
# This will be called by setup.py to configure this module. onStart and
# afterConnect are both lists. Append to onStart the commands you would
# like to be run when the bot is started; append to afterConnect the
# commands you would like to be run when the bot has finished connecting.
from questions import expect, anything, something, yn
onStart.append('load Status')
class UptimeDB(object):
def __init__(self, filename='uptimes'):
self.filename = os.path.join(conf.dataDir, filename)
if os.path.exists(self.filename):
fd = file(self.filename)
s = fd.read()
fd.close()
s = s.replace('\n', ' ')
self.uptimes = eval(s)
else:
self.uptimes = []
def die(self):
fd = file(self.filename, 'w')
2003-10-15 08:25:32 +02:00
fd.write(repr(self.top(50)))
fd.write('\n')
fd.close()
def add(self):
2003-11-23 14:01:19 +01:00
if world.startedAt != 0 and \
not any(lambda t: t[0] == world.startedAt, self.uptimes):
self.uptimes.append((world.startedAt, None))
def top(self, n=3):
def decorator(t):
2003-11-25 10:45:01 +01:00
return t[1] - t[0]
def invertCmp(cmp):
def f(x, y):
return -cmp(x, y)
return f
def notNone(t):
2003-11-25 10:45:01 +01:00
return t[1] is not None and t[0] != 0
utils.sortBy(decorator, self.uptimes, cmp=invertCmp(cmp))
2003-11-23 14:01:19 +01:00
return list(islice(ifilter(notNone, self.uptimes), n))
def update(self):
for (i, t) in enumerate(self.uptimes):
if t[0] == world.startedAt:
self.uptimes[i] = (t[0], time.time())
2003-09-30 12:47:05 +02:00
class Status(callbacks.Privmsg):
def __init__(self):
callbacks.Privmsg.__init__(self)
self.sentMsgs = 0
self.recvdMsgs = 0
self.sentBytes = 0
self.recvdBytes = 0
self.uptimes = UptimeDB()
self.uptimes.add()
self.uptimes.update()
2003-09-30 12:47:05 +02:00
def inFilter(self, irc, msg):
self.uptimes.update()
2003-09-30 12:47:05 +02:00
self.recvdMsgs += 1
self.recvdBytes += len(msg)
2003-09-30 12:47:05 +02:00
return msg
def outFilter(self, irc, msg):
self.sentMsgs += 1
self.sentBytes += len(msg)
2003-09-30 12:47:05 +02:00
return msg
def die(self):
self.uptimes.update()
self.uptimes.die()
def bestuptime(self, irc, msg, args):
"""takes no arguments
Returns the highest uptimes attained by the bot.
"""
L = self.uptimes.top()
if not L:
irc.error(msg, 'I don\'t have enough data to answer that.')
return
def format((started, ended)):
return '%s until %s; up for %s' % \
(time.strftime(conf.humanTimestampFormat,
time.localtime(started)),
time.strftime(conf.humanTimestampFormat,
time.localtime(ended)),
utils.timeElapsed(ended-started))
irc.reply(msg, utils.commaAndify(imap(format, L)))
def net(self, irc, msg, args):
2003-09-30 12:47:05 +02:00
"""takes no arguments
Returns some interesting network-related statistics.
"""
irc.reply(msg,
'I have received %s messages for a total of %s bytes. '\
'I have sent %s messages for a total of %s bytes.' %\
(self.recvdMsgs, self.recvdBytes,
self.sentMsgs, self.sentBytes))
def cpu(self, irc, msg, args):
2003-09-30 12:47:05 +02:00
"""takes no arguments
Returns some interesting CPU-related statistics on the bot.
"""
(user, system, childUser, childSystem, elapsed) = os.times()
now = time.time()
timeRunning = now - world.startedAt
if user+system < timeRunning+1: # Fudge for FPU inaccuracies.
children = 'My children have taken %.2f seconds of user time ' \
'and %.2f seconds of system time ' \
'for a total of %.2f seconds of CPU time. ' % \
(childUser, childSystem, childUser+childSystem)
else:
children = ''
2003-09-30 12:47:05 +02:00
activeThreads = threading.activeCount()
response = ('I have taken %.2f seconds of user time and %.2f seconds '
'of system time, for a total of %.2f seconds of CPU '
'time. %sOut of %s I have %s active.' %
(user, system, user + system, children,
utils.nItems('thread', world.threadsSpawned, 'spawned'),
activeThreads))
mem = 'an unknown amount'
pid = os.getpid()
plat = sys.platform
try:
if plat.startswith('linux') or plat.startswith('sunos') or \
plat.startswith('freebsd') or plat.startswith('openbsd') or \
plat.startswith('darwin'):
try:
r = os.popen('ps -o rss -p %s' % pid)
r.readline() # VSZ Header.
mem = r.readline().strip() + ' kB'
finally:
r.close()
elif sys.platform.startswith('netbsd'):
mem = '%s kB' % os.stat('/proc/%s/mem')[7]
response += ' I\'m taking up %s kB of memory.' % mem
except Exception:
self.log.exception('Uncaught exception in cpu:')
2003-09-30 12:47:05 +02:00
irc.reply(msg, response)
def cmd(self, irc, msg, args):
2003-09-30 13:04:44 +02:00
"""takes no arguments
Returns some interesting command-related statistics.
"""
commands = 0
callbacksPrivmsg = 0
2003-09-30 13:04:44 +02:00
for cb in irc.callbacks:
if isinstance(cb, callbacks.Privmsg) and cb.public:
if not isinstance(cb, callbacks.PrivmsgRegexp):
callbacksPrivmsg += 1
for attr in dir(cb):
if cb.isCommand(attr) and \
attr == callbacks.canonicalName(attr):
commands += 1
s = 'I offer a total of %s in %s. I have processed %s.' % \
(utils.nItems('command', commands),
utils.nItems('plugin', callbacksPrivmsg, 'command-based'),
utils.nItems('command', world.commandsProcessed))
irc.reply(msg, s)
def commands(self, irc, msg, args):
"""takes no arguments
Returns a list of the commands offered by the bot.
"""
commands = sets.Set()
for cb in irc.callbacks:
if isinstance(cb, callbacks.Privmsg) and \
not isinstance(cb, callbacks.PrivmsgRegexp) and cb.public:
2003-09-30 13:04:44 +02:00
for attr in dir(cb):
if cb.isCommand(attr) and \
attr == callbacks.canonicalName(attr):
2003-09-30 13:04:44 +02:00
commands.add(attr)
commands = list(commands)
commands.sort()
irc.reply(msg, utils.commaAndify(commands))
2003-09-30 13:04:44 +02:00
2003-09-30 12:47:05 +02:00
def uptime(self, irc, msg, args):
"""takes no arguments.
Returns the amount of time the bot has been running.
"""
response = 'I have been running for %s.' % \
utils.timeElapsed(time.time() - world.startedAt)
irc.reply(msg, response)
Class = Status
# vim:set shiftwidth=4 tabstop=8 expandtab textwidth=78: