159 lines
4.5 KiB
Python
159 lines
4.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright(C) 2010 Romain Bignon
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, version 3 of the License.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
|
|
|
|
import datetime
|
|
import time
|
|
|
|
from .base import IBaseCap, CapBaseObject, NotLoaded
|
|
|
|
|
|
__all__ = ['ICapMessages', 'ICapMessagesPost', 'Message', 'Thread', 'CantSendMessage']
|
|
|
|
|
|
class Message(CapBaseObject):
|
|
IS_HTML = 0x001 # The content is HTML formatted
|
|
IS_UNREAD = 0x002 # The message is unread
|
|
IS_ACCUSED = 0x004 # The receiver has read this message
|
|
IS_NOT_ACCUSED = 0x008 # The receiver has not read this message
|
|
|
|
def __init__(self, thread, id,
|
|
title,
|
|
sender,
|
|
receiver=NotLoaded,
|
|
date=None,
|
|
parent=NotLoaded,
|
|
content=NotLoaded,
|
|
signature=NotLoaded,
|
|
children=NotLoaded,
|
|
flags=0):
|
|
CapBaseObject.__init__(self, id)
|
|
self.thread = thread
|
|
self.title = title
|
|
self.sender = sender
|
|
self.receiver = receiver
|
|
|
|
if date is None:
|
|
date = datetime.datetime.utcnow()
|
|
self.date = date
|
|
|
|
if isinstance(parent, Message):
|
|
self.parent = parent
|
|
else:
|
|
self.parent = NotLoaded
|
|
self._parent_id = parent
|
|
|
|
self.content = content
|
|
self.signature = signature
|
|
self.children = children
|
|
self.flags = flags
|
|
|
|
@property
|
|
def date_int(self):
|
|
return int(time.strftime('%Y%m%d%H%M%S', self.date.timetuple()))
|
|
|
|
@property
|
|
def full_id(self):
|
|
return '%s.%s' % (self.thread.id, self.id)
|
|
|
|
@property
|
|
def full_parent_id(self):
|
|
if self.parent:
|
|
return self.parent.full_id()
|
|
elif self._parent_id is None:
|
|
return ''
|
|
elif self._parent_id is NotLoaded:
|
|
return NotLoaded
|
|
else:
|
|
return '%s.%s' % (self.thread.id, self._parent_id)
|
|
|
|
def __eq__(self, msg):
|
|
return self.thread.id == msg.thread.id and self.id == msg.id
|
|
|
|
def __repr__(self):
|
|
result = '<Message id="%s" title="%s" date="%s" from="%s">' % (
|
|
self.full_id, self.title, self.date, self.sender)
|
|
return result.encode('utf-8')
|
|
|
|
class Thread(CapBaseObject):
|
|
def __init__(self, id):
|
|
CapBaseObject.__init__(self, id)
|
|
self.root = NotLoaded
|
|
self.title = NotLoaded
|
|
self.date = NotLoaded
|
|
self.nb_messages = NotLoaded
|
|
self.nb_unread = NotLoaded
|
|
|
|
def iter_all_messages(self):
|
|
if self.root:
|
|
yield self.root
|
|
for m in self._iter_all_messages(self.root):
|
|
yield m
|
|
|
|
def _iter_all_messages(self, message):
|
|
if message.children:
|
|
for child in message.children:
|
|
yield child
|
|
for m in self._iter_all_messages(child):
|
|
yield m
|
|
|
|
class ICapMessages(IBaseCap):
|
|
def iter_threads(self):
|
|
"""
|
|
Iterates on threads, from newers to olders.
|
|
|
|
@return [iter] Thread objects
|
|
"""
|
|
raise NotImplementedError()
|
|
|
|
def get_thread(self, id):
|
|
"""
|
|
Get a specific thread.
|
|
|
|
@return [Thread] the Thread object
|
|
"""
|
|
raise NotImplementedError()
|
|
|
|
def iter_unread_messages(self, thread=None):
|
|
"""
|
|
Iterates on messages which hasn't been marked as read.
|
|
|
|
@param thread thread name (optional)
|
|
@return [iter] Message objects
|
|
"""
|
|
raise NotImplementedError()
|
|
|
|
def set_message_read(self, message):
|
|
"""
|
|
Set a message as read.
|
|
|
|
@param [message] message read (or ID)
|
|
"""
|
|
raise NotImplementedError()
|
|
|
|
class CantSendMessage(Exception):
|
|
pass
|
|
|
|
class ICapMessagesPost(IBaseCap):
|
|
def post_message(self, message):
|
|
"""
|
|
Post a reply.
|
|
|
|
@param message Message object
|
|
@return
|
|
"""
|
|
raise NotImplementedError()
|