From 0d68d9dee422befd2e3c1f48d038e1476a3c56ea Mon Sep 17 00:00:00 2001
From: Simone Orsi <simahawk@gmail.com>
Date: Mon, 2 Dec 2019 09:46:50 +0100
Subject: [PATCH] mail_inline_css: unify premailer transform apply

---
 mail_inline_css/models/mail_template.py | 21 +++++++++++----------
 1 file changed, 11 insertions(+), 10 deletions(-)

diff --git a/mail_inline_css/models/mail_template.py b/mail_inline_css/models/mail_template.py
index 647fbdf79..82fdba83d 100644
--- a/mail_inline_css/models/mail_template.py
+++ b/mail_inline_css/models/mail_template.py
@@ -21,19 +21,20 @@ class MailTemplate(models.Model):
         """Use `premailer` to convert styles to inline styles."""
         result = super().generate_email(res_ids, fields=fields)
         if isinstance(res_ids, int):
-            premailer = Premailer(
-                html=result['body_html'],
-                **self._get_premailer_options(),
-            )
-            result['body_html'] = premailer.transform()
+            result['body_html'] = \
+                self._premailer_apply_transform(result["body_html"])
         else:
             for __, data in result.items():
-                premailer = Premailer(
-                    html=data['body_html'],
-                    **self._get_premailer_options(),
-                )
-                data['body_html'] = premailer.transform()
+                data['body_html'] = \
+                    self._premailer_apply_transform(data["body_html"])
         return result
 
+    def _premailer_apply_transform(self, data_html):
+        premailer = Premailer(
+            html=data_html,
+            **self._get_premailer_options(),
+        )
+        return premailer.transform()
+
     def _get_premailer_options(self):
         return {}