77 lines
2.4 KiB
Python
77 lines
2.4 KiB
Python
import os
|
|
import codecs
|
|
import time
|
|
import re
|
|
|
|
from util import hook
|
|
|
|
timestamp_format = '%H:%M:%S'
|
|
|
|
formats = {
|
|
'PRIVMSG': '<%(nick)s> %(msg)s',
|
|
'PART': '-!- %(nick)s [%(user)s@%(host)s] has left %(chan)s',
|
|
'JOIN': '-!- %(nick)s [%(user)s@%(host)s] has joined %(param0)s',
|
|
'MODE': '-!- mode/%(chan)s [%(param_tail)s] by %(nick)s',
|
|
'KICK': '-!- %(param1)s was kicked from %(chan)s by %(nick)s [%(msg)s]',
|
|
'TOPIC': '-!- %(nick)s changed the topic of %(chan)s to: %(msg)s',
|
|
'QUIT': '-!- %(nick)s has quit [%(msg)s]',
|
|
'PING': '',
|
|
'NOTICE': '-%(nick)s- %(msg)s'
|
|
}
|
|
|
|
ctcp_formats = {
|
|
'ACTION': '* %(nick)s %(ctcpmsg)s',
|
|
'VERSION': '%(nick)s has requested CTCP %(ctcpcmd)s from %(chan)s: %(ctcpmsg)s',
|
|
'PING': '%(nick)s has requested CTCP %(ctcpcmd)s from %(chan)s: %(ctcpmsg)s',
|
|
'TIME': '%(nick)s has requested CTCP %(ctcpcmd)s from %(chan)s: %(ctcpmsg)s',
|
|
'FINGER': '%(nick)s has requested CTCP %(ctcpcmd)s from %(chan)s: %(ctcpmsg)s'
|
|
}
|
|
|
|
irc_color_re = re.compile(r'(\x03(\d+,\d+|\d)|[\x0f\x02\x16\x1f])')
|
|
|
|
|
|
def gmtime(format):
|
|
return time.strftime(format, time.gmtime())
|
|
|
|
|
|
def beautify(input):
|
|
format = formats.get(input.command, '%(raw)s')
|
|
args = dict(input)
|
|
|
|
leng = len(args['paraml'])
|
|
for n, p in enumerate(args['paraml']):
|
|
args['param' + str(n)] = p
|
|
args['param_' + str(abs(n - leng))] = p
|
|
|
|
args['param_tail'] = ' '.join(args['paraml'][1:])
|
|
args['msg'] = irc_color_re.sub('', args['msg'])
|
|
|
|
if input.command == 'PRIVMSG' and input.msg.count('\x01') >= 2:
|
|
ctcp = input.msg.split('\x01', 2)[1].split(' ', 1)
|
|
if len(ctcp) == 1:
|
|
ctcp += ['']
|
|
args['ctcpcmd'], args['ctcpmsg'] = ctcp
|
|
format = ctcp_formats.get(args['ctcpcmd'],
|
|
'%(nick)s [%(user)s@%(host)s] requested unknown CTCP '
|
|
'%(ctcpcmd)s from %(chan)s: %(ctcpmsg)s')
|
|
|
|
return format % args
|
|
|
|
@hook.singlethread
|
|
@hook.event('*')
|
|
def log(paraml, input=None, bot=None):
|
|
timestamp = gmtime(timestamp_format)
|
|
|
|
if input.command == 'QUIT': # these are temporary fixes until proper
|
|
input.chan = 'quit' # presence tracking is implemented
|
|
if input.command == 'NICK':
|
|
input.chan = 'nick'
|
|
|
|
beau = beautify(input)
|
|
|
|
if beau == '': # don't log this
|
|
return
|
|
|
|
print timestamp, input.chan, beau.encode('utf8', 'ignore')
|
|
|