From a4c1355bdd670fcb45dbf4b799e41f2f68b822f6 Mon Sep 17 00:00:00 2001
From: DiogoMarques <diogo.marques.29@infoportugal.pt>
Date: Mon, 6 Jul 2015 12:54:58 +0100
Subject: [PATCH] integration of wagtailsearch

---
 .../templatetags/modeltranslation.py                     | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/wagtail_modeltranslation/templatetags/modeltranslation.py b/wagtail_modeltranslation/templatetags/modeltranslation.py
index 25f226d..8acd238 100644
--- a/wagtail_modeltranslation/templatetags/modeltranslation.py
+++ b/wagtail_modeltranslation/templatetags/modeltranslation.py
@@ -28,5 +28,14 @@ def change_lang(context, lang=None, *args, **kwargs):
             activate(current_language)
 
             return translated_url
+        elif match.url_name == 'wagtailsearch_search':
+            path_components = [component for component in non_prefixed_path.split('/') if component]
+
+            translated_url = '/' + lang + '/' + path_components[0] + '/'
+            if request.GET:
+                translated_url += '?'
+                for key, value in request.GET.iteritems():
+                    translated_url += key + '=' + value
+            return translated_url
 
     return ''