From d076cb458794e1c2cfda8d72f0ac89ed698cbdcb Mon Sep 17 00:00:00 2001
From: Oleg Lavrovsky <oleg@utou.ch>
Date: Wed, 26 Apr 2017 15:51:41 +0200
Subject: [PATCH] Smarter language checker

---
 publichealth/home/templatetags/navigation.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/publichealth/home/templatetags/navigation.py b/publichealth/home/templatetags/navigation.py
index 523f310..406b4ed 100644
--- a/publichealth/home/templatetags/navigation.py
+++ b/publichealth/home/templatetags/navigation.py
@@ -7,11 +7,14 @@ register = template.Library()
 # Language switcher
 @register.inclusion_tag('tags/language.html', takes_context=True)
 def language_switcher(context):
-    url = context['page'].url
+    url = context['page'].url.split('/')
+    if len(url) > 2 and len(url[1]) == 2:
+        url[1] = '$lang$'
+        url = '/'.join(url)
     return {
         'languages': [
-            { 'code': 'de', 'title': 'De', 'url': url.replace('/fr/','/de/') },
-            { 'code': 'fr', 'title': 'Fr', 'url': url.replace('/de/','/fr/') }
+            { 'code': 'de', 'title': 'De', 'url': url.replace('$lang$','de') },
+            { 'code': 'fr', 'title': 'Fr', 'url': url.replace('$lang$','fr') }
         ],
         'currentlangcode': translation.get_language(),
         'request': context['request'],