add download message image

app/utils.py

@@ -32,10 +32,10 @@ def get_message_resource(client, message_id, file_key, file_type, save_dir):
     # 处理业务结果
 
     # 处理业务结果
+
     file_path = os.path.join(save_dir, file_key)
-    f = open(file_path, "wb")
+    with open(file_path, "wb") as f:
         f.write(response.file.read())
-    f.close()
 
     return file_path
 
@@ -45,7 +45,7 @@ def upload_media(client, file_path, parent_type, parent_node):
     _, filename = os.path.split(file_path)
     file_size = os.path.getsize(file_path)
 
-    file_bin = open(file_path)
+    with open(file_path) as f:
         request: UploadAllMediaRequest = (
             UploadAllMediaRequest.builder()
             .request_body(
@@ -54,13 +54,12 @@ def upload_media(client, file_path, parent_type, parent_node):
                 .parent_type(parent_type)
                 .parent_node(parent_node)
                 .size(file_size)
-            .file(file_bin)
+                .file(f)
                 .build()
             )
             .build()
         )
 
-    file_bin.close()
 
         # 发起请求
         option = (
@@ -68,7 +67,9 @@ def upload_media(client, file_path, parent_type, parent_node):
             .user_access_token("u-dF0r5rD5BcUFhORI38zmlk1l61TBghPzry00gl.82Haz")
             .build()
         )
-    response: UploadAllMediaResponse = client.drive.v1.media.upload_all(request, option)
+        response: UploadAllMediaResponse = client.drive.v1.media.upload_all(
+            request, option
+        )
 
         # 处理失败返回
         if not response.success():