add more receivers in mail

config/config.yaml

@@ -84,7 +84,9 @@ excel:
 mail:
   smtp_server: smtp.gmail.com
   port: 587
-  receiver: [email protected] # For demo only
+  receivers: # For demo only
+    - [email protected]
+    - [email protected]
 
   modify_invoice_remind:
     subject: "[No Reply - Done extracting Invoice Information]"

src/api.py

@@ -73,7 +73,7 @@ def process_change_stream(ocr_reader, invoice_extractor, config):
                     mongo_db.update_document_by_id(str(document_id), new_data)
 
                     email_sender.send_email(email_type='modify_invoice_remind',
-                                            receiver=None,
+                                            receivers=None,
                                             )
 
                 except Exception as e:
@@ -121,7 +121,7 @@ def process_change_stream(ocr_reader, invoice_extractor, config):
                 employee_expense_report_path, output_2_excel = export_json_to_excel(invoice_pairs=invoice_pairs, logger=logger)
                 if employee_expense_report_path or output_2_excel:
                     email_sender.send_email(email_type='send_excel',
-                                            receiver=None,
+                                            receivers=None,
                                             attachment_paths=[employee_expense_report_path, output_2_excel])
                     logger.debug(f"attachment_paths: {[employee_expense_report_path, output_2_excel]}")
             except Exception as e:

src/mail.py

@@ -23,22 +23,30 @@ def __init__(self, config: dict, logger=None):
         self.logger = logger
 
     def send_email(self, email_type: Literal['modify_invoice_remind', 'send_excel'], 
-                   receiver: str = None, 
+                   receivers: list[str] = None, 
                    attachment_paths: Optional[List[str]] = None):
+        """
+        Function to send email
+        email_type: 
+        - 'modify_invoice_remind': To remind the employees that the invoices have been extracted, go to website and modify the info
+        - 'send_excel': Send email to admin with excel files
+        receivers: List of email string
+        attachment_paths: List of path to attachment files
+        """
         try:
             # Connect to the SMTP server for each email
             server = smtplib.SMTP(self.smtp_server, self.port)
             server.starttls()
             server.login(self.login, self.password)
 
-            if not receiver:
-                receiver = self.config["receiver"]    
+            if not receivers:
+                receivers = self.config["receivers"]    
 
-            message = self._prepare_email(email_type=email_type, receiver=receiver, attachment_paths=attachment_paths)
+            message = self._prepare_email(email_type=email_type, receivers=receivers, attachment_paths=attachment_paths)
             server.send_message(message)
-            print(f"Email sent successfully to {receiver}")
+            print(f"Email sent successfully to {receivers}")
             if self.logger:
-                self.logger.debug(f"Email sent to {receiver}")
+                self.logger.debug(f"Email sent to {receivers}")
 
         except Exception as e:
             msg = f"Error sending email: {str(e)}"
@@ -51,7 +59,7 @@ def send_email(self, email_type: Literal['modify_invoice_remind', 'send_excel'],
             server.quit()
 
     def _prepare_email(self, email_type: Literal['modify_invoice_remind', 'send_excel'], 
-                       receiver: str = None, 
+                       receivers: str = None, 
                        attachment_paths: Optional[List[str]] = None):
 
         config = self.config[email_type]
@@ -61,7 +69,7 @@ def _prepare_email(self, email_type: Literal['modify_invoice_remind', 'send_exce
         message = MIMEMultipart()
         message["Subject"] = subject
         message["From"] = self.login
-        message["To"] = receiver
+        message["To"] = ', '.join(receivers)
         message.attach(MIMEText(body, "plain"))
 
         # Add attachments
@@ -112,13 +120,12 @@ def _attach_files(self, message, attachment_paths):
 
     email_sender.send_email(
         email_type="modify_invoice_remind",
-        receiver="[email protected]",
+        receivers=None,
         # attachment_paths=["output/Stdi_08_24.xlsx", "output/1.4437_10578_A3DS GmbH_04_2024 .xlsm"],  # List of file paths
     )
 
-
     email_sender.send_email(
         email_type="send_excel",
-        receiver="[email protected]",
+        receivers=["[email protected]"],
         attachment_paths=["output/Stdi_08_24.xlsx", "output/1.4437_10578_A3DS GmbH_04_2024 .xlsm", "notfoundfile.txt"],  # List of file paths
     )