Merge pull request #208 from ttt246/feature/email_attach_194

Brain/src/rising_plugin/gmail/email_plugin.py

@@ -219,12 +219,12 @@ class EmailPlugin:
 
                     messages.append(
                         {
-                            "From": from_address,
-                            "To": to_address,
-                            "Date": date,
-                            "CC": cc,
-                            "Subject": subject,
-                            "Message Body": body,
+                            "from": from_address,
+                            "to": to_address,
+                            "date": date,
+                            "cc": cc,
+                            "subject": subject,
+                            "body": body,
                         }
                     )
 
@@ -232,12 +232,12 @@ class EmailPlugin:
         if not messages:
             messages.append(
                 {
-                    "From": "",
-                    "To": "",
-                    "Date": "",
-                    "CC": "",
-                    "Subject": "",
-                    "Message Body": "There are no Emails",
+                    "from": "",
+                    "to": "",
+                    "date": "",
+                    "cc": "",
+                    "subject": "",
+                    "body": "There are no Emails",
                 }
             )
             return json.dumps(messages)
@@ -359,9 +359,16 @@ class EmailPlugin:
 
         return email_body
 
-    def write_attachment(self, filename: str, file_content: str) -> str:
+    def write_attachment(self, filename: str, file_content: str) -> (str, str):
+        # create folder for temporarily saving attached file
+        milliseconds = int(time.time() * 1000)
+        file_path = f"Brain/assets/{milliseconds}/{filename}"
+        file_directory = f"Brain/assets/{milliseconds}"
+        os.mkdir(file_directory)
+
+        # file write
         file_content = base64.b64decode(file_content).decode("utf-8")
-        file = open(filename, "w")
+        file = open(file_path, "w")
         file.write(file_content)
         file.close()
-        return filename
+        return file_path, file_directory

Brain/src/router/email_router.py

@@ -1,5 +1,5 @@
 import json
-
+import shutil
 from fastapi import APIRouter
 
 from Brain.src.common.assembler import Assembler
@@ -30,12 +30,12 @@ def construct_blueprint_email_api() -> APIRouter:
 
         @generator.response(
             status_code=200, schema={"message": "message", "result": [{
-                "From": "testfrom@test.com",
-                "To": "test@gmail.com",
-                "Date": "Tue, 04 Jul 2023 12:55:19 +0000",
-                "CC": "",
-                "Subject": "subject",
-                "Message Body": "message"
+                "from": "testfrom@test.com",
+                "to": "test@gmail.com",
+                "date": "Tue, 04 Jul 2023 12:55:19 +0000",
+                "cc": "",
+                "subject": "subject",
+                "body": "message"
             }]}
         )
 
@@ -119,7 +119,7 @@ def construct_blueprint_email_api() -> APIRouter:
                     to_send=data.data.to_send,
                 )
             else:
-                file_name = email_manager.write_attachment(
+                file_name, file_directory = email_manager.write_attachment(
                     filename=data.data.filename, file_content=data.data.file_content
                 )
 
@@ -133,6 +133,8 @@ def construct_blueprint_email_api() -> APIRouter:
                     filename=file_name,
                 )
 
+                shutil.rmtree(file_directory)
+
         except Exception as e:
             if isinstance(e, BrainException):
                 return e.get_response_exp()