From 6bfb6d934d217d338508bfea850a0107d6d90a11 Mon Sep 17 00:00:00 2001
From: Fabio Manganiello <blacklight86@gmail.com>
Date: Sun, 28 Jan 2018 02:59:19 +0100
Subject: [PATCH] Support for multiple attachments and binary application
 attachments in GMail

---
 platypush/plugins/google/mail.py | 43 +++++++++++++++++++-------------
 1 file changed, 25 insertions(+), 18 deletions(-)

diff --git a/platypush/plugins/google/mail.py b/platypush/plugins/google/mail.py
index 59489a64a6..619f68f93c 100644
--- a/platypush/plugins/google/mail.py
+++ b/platypush/plugins/google/mail.py
@@ -5,8 +5,10 @@ import os
 
 from apiclient import discovery
 
+from email.encoders import encode_base64
+from email.mime.application import MIMEApplication
 from email.mime.audio import MIMEAudio
-from email.mime.base import MIMEBase
+# from email.mime.base import MIMEBase
 from email.mime.image import MIMEImage
 from email.mime.multipart import MIMEMultipart
 from email.mime.text import MIMEText
@@ -22,35 +24,40 @@ class GoogleMailPlugin(GooglePlugin):
         super().__init__(scopes=self.scopes, *args, **kwargs)
 
 
-    def compose(self, sender, to, subject, body, file=None):
-        message = MIMEMultipart() if file else MIMEText(body)
+    def compose(self, sender, to, subject, body, files=None):
+        message = MIMEMultipart() if files else MIMEText(body)
         message['to'] = to
         message['from'] = sender
         message['subject'] = subject
 
-        if file:
-            msg = MIMEText(body)
-            message.attach(msg)
-            content_type, encoding = mimetypes.guess_type(file)
+        if files:
+            for file in files:
+                msg = MIMEText(body)
+                message.attach(msg)
+                content_type, encoding = mimetypes.guess_type(file)
 
-            if content_type is None or encoding is not None:
-                content_type = 'application/octet-stream'
-            main_type, sub_type = content_type.split('/', 1)
+                if content_type is None or encoding is not None:
+                    content_type = 'application/octet-stream'
+
+                main_type, sub_type = content_type.split('/', 1)
+                with open(file, 'rb') as fp: content = fp.read()
 
-            with open(file, 'rb') as fp:
                 if main_type == 'text':
-                    msg = mimetypes.MIMEText(fp.read(), _subtype=sub_type)
+                    msg = mimetypes.MIMEText(content, _subtype=sub_type)
                 elif main_type == 'image':
-                    msg = MIMEImage(fp.read(), _subtype=sub_type)
+                    msg = MIMEImage(content, _subtype=sub_type)
                 elif main_type == 'audio':
-                    msg = MIMEAudio(fp.read(), _subtype=sub_type)
+                    msg = MIMEAudio(content, _subtype=sub_type)
+                elif main_type == 'application':
+                    msg = MIMEApplication(content, _subtype=sub_type,
+                                          _encoder=encode_base64)
                 else:
                     msg = MIMEBase(main_type, sub_type)
-                    msg.set_payload(fp.read())
+                    msg.set_payload(content)
 
-            filename = os.path.basename(file)
-            msg.add_header('Content-Disposition', 'attachment', filename=filename)
-            message.attach(msg)
+                filename = os.path.basename(file)
+                msg.add_header('Content-Disposition', 'attachment', filename=filename)
+                message.attach(msg)
 
         service = self._get_service()
         body = { 'raw': base64.urlsafe_b64encode(message.as_bytes()).decode() }