feature(#194): create folder for saving attach file and delete it after sending email...

Brain/src/rising_plugin/gmail/email_plugin.py

@@ -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
@@ -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()