(20241219) Safety Push.

utils_v2/goog/controllers/gmail/gmail_client.py

@@ -51,7 +51,7 @@ from utils_v2.mail import mail_parser
 from utils_v2.goog.controllers.base import AsyncGoogleBase
 from utils_v2.goog.models.auth_tokens import GoogleAuthTokens
 from utils_v2.goog.models.api_call import GoogleApiResponse
-from utils_v2.goog.gmail.gmail_message import GMailMessage
+from utils_v2.goog.controllers.gmail.gmail_message import GMailMessage
 
 # Related to Google:
 from google.auth.transport.requests import Request

utils_v2/goog/controllers/gmail/gmail_message.py

@@ -101,10 +101,10 @@ class GMailMessage:
     def __init__(
             self,
             from_email: str,
-            to_email: str,
+            to_email: str | List[str],
             subject: str,
-            cc_emails: List[str] = None,
-            bcc_emails: List[str] = None
+            cc_emails: str | List[str] = None,
+            bcc_emails: str | List[str] = None
     ):
 
         """
@@ -119,10 +119,10 @@ class GMailMessage:
         # Create the instance of the message:
         self.message = MIMEMultipart()
         self.message["From"] = from_email
-        self.message["To"] = to_email
+        self.message["To"] = ",".join(to_email if isinstance(to_email, list) else [to_email])
         self.message["Subject"] = subject
-        if cc_emails: self.message["CC"] = ",".join(cc_emails)
-        if bcc_emails: self.message["BCC"] = ",".join(bcc_emails)
+        if cc_emails: self.message["CC"] = ",".join(cc_emails if isinstance(cc_emails, list) else [cc_emails])
+        if bcc_emails: self.message["BCC"] = ",".join(bcc_emails if isinstance(bcc_emails, list) else [bcc_emails])
 
         # Note down the values for accessing later:
         self.__from = from_email
@@ -331,6 +331,6 @@ if __name__ == "__main__":
         """
     )
     my_mail.add_text("This is how you should pet her 👇")
-    my_mail.add_inline_image(r"../../../data/images/cat_petting.png")
-    my_mail.add_attachment(r"../../../data/pdf/sample_label.pdf")
-    print(my_mail.get_raw_message(as_base64 = True))
+    my_mail.add_inline_image(r"../../../../data/images/cat_petting.png")
+    my_mail.add_attachment(r"../../../../data/pdf/sample_label.pdf")
+    print(my_mail.get_raw_message(as_base64 = False))