Category snippet localization

network-api/networkapi/settings.py

@@ -339,6 +339,7 @@
                 'blog_tags': 'networkapi.wagtailpages.templatetags.blog_tags',
                 'card_tags': 'networkapi.wagtailpages.templatetags.card_tags',
                 'class_tags': 'networkapi.wagtailpages.templatetags.class_tags',
+                'debug_tags': 'networkapi.wagtailpages.templatetags.debug_tags',
                 'homepage_tags': 'networkapi.wagtailpages.templatetags.homepage_tags',
                 'localization': 'networkapi.wagtailpages.templatetags.localization',
                 'mini_site_tags': 'networkapi.wagtailpages.templatetags.mini_site_tags',

network-api/networkapi/wagtailpages/pagemodels/products.py

@@ -45,6 +45,8 @@
     ('Bad', 'Bad')
 ]
 
+DEFAULT_LOCALE_ID = Locale.objects.get(language_code=settings.LANGUAGE_CODE).id
+
 
 def get_product_subset(cutoff_date, authenticated, key, products, language_code='en'):
     """
@@ -55,7 +57,7 @@ def get_product_subset(cutoff_date, authenticated, key, products, language_code=
     """
     products = products.filter(
         review_date__gte=cutoff_date,
-        locale=Locale.objects.get(language_code=language_code)
+        # locale=Locale.objects.get(language_code=language_code)
     )
     if not authenticated:
         products = products.live()
@@ -118,13 +120,12 @@ class BuyersGuideProductCategory(TranslatableMixin, models.Model):
     ]
 
     @property
-    def published_product_page_count(self):
-        return ProductPage.objects.filter(product_categories__category=self).live().count()
+    def alias_of(self):
+        return BuyersGuideProductCategory.objects.filter(translation_key=self.translation_key).order_by('locale_id').first()
 
     @property
-    def published_product_count(self):
-        # TODO: REMOVE: LEGACY FUNCTION
-        return ProductPage.objects.filter(product_category=self, draft=False).count()
+    def published_product_page_count(self):
+        return ProductPage.objects.filter(product_categories__category=self).live().count()
 
     def __str__(self):
         return self.name
@@ -1291,7 +1292,9 @@ class Meta:
 
 
 class ExcludedCategories(TranslatableMixin, Orderable):
