Update downloads supernav page for Python install manager

Author: zooba

downloads/models.py

@@ -158,24 +158,29 @@ def update_supernav():
     if not latest_python3:
         return
 
+    try:
+        latest_pymanager = Release.objects.latest_pymanager()
+    except Release.DoesNotExist:
+        latest_pymanager = None
+
     python_files = []
     for o in OS.objects.all():
         data = {
             'os': o,
             'python3': None,
+            'pymanager': None,
         }
 
-        release_file = latest_python3.download_file_for_os(o.slug)
-        if not release_file:
-            continue
-        data['python3'] = release_file
+        data['python3'] = latest_python3.download_file_for_os(o.slug)
+        if latest_pymanager:
+            data['pymanager'] = latest_pymanager.download_file_for_os(o.slug)
 
         python_files.append(data)
 
     if not python_files:
         return
 
-    if not all(f['python3'] for f in python_files):
+    if not all(f['python3'] or f['pymanager'] for f in python_files):
         # We have a latest Python release, different OSes, but don't have release
         # files for the release, so return early.
         return
@@ -287,6 +292,7 @@ def purge_fastly_download_pages(sender, instance, **kwargs):
         purge_url('/downloads/feed.rss')
         purge_url('/downloads/latest/python2/')
         purge_url('/downloads/latest/python3/')
+        purge_url('/downloads/latest/pymanager/')
         purge_url('/downloads/macos/')
         purge_url('/downloads/source/')
         purge_url('/downloads/windows/')
@@ -308,9 +314,7 @@ def update_download_supernav_and_boxes(sender, instance, **kwargs):
         return
 
     if instance.is_published:
-        # Supernav only has download buttons for Python 3.
-        if instance.version == instance.PYTHON3:
-            update_supernav()
+        update_supernav()
         update_download_landing_sources_box()
         update_homepage_download_box()
 

downloads/tests/test_models.py

@@ -1,4 +1,4 @@
-from ..models import Release
+from ..models import Release, ReleaseFile
 from .base import BaseDownloadTests
 
 
@@ -95,3 +95,51 @@ def test_is_version_at_least_with_invalid_name(self):
         self.assertFalse(invalid_release.is_version_at_least_3_5)
         self.assertFalse(invalid_release.is_version_at_least_3_9)
         self.assertFalse(invalid_release.is_version_at_least_3_14)
+
+    def test_update_supernav(self):
+        from ..models import update_supernav
+        from boxes.models import Box
+
+        release = Release.objects.create(
+            name='Python install manager 25.0',
+            version=Release.PYMANAGER,
+        )
+
+        for os, url in [
+            (self.windows, 'python-3.10-windows.msi'),
+            (self.osx, 'python-3.10-macos.pkg'),
+            (self.linux, 'python-3.10-linux.tar.gz'),
+        ]:
+            ReleaseFile.objects.create(
+                os=os,
+                release=self.python_3,
+                name='Python 3.10',
+                url='/ftp/python/{}'.format(url),
+                download_butter=True,
+            )
+
+        update_supernav()
+
+        content = Box.get(label='download-sources').content
+        self.assertIn('class="download-os-windows"', content)
+        self.assertNotIn('pymanager-25.0.msix', content)
+        self.assertIn('python-3.10-windows.msi', content)
+        self.assertIn('class="download-os-macos"', content)
+        self.assertIn('python-3.10-macos.pkg', content)
+        self.assertIn('class="download-os-linux"', content)
+        self.assertIn('python-3.10-linux.tar.gz', content)
+
+        ReleaseFile.objects.create(
+            os=self.windows,
+            release=release,
+            name='MSIX',
+            url='/ftp/python/pymanager/pymanager-25.0.msix',
+            download_button=True,
+        )
+
+        update_supernav()
+
+        content = Box.get(label='download-sources').content
+        self.assertIn('class="download-os-windows"', content)
+        self.assertIn('pymanager-25.0.msix', content)
+        self.assertIn('python-3.10-windows.msi', content)

templates/downloads/supernav.html

@@ -10,8 +10,10 @@ <h4>Download for {{ data.os.name }}</h4>
 
         {% if data.pymanager %}
         <p><a class="button" href="{{ data.pymanager.url }}">Python install manager</a></p>
+        {% if data.python3 %}
         <p>Or get the standalone installer for <a class="button" href="{{ data.python3.url }}">{{ data.python3.release.name }}</a></p>
-        {% else %}
+        {% endif %}
+        {% elif data.python3 %}
         <p>
             <a class="button" href="{{ data.python3.url }}">{{ data.python3.release.name }}</a>
         </p>