69 lines
1.9 KiB
Python
69 lines
1.9 KiB
Python
"""
|
|
log.py: written by Scaevolus 2009
|
|
"""
|
|
|
|
from __future__ import with_statement
|
|
|
|
import os
|
|
import thread
|
|
import codecs
|
|
import time
|
|
|
|
import hook
|
|
|
|
lock = thread.allocate_lock()
|
|
log_fds = {} # '%(net)s %(chan)s' : (filename, fd)
|
|
|
|
timestamp_format = '%H:%M:%S'
|
|
|
|
def get_log_filename(dir, network, chan):
|
|
return os.path.join(dir, 'log', gmtime('%Y'), network,
|
|
gmtime('%%s.%m-%d.log') % chan)
|
|
|
|
def gmtime(format):
|
|
return time.strftime(format, time.gmtime())
|
|
|
|
def load_memory(filename, mtimes={}):
|
|
if not os.path.exists(filename):
|
|
return {}
|
|
mtime = os.stat(filename).st_mtime
|
|
if mtimes.get(filename, 0) != mtime:
|
|
mtimes[filename] = mtime
|
|
return dict((x.split(None, 1)[0].lower(), x.strip()) for x in
|
|
codecs.open(filename, 'r', 'utf-8'))
|
|
|
|
def save_memory(filename, memory):
|
|
out = codecs.open(filename, 'w', 'utf-8')
|
|
out.write('\n'.join(sorted(memory.itervalues())))
|
|
out.flush()
|
|
out.close()
|
|
|
|
def get_log_fd(dir, network, chan):
|
|
fn = get_log_filename(dir, network, chan)
|
|
cache_key = '%s %s' % (network, chan)
|
|
filename, fd = log_fds.get(cache_key, ('', 0))
|
|
|
|
if fn != filename: # we need to open a file for writing
|
|
if fd != 0: # is a valid fd
|
|
fd.flush()
|
|
fd.close()
|
|
dir = os.path.split(fn)[0]
|
|
if not os.path.exists(dir):
|
|
os.makedirs(dir)
|
|
fd = codecs.open(fn, 'wab', 'utf-8')
|
|
log_fds[cache_key] = (fn, fd)
|
|
|
|
return fd
|
|
|
|
@hook.event(ignorebots=False)
|
|
def log(bot, input):
|
|
".remember <word> <data> -- maps word to data in the memory"
|
|
with lock:
|
|
fd = get_log_fd(bot.persist_dir, bot.network, 'RAW')
|
|
fd.write(gmtime(timestamp_format) + ' ' + input.raw + '\n')
|
|
|
|
if input.chan:
|
|
fd = get_log_fd(bot.persist_dir, bot.network, input.chan)
|
|
fd.write(gmtime(timestamp_format) + ' ' + input.raw + '\n')
|
|
|