import logging import os import subprocess import tempfile from contextlib import closing from datetime import datetime from html import escape import re from odoo import models, api from odoo.tools.misc import find_in_path _logger = logging.getLogger(__name__) class AccountMove(models.Model): _inherit = "account.move" @api.model def _routing_check_route(self, message, message_dict, route, raise_exception=True): """Override to create a PDF attachment from the email content before checking the route. The standard Odoo behavior bounces emails without attachments, but we want to process them by generating a PDF from the email content. """ if route[0] == "account.move" and ( len(message_dict.get("attachments", [])) < 1 or message_dict["attachments"][0][0].lower().endswith(".eml") ): try: # Create a PDF from the email content pdf_attachment = self._create_pdf_from_email(message_dict) if pdf_attachment: # Add the PDF to the message's attachments if not message_dict.get("attachments"): message_dict["attachments"] = [] # Convert the attachment to the format expected by mail_thread # Format: (name, base64_data, info_dict) attachment_data = ( pdf_attachment["name"], pdf_attachment["datas"], {"mimetype": "application/pdf"}, ) message_dict["attachments"].append(attachment_data) except Exception as e: _logger.exception( "Error creating PDF from email in _routing_check_route: %s", e ) return super()._routing_check_route( message, message_dict, route, raise_exception=raise_exception ) @classmethod def _html_to_pdf(cls, html_content): """Convert HTML content to PDF using wkhtmltopdf. Args: html_content (str): HTML content to convert to PDF Returns: bytes: PDF content as bytes or False if conversion failed """ # Check if the content is plain text (not HTML) # More comprehensive check for HTML content is_html = bool(re.search(r" 
{escaped_content}
""" # Check if wkhtmltopdf is installed wkhtmltopdf_bin = find_in_path("wkhtmltopdf") if not wkhtmltopdf_bin: _logger.error("Cannot find wkhtmltopdf executable in system path") return False # Create temporary files for the HTML input and PDF output html_file_fd, html_file_path = tempfile.mkstemp( suffix=".html", prefix="email_to_pdf." ) pdf_file_fd, pdf_file_path = tempfile.mkstemp( suffix=".pdf", prefix="email_to_pdf." ) try: # Write the HTML content to the temporary file with closing(os.fdopen(html_file_fd, "wb")) as html_file: encoded_content = html_content.encode("utf-8") html_file.write(encoded_content) # Close the PDF file descriptor as wkhtmltopdf will write to it os.close(pdf_file_fd) # Basic wkhtmltopdf command arguments command = [wkhtmltopdf_bin] command.extend(["--encoding", "utf-8"]) command.extend(["--page-size", "A4"]) command.extend(["--margin-top", "10mm"]) command.extend(["--margin-bottom", "10mm"]) command.extend(["--margin-left", "10mm"]) command.extend(["--margin-right", "10mm"]) command.append(html_file_path) command.append(pdf_file_path) process = subprocess.Popen( command, stdout=subprocess.PIPE, stderr=subprocess.PIPE ) out, err = process.communicate() if process.returncode != 0: _logger.error( "wkhtmltopdf failed with error code %s: %s", process.returncode, err ) return False # Read the generated PDF try: with open(pdf_file_path, "rb") as pdf_file: pdf_content = pdf_file.read() return pdf_content except Exception as e: _logger.exception("Error reading generated PDF file: %s", e) return False except Exception as e: _logger.exception("Error during PDF generation: %s", e) return False finally: # Clean up temporary files try: os.unlink(html_file_path) os.unlink(pdf_file_path) except (OSError, IOError): _logger.error("Failed to remove temporary files") def _create_pdf_from_email(self, message_dict): """Create a PDF attachment from an email message. Args: message_dict (dict): Email message dictionary Returns: dict: Dictionary with keys `name` and `datas` containing the name and base64 encoded data of the attachment """ # Log message dict keys for debugging _logger.info(f"Message dict keys: {list(message_dict.keys())}") # Extract email details email_from = message_dict.get("email_from", "Unknown Sender") email_date = message_dict.get("date", datetime.now()) subject = message_dict.get("subject", "No Subject") body = message_dict.get("body", "") # Format the date if it's a datetime object if isinstance(email_date, datetime): email_date = email_date.strftime("%Y-%m-%d %H:%M:%S") # Check if body is empty or None if not body: body = "

This email did not contain any body content.

" # Check if the body is plain text based on content-type # The content type can be found in the message headers or directly in the message_dict content_type = "" # Try to get content type from various places in the message dict if "content-type" in message_dict: content_type = message_dict["content-type"].lower() # Try to get from headers headers = message_dict.get("headers", {}) if not content_type and headers: if isinstance(headers, dict): content_type = headers.get("Content-Type", "").lower() elif isinstance(headers, list): for header in headers: if ( isinstance(header, tuple) and len(header) >= 2 and header[0].lower() == "content-type" ): content_type = header[1].lower() break # Try to determine from the body content if we still don't have a content type if not content_type: if re.search(r" tags and wrap in paragraph tags # Escape HTML special characters to prevent injection body = f"
{escape(body)}
" # Create HTML content for the PDF html_content = f""" """ # Convert HTML to PDF pdf_content = self._html_to_pdf(html_content) if not pdf_content: _logger.error("PDF generation failed") return False # Create a proper ir.attachment record filename = f"Email_{subject.replace(' ', '_')[:30]}.pdf" # Note: No need to base64 encode the PDF content here # When this attachment is added to message_dict["attachments"], Odoo expects raw binary data # Odoo will handle the base64 encoding when creating the actual ir.attachment record attachment = { "name": filename, "datas": pdf_content, } return attachment