From 3d52b106cf360d09e47f48c4f3b54e87627eda65 Mon Sep 17 00:00:00 2001 From: Patrick Ulbrich Date: Tue, 17 Dec 2013 17:59:17 +0100 Subject: [PATCH] imap msg dict doesn't seem to throw a keyerror, so throw it explicitely --- Mailnag/daemon/mails.py | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) diff --git a/Mailnag/daemon/mails.py b/Mailnag/daemon/mails.py index b327246..7447bac 100644 --- a/Mailnag/daemon/mails.py +++ b/Mailnag/daemon/mails.py @@ -154,7 +154,7 @@ class MailCollector: return mail_list - def _get_header(self, msg_dict): + def _get_header(self, msg_dict): try: content = self._get_header_field(msg_dict, 'From') sender = self._format_header_field('sender', content) @@ -184,14 +184,13 @@ class MailCollector: def _get_header_field(self, msg_dict, key): - try: + if msg_dict.has_key(key): value = msg_dict[key] - except KeyError: - try: - value = msg_dict[key.lower()] - except KeyError, err: - logging.debug("Couldn't get %s from message." % key) - raise err + elif msg_dict.has_key(key.lower()): + value = msg_dict[key.lower()] + else: + logging.debug("Couldn't get %s from message." % key) + raise KeyError return value