简体   繁体   中英

Attach Pandas Dataframe as .csv in email Without using .to_csv

I have a pandas data frame that I want to send out as a.csv file in an email attachment. So right now when I use df.to_csv() the file is downloaded every time.

I do not want to save the file in the system and instead just want to pass it straight as a.csv file. Is there a way to do this?

import pandas as pd

df = pd.DataFrame(output of sqlquery) # this data is dataframe output of a sql query. 

def send_email(sender, recipient, aws_region, subject, df):

client = boto3.client('ses', region_name=aws_region)

BODY_TEXT = "Hello,\r\nPlease find the attached file."
BODY_HTML = """\
<html>
<head></head>
<body>
<h1>Hello!</h1>
<p>Please find the attached file.</p>
</body>
</html>
"""

msg = MIMEMultipart('mixed')
msg['From'] = sender
print(msg['From'])
msg['To'] = recipient
msg['Subject'] = 'TOI Order Alert'

# The character encoding for the email.
CHARSET = "UTF-8"

msg_body = MIMEMultipart('alternative')
textpart = MIMEText(BODY_TEXT.encode(CHARSET), 'plain', CHARSET)
htmlpart = MIMEText(BODY_HTML.encode(CHARSET), 'html', CHARSET)


# Add the text and HTML parts to the child container.
msg_body.attach(textpart)
msg_body.attach(htmlpart)





# # Define the attachment part and encode it using MIMEApplication.
att = MIMEApplication(df.to_csv('test.csv'))
att.add_header('Content-Disposition','attachment; filename='+ 'test.csv')

# Attach the multipart/alternative child container to the multipart/mixed
# parent container.
msg.attach(msg_body)

# Add the attachment to the parent container.
msg.attach(att)



#Provide the contents of the email.
response = client.send_raw_email(
        Source=msg['From'],
        Destinations=[
            msg['To']
        ],
        RawMessage={
            'Data':msg.as_string(),
        }
    )
 

I'd suggest the following

import io

s_buf = io.StringIO() 
df.to_csv(s_buf)
byte_buf = s_buf.encode()

And pass in the byte buffer

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM