apps/catalogue/templatetags/flat_lang.py

   1 # -*- coding: utf-8 -*-
   2 #
   3 # This file is part of MIL/PEER, licensed under GNU Affero GPLv3 or later.
   4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   5 #
   6 from django.core.exceptions import ObjectDoesNotExist, MultipleObjectsReturned
   7 from django.utils import translation
   8
   9 from django import template
  10
  11 register = template.Library()
  12
  13
  14 @register.assignment_tag
  15 def flat_lang(page):
  16     try:
  17         return type(page).objects.get(url="%s%s/" % (page.url, translation.get_language()))
  18     except (ObjectDoesNotExist, MultipleObjectsReturned):
  19         return page