refactor and black

app/config.py

@@ -10,6 +10,7 @@ class SMTPConfig(BaseModel):
     port: int = os.getenv("EMAIL_PORT", 587)
     username: str = os.getenv("EMAIL_HOST_USER", "smtp_user")
     password: str = os.getenv("EMAIL_HOST_PASSWORD", "smtp_password")
+    template_path: str = os.getenv("EMAIL_TEMPLATE_PATH", "templates")
 
 
 class Settings(BaseSettings):

app/services/scheduler.py

@@ -45,4 +45,3 @@ class SchedulerMiddleware:
                 await self.app(scope, receive, send)
         else:
             await self.app(scope, receive, send)
-

app/services/smtp.py

@@ -1,3 +1,4 @@
+from attrs import define, field
 import smtplib
 from email.mime.multipart import MIMEMultipart
 from email.mime.text import MIMEText
@@ -15,23 +16,35 @@ from app.utils.singleton import SingletonMetaNoArgs
 logger = AppLogger().get_logger()
 
 
+@define
 class SMTPEmailService(metaclass=SingletonMetaNoArgs):
-    def __init__(self):
+    # SMTP configuration
-        self.server = smtplib.SMTP(
+    server_host: str = field(default=global_settings.smtp.server)
-            global_settings.smtp.server, global_settings.smtp.port
+    server_port: int = field(default=global_settings.smtp.port)
-        )
+    username: str = field(default=global_settings.smtp.username)
-        self.server.starttls()
+    password: str = field(default=global_settings.smtp.password)
-        self.server.login(global_settings.smtp.username, global_settings.smtp.password)
-        self.templates = Jinja2Templates("templates")
 
-    def send_email(
+    # Dependencies
+    templates: Jinja2Templates = field(
+        factory=lambda: Jinja2Templates(global_settings.templates_dir)
+    )
+    server: smtplib.SMTP = field(init=False)  # Deferred initialization in post-init
+
+    def __attrs_post_init__(self):
+        """Initialize the SMTP server connection after object creation."""
+        self.server = smtplib.SMTP(self.server_host, self.server_port)
+        self.server.starttls()
+        self.server.login(self.username, self.password)
+
+    def _prepare_email(
         self,
         sender: EmailStr,
         recipients: list[EmailStr],
         subject: str,
-        body_text: str = "",
+        body_text: str,
-        body_html=None,
+        body_html: str,
-    ):
+    ) -> MIMEMultipart:
+        """Prepare the email message."""
         msg = MIMEMultipart()
         msg["From"] = sender
         msg["To"] = ",".join(recipients)
@@ -39,16 +52,29 @@ class SMTPEmailService(metaclass=SingletonMetaNoArgs):
         msg.attach(MIMEText(body_text, "plain"))
         if body_html:
             msg.attach(MIMEText(body_html, "html"))
+        return msg
+
+    def send_email(
+        self,
+        sender: EmailStr,
+        recipients: list[EmailStr],
+        subject: str,
+        body_text: str = "",
+        body_html: str = None,
+    ):
+        """Send a regular email (plain text or HTML)."""
+        msg = self._prepare_email(sender, recipients, subject, body_text, body_html)
         self.server.sendmail(sender, recipients, msg.as_string())
 
     def send_template_email(
         self,
         recipients: list[EmailStr],
         subject: str,
-        template: str = None,
+        template: str,
-        context: dict = None,
+        context: dict,
-        sender: EmailStr = global_settings.smtp.from_email,
+        sender: EmailStr,
     ):
+        """Send an email using a template with the provided context."""
         template_str = self.templates.get_template(template)
         body_html = template_str.render(context)
         self.send_email(sender, recipients, subject, body_html=body_html)