mail_inline_css: unify premailer transform apply

pull/534/head
Simone Orsi 2019-12-02 09:46:50 +01:00
parent b6dc1918b4
commit 0d68d9dee4
1 changed files with 11 additions and 10 deletions

View File

@ -21,19 +21,20 @@ class MailTemplate(models.Model):
"""Use `premailer` to convert styles to inline styles.""" """Use `premailer` to convert styles to inline styles."""
result = super().generate_email(res_ids, fields=fields) result = super().generate_email(res_ids, fields=fields)
if isinstance(res_ids, int): if isinstance(res_ids, int):
premailer = Premailer( result['body_html'] = \
html=result['body_html'], self._premailer_apply_transform(result["body_html"])
**self._get_premailer_options(),
)
result['body_html'] = premailer.transform()
else: else:
for __, data in result.items(): for __, data in result.items():
premailer = Premailer( data['body_html'] = \
html=data['body_html'], self._premailer_apply_transform(data["body_html"])
**self._get_premailer_options(),
)
data['body_html'] = premailer.transform()
return result 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): def _get_premailer_options(self):
return {} return {}