dam_helper/email_manager/email_manager.py

from email.message import EmailMessage

from dam_helper.enums.enums import Questions
from dam_helper.prompts.prompts_manager import PromptsManager
from dam_helper.repository import Repository


class EmailManager:

    def __init__(self, pm: PromptsManager, repository:Repository):
        self.pm = pm
        self.repository = repository

    def create_email(self, answers: dict[Questions, str], pdf_form, filename) -> EmailMessage:
        msg = EmailMessage()
        msg["Subject"] = (f"Minor work permit for unit {answers[Questions.UNIT_APT_NUMBER]} in "
                          f"{answers[Questions.COMMUNITY]}")
        msg["From"] = answers[Questions.YOUR_EMAIL]
        msg["To"] = "[email protected]"
        msg.set_content(self.repository.send_prompt(self.pm.compose_email(answers), True)['content'])
        msg.add_attachment(pdf_form, maintype="application", subtype="pdf", filename=filename)
        return msg