diff --git a/.flake8 b/.flake8 new file mode 100644 index 0000000..1f20aa3 --- /dev/null +++ b/.flake8 @@ -0,0 +1,2 @@ +[flake8] +exclude = django_mailbox/migrations, django_mailbox/south_migrations diff --git a/django_mailbox/admin.py b/django_mailbox/admin.py index 6bfebd4..25eaa71 100644 --- a/django_mailbox/admin.py +++ b/django_mailbox/admin.py @@ -23,6 +23,8 @@ def get_new_mail(mailbox_admin, request, queryset): for mailbox in queryset.all(): logger.debug('Receiving mail for %s' % mailbox) mailbox.get_new_mail() + + get_new_mail.short_description = 'Get new mail' @@ -103,6 +105,7 @@ class MessageAdmin(admin.ModelAdmin): ) actions = [resend_message_received_signal] + if getattr(settings, 'DJANGO_MAILBOX_ADMIN_ENABLED', True): admin.site.register(Message, MessageAdmin) admin.site.register(MessageAttachment, MessageAttachmentAdmin) diff --git a/django_mailbox/management/commands/rebuildmessageattachments.py b/django_mailbox/management/commands/rebuildmessageattachments.py index 3bd2eb3..6bfc24f 100644 --- a/django_mailbox/management/commands/rebuildmessageattachments.py +++ b/django_mailbox/management/commands/rebuildmessageattachments.py @@ -23,7 +23,7 @@ class Command(BaseCommand): """ def handle(self, *args, **options): - ATTACHMENT_HASH_MAP = {} + attachment_hash_map = {} attachments_without_messages = MessageAttachment.objects.filter( message=None @@ -38,7 +38,7 @@ class Command(BaseCommand): md5 = hashlib.md5() for chunk in attachment.document.file.chunks(): md5.update(chunk) - ATTACHMENT_HASH_MAP[md5.hexdigest()] = attachment.pk + attachment_hash_map[md5.hexdigest()] = attachment.pk for message_record in Message.objects.all().order_by('id'): message = email.message_from_string(message_record.body) @@ -51,9 +51,9 @@ class Command(BaseCommand): md5 = hashlib.md5() md5.update(part.get_payload(decode=True)) digest = md5.hexdigest() - if digest in ATTACHMENT_HASH_MAP: + if digest in attachment_hash_map: attachment = MessageAttachment.objects.get( - pk=ATTACHMENT_HASH_MAP[digest] + pk=attachment_hash_map[digest] ) attachment.message = message_record attachment.save() diff --git a/django_mailbox/models.py b/django_mailbox/models.py index 89eb0ad..2e6a90c 100644 --- a/django_mailbox/models.py +++ b/django_mailbox/models.py @@ -733,7 +733,7 @@ class MessageAttachment(models.Model): try: headers = headers.encode('utf-8') except UnicodeDecodeError: - headers = unicode(headers, 'utf-8').encode('utf-8') + headers = headers.decode('utf-8').encode('utf-8') return email.message_from_string(headers) def _set_dehydrated_headers(self, email_object): diff --git a/django_mailbox/tests/base.py b/django_mailbox/tests/base.py index 0f5516c..5553aa1 100644 --- a/django_mailbox/tests/base.py +++ b/django_mailbox/tests/base.py @@ -171,7 +171,7 @@ class EmailMessageTestCase(TestCase): ) ) - def assertEqual(self, left, right): + def assertEqual(self, left, right): # noqa: N802 if not isinstance(left, email.message.Message): return super(EmailMessageTestCase, self).assertEqual(left, right) return self.compare_email_objects(left, right) diff --git a/django_mailbox/tests/test_process_email.py b/django_mailbox/tests/test_process_email.py index f265bdc..de123c2 100644 --- a/django_mailbox/tests/test_process_email.py +++ b/django_mailbox/tests/test_process_email.py @@ -213,7 +213,7 @@ class TestProcessEmail(EmailMessageTestCase): success = True try: - msg = self.mailbox.process_incoming_message(email_object) + self.mailbox.process_incoming_message(email_object) except ValueError: success = False @@ -232,7 +232,8 @@ class TestProcessEmail(EmailMessageTestCase): # this is the code of _process_message() msg = Message() # if STORE_ORIGINAL_MESSAGE: - # msg.eml.save('%s.eml' % uuid.uuid4(), ContentFile(message), save=False) + # msg.eml.save('%s.eml' % uuid.uuid4(), ContentFile(message), + # save=False) msg.mailbox = self.mailbox if 'subject' in message: msg.subject = convert_header_to_unicode(message['subject'])[0:255] @@ -246,7 +247,7 @@ class TestProcessEmail(EmailMessageTestCase): msg.to_header = convert_header_to_unicode(message['Delivered-To']) msg.save() - #here the message is ok + # here the message is ok str_msg = message.as_string() message = self.mailbox._get_dehydrated_message(message, msg) try: @@ -358,7 +359,8 @@ class TestProcessEmail(EmailMessageTestCase): self.assertEqual(expected_from, actual_from) def test_message_reply(self): - email_object = EmailMessage('Test subject', # subject + email_object = EmailMessage( + 'Test subject', # subject 'Test body', # body 'username@example.com', # from ['mr.test32@mail.ru'], # to diff --git a/django_mailbox/tests/test_transports.py b/django_mailbox/tests/test_transports.py index aaa0d0d..1d3c68f 100644 --- a/django_mailbox/tests/test_transports.py +++ b/django_mailbox/tests/test_transports.py @@ -59,6 +59,7 @@ class IMAPTestCase(EmailMessageTestCase): return FAKE_UID_FETCH_SIZES if arg2 == '(RFC822)': return FAKE_UID_FETCH_MSG + def imap_server_list_method(pattern=None): return FAKE_LIST_ARCHIVE_FOLDERS_ANSWERS @@ -88,6 +89,7 @@ class TestImapTransport(IMAPTestCase): expected_message = self._get_email_object('generic_message.eml') self.assertEqual(expected_message, actual_message) + class TestImapArchivedTransport(TestImapTransport): def setUp(self): super(TestImapArchivedTransport, self).setUp() @@ -100,6 +102,7 @@ class TestImapArchivedTransport(TestImapTransport): ) self.transport.server = self.imap_server + class TestMaxSizeImapTransport(TestImapTransport): @override_settings(DJANGO_MAILBOX_MAX_MESSAGE_SIZE=5807) @@ -115,7 +118,9 @@ class TestMaxSizeImapTransport(TestImapTransport): def test_size_limit(self): all_message_ids = self.transport._get_all_message_ids() - small_message_ids = self.transport._get_small_message_ids(all_message_ids) + small_message_ids = self.transport._get_small_message_ids( + all_message_ids, + ) self.assertEqual(len(small_message_ids), 1) def test_get_email_message(self): @@ -163,4 +168,3 @@ class TestPop3Transport(EmailMessageTestCase): expected_message = self._get_email_object('generic_message.eml') self.assertEqual(expected_message, actual_message) - diff --git a/django_mailbox/transports/__init__.py b/django_mailbox/transports/__init__.py index 03b31f0..7aeef95 100644 --- a/django_mailbox/transports/__init__.py +++ b/django_mailbox/transports/__init__.py @@ -1,3 +1,5 @@ +# all imports below are only used by external modules +# flake8: noqa from django_mailbox.transports.imap import ImapTransport from django_mailbox.transports.pop3 import Pop3Transport from django_mailbox.transports.maildir import MaildirTransport diff --git a/django_mailbox/transports/base.py b/django_mailbox/transports/base.py index 753df52..d65ad35 100644 --- a/django_mailbox/transports/base.py +++ b/django_mailbox/transports/base.py @@ -4,9 +4,9 @@ import six # Do *not* remove this, we need to use this in subclasses of EmailTransport if six.PY3: - from email.errors import MessageParseError + from email.errors import MessageParseError # noqa: F401 else: - from email.Errors import MessageParseError + from email.Errors import MessageParseError # noqa: F401 class EmailTransport(object): diff --git a/django_mailbox/utils.py b/django_mailbox/utils.py index 3159c2c..17a4b37 100644 --- a/django_mailbox/utils.py +++ b/django_mailbox/utils.py @@ -89,7 +89,7 @@ def convert_header_to_unicode(header): header, default_charset, ) - return unicode(header, default_charset, 'replace') + return header.decode(default_charset, 'replace') def get_body_from_message(message, maintype, subtype): diff --git a/tox.ini b/tox.ini index 8acbaf8..4255817 100644 --- a/tox.ini +++ b/tox.ini @@ -1,5 +1,5 @@ [tox] -envlist=py27,py34,py35 +envlist=py27,py34,py35,flake8 [testenv] deps= @@ -8,3 +8,7 @@ deps= sitepackages=False commands= py.test {posargs} + +[testenv:flake8] +deps=flake8 +commands=flake8 django_mailbox