def test_decorator(self):
class Event: pass
dummy = []
events.subscribe_to(Event)(dummy.append)
e = Event()
events.notify(e)
self.assertEqual(dummy, [e])
def test_notify(self):
class Event: pass
dummy = []
events.subscribe(dummy.append, Event)
e = Event()
events.notify(e)
self.assertEqual(dummy, [e])
def test_decorator_staticmethod(self):
class Event: pass
dummy = []
class Target:
@events.subscribe_to(Event)
def run(e):
dummy.append(e)
e = Event()
events.notify(e)
self.assertEqual(dummy, [e])
def add_to_list(self, mlist, message):
list_name = unicode(mlist.fqdn_listname)
# Create the list if it does not exist
l = self.db.find(List, List.name == list_name).one()
if l is None:
l = List(list_name)
# Don't wait for the cache to set those properties
for propname in l.mailman_props:
setattr(l, propname, getattr(mlist, propname))
self.db.add(l)
if mlist.archive_policy == ArchivePolicy.never:
logger.info("Archiving disabled by list policy for %s" % list_name)
return None
if not message.has_key("Message-Id"):
raise ValueError("No 'Message-Id' header in email", message)
msg_id = unicode(unquote(message["Message-Id"]))
# Protect against extremely long Message-Ids (there is no limit in the
# email spec), it's set to VARCHAR(255) in the database
if len(msg_id) >= 255:
msg_id = msg_id[:254]
email = Email(list_name, msg_id)
if self.is_message_in_list(list_name, email.message_id):
logger.info("Duplicate email from %s: %s" % (message["From"], message.get("Subject", '""')))
return email.message_id_hash
# if not getattr(settings.KITTYSTORE_FULL_EMAIL):
# # If it's a valid value, leave it to the "prototype" archiver
# # Note: the message.as_string() call must be done before scrubbing
# email_full = EmailFull(list_name, msg_id, message.as_string())
# self.db.add(email_full)
# Find thread id
new_thread = False
ref, thread_id = get_ref_and_thread_id(message, list_name, self)
if thread_id is None:
new_thread = True
# make up the thread_id if not found
thread_id = email.message_id_hash
email.thread_id = thread_id
email.in_reply_to = ref
try:
from_name, from_email = parseaddr(message["From"])
from_name = header_to_unicode(from_name).strip()
email.sender_email = unicode(from_email).strip()
except (UnicodeDecodeError, UnicodeEncodeError):
raise ValueError("Non-ascii sender address", message)
sender = self.db.find(Sender, Sender.email == email.sender_email).one()
if sender is None:
sender = Sender(email.sender_email, from_name)
self.db.add(sender)
else:
sender.name = from_name # update the name if needed
email.subject = header_to_unicode(message.get("Subject"))
if email.subject is not None:
# limit subject size to 2000 chars or PostgreSQL may complain
email.subject = email.subject[:2000]
msg_date = parsedate(message.get("Date"))
if msg_date is None:
# Absent or unparseable date
msg_date = datetime.datetime.utcnow()
utcoffset = msg_date.utcoffset()
if msg_date.tzinfo is not None:
msg_date = msg_date.astimezone(tzutc()).replace(tzinfo=None)
email.date = msg_date
if utcoffset is None:
email.timezone = 0
else:
# in minutes
email.timezone = ((utcoffset.days * 24 * 60 * 60) + utcoffset.seconds) / 60
scrubber = Scrubber(list_name, message)
# warning: scrubbing modifies the msg in-place
email.content, attachments = scrubber.scrub()
# category = 'Question' # TODO: enum + i18n ?
# if ('agenda' in message.get('Subject', '').lower() or
# 'reminder' in message.get('Subject', '').lower()):
# # i18n!
# category = 'Agenda'
if new_thread:
thread = Thread(list_name, thread_id, email.date)
else:
thread = self.db.find(Thread, And(Thread.list_name == list_name, Thread.thread_id == thread_id)).one()
thread.date_active = email.date
self.db.add(thread)
self.db.add(email)
compute_thread_order_and_depth(thread)
for attachment in attachments:
self.add_attachment(list_name, msg_id, *attachment)
self.flush()
# invalidate the cache
events.notify(events.NewMessage(self, mlist, email))
if new_thread:
events.notify(events.NewThread(self, mlist, thread))
# search indexing
# do it after caching because we need some list properties (like
# archive_policy)
if self.search_index is not None:
self.search_index.add(email)
return email.message_id_hash
