Merge pull request #1 from MattyBoy4444/MattyBoy4444-pngsupport-1

Added support for PNG's.
MattyBoy4444 · Jan 16, 2019 · 73fe5cd · 73fe5cd
2 parents 2a4a23b + 9101a53
commit 73fe5cd
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/titanium_backup_formatter.py b/titanium_backup_formatter.py
@@ -134,8 +134,12 @@ def create_output_file(self, conversations, self_gaia_id, output_file_name):
                                         if attachment.original_content_url is not None:
                                             data = self._convert_url_to_base64_data(attachment.original_content_url)
                                         if data is not None:
-                                            message_string += MMS_PART.format("image/jpeg", order, "base64", data)
-                                            order += 1
+                                            if ".png" in attachment.download_url:
+                                                message_string += MMS_PART.format("image/png", order, "base64", data)
+                                                order += 1
+                                            else:
+                                                message_string += MMS_PART.format("image/jpeg", order, "base64", data)
+                                                order += 1
                                         else:
                                             print("Error: unable to download image data!")
                                     elif attachment.media_type == "ANIMATED_PHOTO":