feature(#98): updated send email api

Brain/src/rising_plugin/gmail/email_plugin.py

@@ -6,6 +6,7 @@ import os
 import re
 import smtplib
 import time
+import base64
 from email.header import decode_header
 from email.message import EmailMessage
 
@@ -15,7 +16,7 @@ from bs4 import BeautifulSoup
 EMAIL_SMTP_HOST = "smtp.gmail.com"
 EMAIL_SMTP_PORT = 587
 EMAIL_IMAP_SERVER = "imap.gmail.com"
-EMAIL_SIGNATURE = "This was sent by Brain Service"
+EMAIL_SIGNATURE = "This was sent by Rising Brain"
 
 
 class EmailPlugin:
@@ -357,3 +358,10 @@ class EmailPlugin:
         email_body = re.sub(r"http\S+", "", email_body)
 
         return email_body
+
+    def write_attachment(self, filename: str, file_content: str) -> str:
+        file_content = base64.b64decode(file_content).decode("utf-8")
+        file = open(filename, "w")
+        file.write(file_content)
+        file.close()
+        return filename

Brain/src/router/email_router.py

@@ -119,6 +119,10 @@ def construct_blueprint_email_api() -> APIRouter:
                     to_send=data.data.to_send,
                 )
             else:
+                file_name = email_manager.write_attachment(
+                    filename=data.data.filename, file_content=data.data.file_content
+                )
+
                 result += email_manager.send_email_with_attachment(
                     sender=data.data.sender,
                     pwd=data.data.pwd,
@@ -126,8 +130,9 @@ def construct_blueprint_email_api() -> APIRouter:
                     subject=data.data.subject,
                     body=data.data.body,
                     to_send=data.data.to_send,
-                    filename=data.data.filename,
+                    filename=file_name,
                 )
+
         except Exception as e:
             if isinstance(e, BrainException):
                 return e.get_response_exp()

Brain/tests/functional/test_api.py

@@ -235,3 +235,64 @@ def test_train_contacts(body):
 def test_delete_data(body):
     response = client.post("/auto_task/delete", json=body)
     assert response.status_code == 200
+
+
+@pytest.mark.parametrize(
+    "body",
+    [
+        (
+            {
+                "data": {
+                    "sender": "test@gmail.com",
+                    "pwd": "use app password of your google account",
+                    "imap_folder": "inbox or drafts",
+                },
+                "confs": {
+                    "token": "test_token",
+                    "uuid": "test-uuid",
+                    "openai_key": "",
+                    "pinecone_key": "",
+                    "pinecone_env": "",
+                    "firebase_key": "",
+                    "settings": {"temperature": 0.6},
+                },
+            }
+        )
+    ],
+)
+def test_read_emails(body):
+    response = client.post("/email/read_emails", json=body)
+    assert response.status_code == 200
+
+
+@pytest.mark.parametrize(
+    "body",
+    [
+        (
+            {
+                "data": {
+                    "sender": "testsender@gmail.com",
+                    "pwd": "use app password of your google account",
+                    "to": "testto@gmail.com",
+                    "subject": "title of email",
+                    "body": "email content",
+                    "to_send": "true or false whether send to inbox or drafts",
+                    "filename": "test.txt",
+                    "file_content": "string encoded with base64",
+                },
+                "confs": {
+                    "token": "test_token",
+                    "uuid": "test-uuid",
+                    "openai_key": "",
+                    "pinecone_key": "",
+                    "pinecone_env": "",
+                    "firebase_key": "",
+                    "settings": {"temperature": 0.6},
+                },
+            }
+        )
+    ],
+)
+def test_send_email(body):
+    response = client.post("/email/send_email", json=body)
+    assert response.status_code == 200