-    """This allows us to select one or more blog authors from Snippets."""
+    """
+    This allows us to filter categories from showing up on the PNI site
+    """
 
     page = ParentalKey("wagtailpages.BuyersGuidePage", related_name="excluded_categories")
     category = models.ForeignKey(
@@ -1449,31 +1452,46 @@ def product_view(self, request, slug):
     def categories_page(self, request, slug):
         context = self.get_context(request, bypass_products=True)
         language_code = self.get_language_code(request)
+        locale_id = Locale.objects.get(language_code=language_code).id
         slug = slugify(slug)
 
-        # If getting by slug fails, also try to get it by name.
+        # because we may be working with localized content, and the slug
+        # will always be our english slug, we need to find the english
+        # category first, and then find its corresponding localized version
         try:
-            category = BuyersGuideProductCategory.objects.get(slug=slug)
+            original_category = BuyersGuideProductCategory.objects.get(slug=slug, locale_id=DEFAULT_LOCALE_ID)
         except BuyersGuideProductCategory.DoesNotExist:
-            category = get_object_or_404(BuyersGuideProductCategory, name__iexact=slug)
+            original_category = get_object_or_404(BuyersGuideProductCategory, name__iexact=slug)
+
+        if locale_id != DEFAULT_LOCALE_ID:
+            try:
+                category = BuyersGuideProductCategory.objects.get(
+                    translation_key=original_category.translation_key,
+                    locale_id=DEFAULT_LOCALE_ID,
+                )
+            except BuyersGuideProductCategory.DoesNotExist:
+                category = original_category
+        else:
+            category = original_category
 
         authenticated = request.user.is_authenticated
         key = f'cat_product_dicts_{slug}_auth' if authenticated else f'cat_product_dicts_{slug}_live'
         key = f'{language_code}_{key}'
-        products = cache.get(key)
+        products = None # cache.get(key)
         exclude_cat_ids = [excats.category.id for excats in self.excluded_categories.all()]
 
         if products is None:
             products = get_product_subset(
                 self.cutoff_date,
                 authenticated,
                 key,
-                ProductPage.objects.filter(product_categories__category__in=[category])
+                ProductPage.objects.filter(product_categories__category__in=[original_category])
                                    .exclude(product_categories__category__id__in=exclude_cat_ids),
                 language_code=language_code
             )
 
-        context['category'] = category.slug
+        context['category'] = slug
+        context['current_category'] = category
         context['products'] = products
         context['pageTitle'] = f'{category} | ' + gettext("Privacy & security guide") + ' | Mozilla Foundation'
         context['template_cache_key_fragment'] = f'{category.slug}_{request.LANGUAGE_CODE}'

network-api/networkapi/wagtailpages/templates/buyersguide/bg_base.html

@@ -1,6 +1,6 @@
 {% extends "pages/base.html" %}
 
-{% load bg_nav_tags localization i18n static wagtailroutablepage_tags wagtailmetadata_tags %}
+{% load bg_nav_tags localization i18n static wagtailroutablepage_tags wagtailmetadata_tags debug_tags %}
 
 {% get_current_language as lang_code %}
 
@@ -85,15 +85,18 @@
       <div class="row">
         <div class="col">
           {% if pagetype == "product" or pagetype == "about" %}
-            <a class="multipage-link active" href="{{ page.url }}">{% trans "All" %}</a>
+            <a class="multipage-link active" href="{% relocalized_url page.localized.url lang_code %}">{% trans "All" %}</a>
           {% else %}
-            <a class="multipage-link {% if not category %} active{% endif %}" href="{{ page.url }}">{% trans "All" %}</a>
+            <a class="multipage-link {% if not category %} active{% endif %}" href="{% relocalized_url page.localized.url lang_code %}">{% trans "All" %}</a>
           {% endif %}
+
           {% for cat in categories %}
-            {% if cat.published_product_page_count > 0 %}
-              {% routablepageurl home_page 'category-view' cat.slug as cat_url %}
-              <a class="multipage-link {% check_active_category current_category cat %}{% if cat.featured is True %} featured{% endif %}{% if category == cat.slug %} active{% endif %}" href="{{ cat_url }}" data-name="{{ cat.name }}">{{ cat.name }}</a>
+            {% with original=cat.alias_of %}
+            {% if original.published_product_page_count > 0 %}
+              {% localizedroutablepageurl home_page 'category-view' lang_code original.slug as cat_url %}
+              <a class="multipage-link {% check_active_category current_category cat %}{% if original.featured is True %} featured{% endif %}" href="{{ cat_url }}" data-name="{{ cat.name }}">{{ cat.name }}</a>
             {% endif %}
+            {% endwith %}
           {% endfor %}
         </div>
       </div>

network-api/networkapi/wagtailpages/templates/buyersguide/fragments/category_nav_links.html

@@ -5,13 +5,15 @@
 {% endcomment %}
 
 {% for cat in sorted_categories %}
+  {% with original=cat.alias_of %}
   {% if cat.hidden == False and cat.published_product_page_count > 0 %}
-    {% url 'category-view' cat.slug as cat_url %}
+    {% localizedroutablepageurl home_page 'category-view' lang_code original.slug as cat_url %}
     <a
-      class="multipage-link {% check_active_category current_category cat %}{% if cat.featured is True %} featured{% endif %}"
+      class="multipage-link {% check_active_category current_category cat %}{% if original.featured is True %} featured{% endif %}"
       href="{{ cat_url }}"
       data-name="{{ cat.name }}">
       {{ cat.name }}
     </a>
   {% endif %}
+  {% endwith %}
 {% endfor %}

network-api/networkapi/wagtailpages/templatetags/bg_nav_tags.py

@@ -7,6 +7,10 @@
 # Determine if a category nav link should be marked active
 @register.simple_tag(name='check_active_category')
 def check_active_category(current_category, target_category):
+    # because we're working with potentially localized data,
+    # make sure to compare the linguistic originals.
+    current_category = getattr(current_category, 'alias_of', current_category)
+    target_category = getattr(target_category, 'alias_of', target_category)
     return 'active' if current_category == target_category else ''
 
 
@@ -15,7 +19,7 @@ def check_active_category(current_category, target_category):
 def bg_active_nav(current, target):
     return 'active' if urlparse(current).path == urlparse(target).path else ''
 
-
+"""
 # Instantiate a list of category page links based on the current page's relation to them
 # NOTE: this points to the new, namespaced category_nav_links. If we need to revert to the old app, change this back.
 @register.inclusion_tag('buyersguide/fragments/category_nav_links.html', takes_context=True)
@@ -25,3 +29,4 @@ def category_nav(context, current_url, current_category, all_categories):
         'current_category': current_category,
         'sorted_categories': all_categories.order_by('-featured', 'sort_order', 'name'),  # featured categories first
     }
+"""

network-api/networkapi/wagtailpages/templatetags/debug_tags.py

@@ -0,0 +1,12 @@
+from django.template import Library
+from django.conf import settings
+register = Library()
+
+if settings.DEBUG:
+    print('→ Running in DEBUG mode: enabling debug template tag "inspect_object" in tag collection "debug_tags".\n')
+
+    @register.simple_tag
+    def inspect_object(instance, prefix_label=""):
+        output = str(dir(instance)).replace(', ', ',\n')
+        prefix_label = str(prefix_label)
+        return f'<pre>{prefix_label} {output}</pre>'

network-api/networkapi/wagtailpages/templatetags/localization.py

@@ -4,11 +4,10 @@
 from django import template
 from django.conf import settings
 from django.utils.translation import get_language_info
+from wagtail.contrib.routable_page.templatetags.wagtailroutablepage_tags import routablepageurl
 
 register = template.Library()
 
-DEFAULT_LOCALE = 'en_US'
-
 mappings = {
     'en': 'en_US',
     'de': 'de_DE',
@@ -20,6 +19,9 @@
     'pt-BR': 'pt_BR',
 }
 
+DEFAULT_LOCALE_CODE = settings.LANGUAGE_CODE
+DEFAULT_LOCALE = mappings.get(DEFAULT_LOCALE_CODE)
+
 
 # This filter turns Wagtail language codes into OpenGraph locale strings
 @register.filter
@@ -49,3 +51,20 @@ def get_local_language_names():
 @register.simple_tag()
 def get_unlocalized_url(page, locale):
     return page.get_url().replace(f'/{locale}/', '/', 1)
+
+# Gorce-relocalize a URL
+@register.simple_tag()
+def relocalized_url(url, locale_code):
+    if locale_code == DEFAULT_LOCALE_CODE:
+        return url
+    return url.replace(f'/{DEFAULT_LOCALE_CODE}/', f'/{locale_code}/')
+
+
+# Overcome a limitation of the routablepageurl tag
+@register.simple_tag(takes_context=True)
+def localizedroutablepageurl(context, page, url_name, locale_code, *args, **kwargs):
+    url = relocalized_url(
+        routablepageurl(context, page, url_name, *args, **kwargs),
+        locale_code,
+    )
+    return url