70 lines
2.5 KiB
Python
70 lines
2.5 KiB
Python
# Copyright 2010-2011 Florent Le Coz <louiz@louiz.org>
|
|
#
|
|
# This file is part of Poezio.
|
|
#
|
|
# Poezio is free software: you can redistribute it and/or modify
|
|
# it under the terms of the zlib license. See the COPYING file.
|
|
|
|
"""
|
|
Define the TextBuffer class
|
|
"""
|
|
|
|
import logging
|
|
log = logging.getLogger(__name__)
|
|
|
|
import collections
|
|
|
|
from datetime import datetime
|
|
from config import config
|
|
|
|
Message = collections.namedtuple('Message', 'txt nick_color time str_time nickname user')
|
|
|
|
class TextBuffer(object):
|
|
"""
|
|
This class just keep trace of messages, in a list with various
|
|
informations and attributes.
|
|
"""
|
|
def __init__(self, messages_nb_limit=config.get('max_messages_in_memory', 2048)):
|
|
self.messages_nb_limit = messages_nb_limit
|
|
self.messages = [] # Message objects
|
|
self.windows = [] # we keep track of one or more windows
|
|
# so we can pass the new messages to them, as they are added, so
|
|
# they (the windows) can build the lines from the new message
|
|
|
|
def add_window(self, win):
|
|
self.windows.append(win)
|
|
|
|
def add_message(self, txt, time=None, nickname=None, nick_color=None, history=None, user=None):
|
|
time = time or datetime.now()
|
|
if txt.startswith('/me '):
|
|
if nick_color:
|
|
color = nick_color[0]
|
|
elif user:
|
|
color = user.color[0]
|
|
else:
|
|
color = None
|
|
# TODO: display the bg color too.
|
|
txt = ("\x19%s}* \x195}" % (color or 5,))+ nickname + ' ' + txt[4:]
|
|
nickname = None
|
|
msg = Message(txt='%s\x19o'%(txt.replace('\t', ' '),), nick_color=nick_color,
|
|
time=time, str_time=time.strftime("%Y-%m-%d %H:%M:%S")\
|
|
if history else time.strftime("%H:%M:%S"),\
|
|
nickname=nickname, user=user)
|
|
self.messages.append(msg)
|
|
while len(self.messages) > self.messages_nb_limit:
|
|
self.messages.pop(0)
|
|
ret_val = None
|
|
for window in self.windows: # make the associated windows
|
|
# build the lines from the new message
|
|
nb = window.build_new_message(msg, history=history)
|
|
if ret_val is None:
|
|
ret_val = nb
|
|
if window.pos != 0:
|
|
window.scroll_up(nb)
|
|
return ret_val or 1
|
|
|
|
def del_window(self, win):
|
|
self.windows.remove(win)
|
|
|
|
def __del__(self):
|
|
log.debug('** Deleting %s messages from textbuffer' % (len(self.messages)))
|