Spaces:
Sleeping
Sleeping
| """ | |
| PDF export utilities for email drafts. | |
| """ | |
| from fpdf import FPDF | |
| from typing import Optional | |
| class EmailPDFExporter: | |
| """Export emails to PDF format.""" | |
| def __init__(self): | |
| self.pdf = None | |
| def export_to_pdf(self, subject: str, email_content: str, | |
| sender_name: Optional[str] = None, | |
| recipient_name: Optional[str] = None) -> bytes: | |
| """Export email to PDF and return as bytes.""" | |
| self.pdf = FPDF() | |
| self.pdf.add_page() | |
| self.pdf.set_auto_page_break(auto=True, margin=15) | |
| # Calculate usable width | |
| effective_width = self.pdf.w - 2 * self.pdf.l_margin | |
| # Set up fonts | |
| self.pdf.set_font("Helvetica", "B", 16) | |
| # Title | |
| self.pdf.cell(effective_width, 10, "Email Draft", new_x="LMARGIN", new_y="NEXT", align="C") | |
| self.pdf.ln(5) | |
| # Subject - handle long subjects | |
| self.pdf.set_font("Helvetica", "B", 12) | |
| subject_text = f"Subject: {subject}" if subject else "Subject: (No subject)" | |
| self.pdf.multi_cell(effective_width, 8, subject_text) | |
| self.pdf.ln(2) | |
| # From/To if provided | |
| if sender_name: | |
| self.pdf.set_font("Helvetica", "", 10) | |
| self.pdf.cell(effective_width, 8, f"From: {sender_name}", new_x="LMARGIN", new_y="NEXT") | |
| if recipient_name: | |
| self.pdf.set_font("Helvetica", "", 10) | |
| self.pdf.cell(effective_width, 8, f"To: {recipient_name}", new_x="LMARGIN", new_y="NEXT") | |
| self.pdf.ln(5) | |
| # Horizontal line | |
| self.pdf.set_draw_color(200, 200, 200) | |
| self.pdf.line(self.pdf.l_margin, self.pdf.get_y(), self.pdf.w - self.pdf.r_margin, self.pdf.get_y()) | |
| self.pdf.ln(5) | |
| # Email content | |
| self.pdf.set_font("Helvetica", "", 11) | |
| # Handle multi-line content safely | |
| for line in email_content.split('\n'): | |
| if line.strip(): | |
| # Encode to handle special characters | |
| safe_line = line.encode('latin-1', errors='replace').decode('latin-1') | |
| self.pdf.multi_cell(effective_width, 6, safe_line) | |
| else: | |
| self.pdf.ln(4) | |
| # Return as bytes (fpdf2 returns bytes directly) | |
| return bytes(self.pdf.output()) | |
| def save_to_file(self, filepath: str, subject: str, email_content: str, | |
| sender_name: Optional[str] = None, | |
| recipient_name: Optional[str] = None): | |
| """Export email to PDF file.""" | |
| pdf_bytes = self.export_to_pdf(subject, email_content, sender_name, recipient_name) | |
| with open(filepath, 'wb') as f: | |
| f.write(pdf_bytes) | |
| def export_email_to_pdf(subject: str, email_content: str, | |
| sender_name: Optional[str] = None, | |
| recipient_name: Optional[str] = None) -> bytes: | |
| """Convenience function to export email to PDF bytes.""" | |
| exporter = EmailPDFExporter() | |
| return exporter.export_to_pdf(subject, email_content, sender_name, recipient_name) | |