Clean session nav_menu cache when switch the language.

sshwsfc · sshwsfc · commit 614439db811d · 2016-09-21T21:55:29.000+08:00
diff --git a/xadmin/plugins/language.py b/xadmin/plugins/language.py
@@ -1,9 +1,10 @@
 
 from django.conf import settings
 from django.template import loader
+from django.views.i18n import set_language
 from xadmin.plugins.utils import get_context_dict
 from xadmin.sites import site
-from xadmin.views import BaseAdminPlugin, CommAdminView
+from xadmin.views import BaseAdminPlugin, CommAdminView, BaseAdminView
 
 
 class SetLangNavPlugin(BaseAdminPlugin):
@@ -13,6 +14,13 @@ def block_top_navmenu(self, context, nodes):
         context['redirect_to'] = self.request.get_full_path()
         nodes.append(loader.render_to_string('xadmin/blocks/comm.top.setlang.html', context=context))
 
+class SetLangView(BaseAdminView):
+
+    def post(self, request, *args, **kwargs):
+        if 'nav_menu' in request.session:
+            del request.session['nav_menu']
+        return set_language(request)
+
 if settings.LANGUAGES and 'django.middleware.locale.LocaleMiddleware' in settings.MIDDLEWARE_CLASSES:
     site.register_plugin(SetLangNavPlugin, CommAdminView)
-    site.register_view(r'^i18n/', lambda site: 'django.conf.urls.i18n', 'i18n')
+    site.register_view(r'^i18n/setlang/$', SetLangView, 'set_language')
