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