用Python一次性从邮件附件中提取ZIP文件中的CSV文件

import imaplib
import email
import zipfile
import csv

# 连接到邮件服务器
mail = imaplib.IMAP4('mail.example.com')
mail.login('username', 'password')
mail.select('inbox')

# 搜索邮件
result, data = mail.search(None, 'ALL')
email_ids = data[0].split()

# 遍历邮件
for email_id in email_ids:
    # 获取邮件内容
    result, data = mail.fetch(email_id, '(RFC822)')
    raw_email = data[0][1]
    email_message = email.message_from_bytes(raw_email)

    # 遍历附件
    for part in email_message.iter_attachments():
        if part.get_content_type() == 'application/zip':
            # 获取附件数据
            zip_data = part.get_payload(decode=True)

            # 解压缩ZIP文件
            with zipfile.ZipFile(zip_data) as zip_file:
                # 遍历ZIP文件中的文件
                for file_name in zip_file.namelist():
                    if file_name.endswith('.csv'):
                        # 读取CSV文件内容
                        with zip_file.open(file_name) as csv_file:
                            csv_reader = csv.reader(csv_file)
                            for row in csv_reader:
                                # 处理CSV文件内容
                                print(row)

# 关闭邮件连接
mail.logout()