Mikaela/Limnoria
1
0
Fork 0
mirror of https://github.com/Mikaela/Limnoria.git synced 2024-11-23 19:19:32 +01:00
Code Issues Projects Releases Wiki Activity

Updated to use plugins.ChannelUserDB.

Browse Source
This commit is contained in:
Jeremy Fincher 2004-02-08 06:48:07 +00:00
parent 04b2ecd5e6
commit 645d654ada
1 changed files with 19 additions and 59 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

74
plugins/ChannelStats.py
View File

@ -152,59 +152,20 @@ class UserStat(ChannelStat):
self.doPayload(msg.args[0], msg.args[2])
self.kicks += 1
class StatsDB(object):
def __init__(self, filename):
self.filename = filename
self.channels = ircutils.IrcDict()
class StatsDB(plugins.ChannelUserDB):
def __init__(self, *args, **kwargs):
self.channelStats = ircutils.IrcDict()
try:
fd = file(filename)
except EnvironmentError, e:
log.info('Couldn\'t open %s: %s', filename, e)
return
lineno = 0
for line in fd:
lineno += 1
line = line.rstrip('\r\n')
try:
(channelOrId, rest) = line.split(':', 1)
if ircutils.isChannel(channelOrId):
channel = channelOrId
values = map(int, rest.split(','))
self.channelStats[channel] = ChannelStat(*values)
else:
id = int(channelOrId)
(channel, rest) = rest.split(':', 1)
d = self.channels.setdefault(channel, {})
values = map(int, rest.split(','))
d[id] = UserStat(id, *values)
except ValueError:
log.warning('Invalid line (#%s): %r', lineno, line)
continue
fd.close()
plugins.ChannelUserDB.__init__(self, *args, **kwargs)
def flush(self):
fd = file(self.filename, 'w')
L = self.channelStats.items()
L.sort()
for (channel, stat) in L:
fd.write('%s:%s' % (channel, ','.join(map(str, stat.values()))))
fd.write(os.linesep)
if channel not in self.channels:
continue
d = self.channels[channel]
LL = d.items()
LL.sort()
for (id, stat) in LL:
fd.write('%s:%s:%s' % (id, channel,
','.join(map(str, stat.values()))))
fd.write(os.linesep)
fd.close()
def serialize(self, v):
return v.values()
def close(self):
self.flush()
self.channels.clear()
self.channelStats.clear()
def deserialize(self, channel, id, L):
L = map(int, L)
if id == 'channelStats':
return ChannelStat(L)
else:
return UserStat(L)
def addMsg(self, msg, id=None):
channel = msg.args[0]
@ -223,12 +184,11 @@ class StatsDB(object):
self.channels[channel][id] = UserStat(id)
self.channels[channel][id].addMsg(msg)
def get(self, channel, id=None):
if id is None:
return self.channelStats[channel]
else:
return self.channels[channel][id]
def getChannelStats(self, channel):
return self[channel, -1]
def getUserStats(self, channel, id):
return self[channel, id]
class ChannelStats(callbacks.Privmsg):
noIgnore = True
@ -329,7 +289,7 @@ class ChannelStats(callbacks.Privmsg):
else:
id = ircdb.users.getUserId(name)
try:
stats = self.db.get(channel, id)
stats = self.db.getUserStats(channel, id)
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, ' \
@ -362,7 +322,7 @@ class ChannelStats(callbacks.Privmsg):
"""
channel = privmsgs.getChannel(msg, args)
try:
stats = self.db.get(channel)
stats = self.db.getChannelStats(channel)
s = 'On %s there have been %s messages, containing %s ' \
'characters, %s, %s, and %s; ' \
'%s of those messages %s. There have been ' \