Add a count attribute to Response, which show the number of left results

This commit is contained in:
Némunaire 2012-08-31 02:58:06 +02:00
parent 637861b3ae
commit 358deb99f0

View File

@ -21,7 +21,7 @@ import sys
class Response: class Response:
def __init__(self, sender, message=None, channel=None, nick=None, server=None, def __init__(self, sender, message=None, channel=None, nick=None, server=None,
nomore="No more message", title=None, more="(suite) "): nomore="No more message", title=None, more="(suite) ", count=None):
self.nomore = nomore self.nomore = nomore
self.more = more self.more = more
self.title = title self.title = title
@ -34,6 +34,7 @@ class Response:
self.nick = nick self.nick = nick
self.set_sender(sender) self.set_sender(sender)
self.alone = True self.alone = True
self.count = count
def set_sender(self, sender): def set_sender(self, sender):
if sender is None or sender.find("!") < 0: if sender is None or sender.find("!") < 0:
@ -46,8 +47,9 @@ class Response:
self.sender = sender self.sender = sender
def append_message(self, message): def append_message(self, message):
self.alone = False if message is not None and len(message) > 0:
self.messages.append(message) self.alone = False
self.messages.append(message)
@property @property
def empty(self): def empty(self):
@ -91,7 +93,10 @@ class Response:
if len(elts) <= 432: if len(elts) <= 432:
self.messages.pop(0) self.messages.pop(0)
self.elt = 0 self.elt = 0
return msg + elts if self.count is not None:
return msg + elts + (self.count % len(self.messages))
else:
return msg + elts
else: else:
words = elts.split(' ') words = elts.split(' ')