Spaces:
Sleeping
Sleeping
File size: 3,903 Bytes
4ded330 5c04706 4ded330 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 |
import imaplib
import email
from email.header import decode_header
import os
from dotenv import load_dotenv
from bs4 import BeautifulSoup # A library for parsing HTML
# Load the environment variables from your .env file
load_dotenv()
def fetch_latest_emails(num_emails=5):
try:
# Get credentials securely from the .env file
username = os.getenv("GMAIL_USER")
password = os.getenv("GMAIL_PASSWORD")
if not username or not password:
print("Error: GMAIL_USER and GMAIL_PASSWORD must be set in the .env file.")
return []
# Connect to Gmail's server
mail = imaplib.IMAP4_SSL("imap.gmail.com")
mail.login(username, password)
mail.select("inbox")
# Search for all emails and get their IDs
result, data = mail.search(None, "ALL")
mail_ids = data[0].split()
emails = []
# Fetch the most recent emails based on num_emails
for i in mail_ids[-num_emails:]:
result, data = mail.fetch(i, "(RFC822)")
raw_email = data[0][1]
msg = email.message_from_bytes(raw_email)
# Decode the subject line safely
subject, encoding = decode_header(msg["Subject"])[0]
if isinstance(subject, bytes):
subject = subject.decode(encoding or "utf-8", errors="ignore")
# Get the sender's address
from_ = msg.get("From", "")
# --- Improved Body Parsing ---
body = ""
if msg.is_multipart():
for part in msg.walk():
content_type = part.get_content_type()
content_disposition = str(part.get("Content-Disposition"))
# Skip attachments
if "attachment" not in content_disposition:
if content_type == "text/plain":
try:
body = part.get_payload(decode=True).decode()
break # Found plain text, stop searching
except:
continue
elif content_type == "text/html":
# If we only find HTML, parse it to get clean text
try:
html_body = part.get_payload(decode=True).decode()
soup = BeautifulSoup(html_body, "html.parser")
body = soup.get_text()
except:
continue
else:
# For non-multipart emails, just get the payload
try:
body = msg.get_payload(decode=True).decode()
except:
body = ""
emails.append({
"From": from_,
"Subject": subject,
"Body": body.strip()
})
mail.logout()
# Return emails in reverse order to show the newest one first
return emails[::-1]
except Exception as e:
print(f"An error occurred while fetching emails: {e}")
return []
# NEW, ROBUST CODE for the end of gmail_fetcher.py
if __name__ == "__main__":
latest_emails = fetch_latest_emails()
if latest_emails:
print(f"Successfully fetched {len(latest_emails)} emails.")
for e in latest_emails:
# Safely encode and decode each part to prevent crashes in the terminal
safe_from = e['From'].encode('utf-8', 'replace').decode('utf-8')
safe_subject = e['Subject'].encode('utf-8', 'replace').decode('utf-8')
safe_body = e['Body'][:100].encode('utf-8', 'replace').decode('utf-8')
print(f"From: {safe_from}\nSubject: {safe_subject}\nBody: {safe_body}...\n")
|