Fix language switching bug

molgen · Sep 14, 2019 · fb4848d · fb4848d
1 parent 16f1bf7
commit fb4848d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/mpicms/base/templatetags/base_tags.py b/mpicms/base/templatetags/base_tags.py
@@ -18,4 +18,11 @@ def is_subscribed(page, user):
 
 @register.simple_tag
 def room_link(room):
-    return get_room_link(room)
+    return get_room_link(room)
+
+
+@register.filter
+def remove_i18n(url):
+    if url.startswith('/en') or url.startswith('/de'):
+        return url[3:] 
+    return url
diff --git a/mpicms/templates/menus/main.html b/mpicms/templates/menus/main.html
@@ -1,4 +1,4 @@
-{% load static i18n wagtailcore_tags %}
+{% load static i18n wagtailcore_tags base_tags %}
 
 {% get_current_language as LANGUAGE_CODE %}
 
@@ -37,6 +37,7 @@
               {% else %}
                 <form action="{% url 'set_language' %}" method="post" id=language_form>
                   {% csrf_token %}
+                  <input type="hidden" name="next" value="{{ page.get_url|remove_i18n }}" />
                   <select name="language" onchange="language_form.submit()">
               {% endif %}
                     {% get_available_languages as LANGUAGES %}