68c6c06095
Python recursion sucks, and namedtuple.repr() uses that, so I rewrote __str__ and __repr__ in order to make it the iterative way instead.
121 lines
4.5 KiB
Python
121 lines
4.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
|
|
from theming import get_theme
|
|
|
|
message_fields = 'txt nick_color time str_time nickname user identifier highlight me old_message revisions'
|
|
Message = collections.namedtuple('Message', message_fields)
|
|
|
|
def other_elems(self):
|
|
acc = ['Message(']
|
|
fields = message_fields.split()
|
|
fields.remove('old_message')
|
|
for field in fields:
|
|
acc.append('%s=%s' % (field, getattr(self, field)))
|
|
return (', '.join(acc) + ', old_message=')
|
|
|
|
def repr_message(self):
|
|
init = other_elems(self)
|
|
acc = []
|
|
next = self.old_message
|
|
rev = 0
|
|
while next:
|
|
acc.append(other_elems(next))
|
|
next = next.old_message
|
|
rev += 1
|
|
acc.append('None')
|
|
while rev:
|
|
acc.append(')')
|
|
rev -= 1
|
|
return ''.join(acc)
|
|
|
|
Message.__repr__ = repr_message
|
|
Message.__str__ = repr_message
|
|
|
|
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)
|
|
|
|
@property
|
|
def last_message(self):
|
|
return self.messages[-1] if self.messages else None
|
|
|
|
|
|
def make_message(self, txt, time, nickname, nick_color, history, user, identifier, str_time=None, highlight=False, old_message=None, revisions=0):
|
|
time = time or datetime.now()
|
|
me = False
|
|
if txt.startswith('/me '):
|
|
me = True
|
|
txt = '\x19%(info_col)s}' % {'info_col': get_theme().COLOR_ME_MESSAGE[0]} + txt[4:]
|
|
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")) if str_time is None else '',
|
|
nickname=nickname,
|
|
user=user,
|
|
identifier=identifier,
|
|
highlight=highlight,
|
|
me=me,
|
|
old_message=old_message,
|
|
revisions=revisions)
|
|
log.debug('Set message %s with %s.', identifier, msg)
|
|
return msg
|
|
|
|
def add_message(self, txt, time=None, nickname=None, nick_color=None, history=None, user=None, highlight=False, identifier=None, str_time=None):
|
|
msg = self.make_message(txt, time, nickname, nick_color, history, user, identifier, str_time=str_time, highlight=highlight)
|
|
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, highlight=highlight, timestamp=config.get("show_timestamps", "true") != 'false')
|
|
if ret_val is None:
|
|
ret_val = nb
|
|
if window.pos != 0:
|
|
window.scroll_up(nb)
|
|
return ret_val or 1
|
|
|
|
def modify_message(self, txt, old_id, new_id, highlight=False, time=None):
|
|
for i in range(len(self.messages) -1, -1, -1):
|
|
msg = self.messages[i]
|
|
if msg.identifier == old_id:
|
|
message = self.make_message(txt, time if time else msg.time, msg.nickname, msg.nick_color, None, msg.user, new_id, highlight=highlight, old_message=msg, revisions=msg.revisions + 1)
|
|
self.messages[i] = message
|
|
log.debug('Replacing message %s with %s.', old_id, new_id)
|
|
return message
|
|
log.debug('Message %s not found in text_buffer, abort replacement.', old_id)
|
|
return
|
|
|
|
def del_window(self, win):
|
|
self.windows.remove(win)
|
|
|
|
def __del__(self):
|
|
log.debug('** Deleting %s messages from textbuffer', len(self.messages))
|