From 08bb95422bd43ce729992673b27117088f5454d9 Mon Sep 17 00:00:00 2001
From: zorun <github@bitsofnetworks.org>
Date: Sun, 26 Apr 2020 23:12:33 +0200
Subject: [PATCH] Fix crash when a localized email template is missing (#592)

---
 ihatemoney/utils.py | 17 ++++++++++++++++-
 ihatemoney/web.py   | 16 +++++++---------
 2 files changed, 23 insertions(+), 10 deletions(-)

diff --git a/ihatemoney/utils.py b/ihatemoney/utils.py
index c8daa567..7fdad61a 100644
--- a/ihatemoney/utils.py
+++ b/ihatemoney/utils.py
@@ -9,7 +9,8 @@ import os
 import re
 
 from babel import Locale
-from flask import current_app, redirect
+from flask import current_app, redirect, render_template
+from flask_babel import get_locale
 import jinja2
 from werkzeug.routing import HTTPException, RoutingException
 
@@ -278,3 +279,17 @@ class FormEnum(Enum):
 
     def __str__(self):
         return str(self.value)
+
+
+def render_localized_template(template_name_prefix, **context):
+    """Like render_template(), but selects the right template according to the
+    current user language.  Fallback to English if a template for the
+    current language does not exist.
+    """
+    fallback = "en"
+    templates = [
+        f"{template_name_prefix}.{lang}.j2"
+        for lang in (get_locale().language, fallback)
+    ]
+    # render_template() supports a list of templates to try in order
+    return render_template(templates, **context)
diff --git a/ihatemoney/web.py b/ihatemoney/web.py
index d799420b..b5998c68 100644
--- a/ihatemoney/web.py
+++ b/ihatemoney/web.py
@@ -30,7 +30,7 @@ from flask import (
     session,
     url_for,
 )
-from flask_babel import get_locale, gettext as _
+from flask_babel import gettext as _
 from flask_mail import Message
 from sqlalchemy import orm
 from sqlalchemy_continuum import Operation
@@ -57,6 +57,7 @@ from ihatemoney.utils import (
     get_members,
     list_of_dicts2csv,
     list_of_dicts2json,
+    render_localized_template,
     same_bill,
 )
 
@@ -301,7 +302,7 @@ def create_project():
                 project=g.project.name,
             )
 
-            message_body = render_template(f"reminder_mail.{get_locale().language}.j2")
+            message_body = render_localized_template("reminder_mail")
 
             msg = Message(
                 message_title, body=message_body, recipients=[project.contact_email]
@@ -335,11 +336,12 @@ def remind_password():
             # get the project
             project = Project.query.get(form.id.data)
             # send a link to reset the password
-            password_reminder = f"password_reminder.{get_locale().language}.j2"
             current_app.mail.send(
                 Message(
                     "password recovery",
-                    body=render_template(password_reminder, project=project),
+                    body=render_localized_template(
+                        "password_reminder", project=project
+                    ),
                     recipients=[project.contact_email],
                 )
             )
@@ -566,11 +568,7 @@ def invite():
     if request.method == "POST":
         if form.validate():
             # send the email
-
-            message_body = render_template(
-                f"invitation_mail.{get_locale().language}.j2"
-            )
-
+            message_body = render_localized_template("invitation_mail")
             message_title = _(
                 "You have been invited to share your " "expenses for %(project)s",
                 project=g.project.name,