python poplib получить вложение - PullRequest
5 голосов
/ 01 июля 2011

Я пытаюсь получить доступ к серверу электронной почты POP3.Я буду опрашивать сообщения и загружать вложения для каждого из них.Я могу успешно войти в систему и получить сообщения, но не могу понять, как на самом деле получить вложение, которое мне нужно будет проанализировать позже.Я думаю, что смогу сохранить в tmp dir, пока не обработаю его.

Вот что я получил до сих пор:

pop = poplib.POP3_SSL(server)
pop.user(usr)
pop.pass_(pwd)

f = open(file_dir, 'w')
num_msgs = len(pop.list()[1])
for msg_list in range(num_msgs):
    for msg in pop.retr(msg_list+1)[1]:
        mail = email.message_from_string(msg)
        for part in mail.walk():
            f.write(part.get_payload(decode=True))
f.close()

Это код, который я собрал воедино из примеров, которые я нашел в Интернете, но не являлся надежным примером фактического получения вложения.Файл, в который я пишу, пуст.Что мне здесь не хватает?Заранее спасибо.

Ответы [ 2 ]

3 голосов
/ 11 июня 2015

Пожалуйста, посмотрите полный пример ниже.

Импорт poplib и парсера

import poplib from email import parser

Функция для возврата соединения с поп-сервером:

def mail_connection(server='pop.mymailserver.com'):
    pop_conn = poplib.POP3(server)
    pop_conn.user('someuser@server')
    pop_conn.pass_('password')
    return pop_conn

Функция для получения почты:

def fetch_mail(delete_after=False):
    pop_conn = mail_connection() 
    messages = [pop_conn.retr(i) for i in range(1, len(pop_conn.list()[1]) + 1)]
    messages = ["\n".join(mssg[1]) for mssg in messages]
    messages = [parser.Parser().parsestr(mssg) for mssg in messages]
    if delete_after == True:
        delete_messages = [pop_conn.dele(i) for i in range(1, len(pop_conn.list()[1]) + 1)]
    pop_conn.quit()
    return messages

Затем функция для сохранения вложений в виде файлов. NB, allowed mimetypes; у вас мог бы быть их список, такой как:

allowed_mimetypes = ["application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"]

и т. Д.

def get_attachments():
    messages = fetch_mail()
    attachments = []
    for msg in messages:
        for part in msg.walk():
            if part.get_content_type() in allowed_mimetypes:
                name = part.get_filename()
                data = part.get_payload(decode=True)
                f = file(name,'wb')
                f.write(data)
                f.close()
                attachments.append(name)
    return attachments
3 голосов
/ 20 января 2012

Я знаю, что это старый вопрос, но на всякий случай: значение, которое вы передаете email.message_from_string, на самом деле представляет собой список содержимого письма, где каждый элемент представляет собой строку.Вам нужно присоединиться к нему, чтобы получить строковое представление этого письма:

mail = email.message_from_string("".join(msg))
...