Final mongodb edition.

Author: mikeckennedy

app/ch16_mongodb/final/pypi_org/services/package_service.py

@@ -35,3 +35,11 @@ def get_package_by_id(package_id: str) -> Optional[Package]:
 
 def all_packages(limit: int) -> List[Package]:
     return list(Package.objects().limit(limit))
+
+
+def get_packages_by_ids(package_ids: List[str]) -> List[Package]:
+    return list(Package.objects(id__in=package_ids))
+
+
+def get_latest_release_for_package(package_id: str) -> Optional[Release]:
+    return Release.objects(package_id=package_id).order_by('-created_date').first()

app/ch16_mongodb/final/pypi_org/templates/home/index.html

@@ -28,10 +28,11 @@ <h2>New releases</h2>
                 </div>
 
                 {% for r in releases %}
+                    {% set package = package_lookup.get(r.package_id) %}
                     <div class="project">
                         <div class="title">
-                            <a href="/project/{{ r.package.id }}">{{ r.package.id }} {{ r.version_text }}</a>
-                            <div class="desc">{{ r.package.summary }}</div>
+                            <a href="/project/{{ package.id }}">{{ package.id }} {{ r.version_text }}</a>
+                            <div class="desc">{{ package.summary }}</div>
                         </div>
                     </div>
                 {% endfor %}

app/ch16_mongodb/final/pypi_org/viewmodels/home/index_viewmodel.py

@@ -10,3 +10,8 @@ def __init__(self):
         self.package_count = package_service.get_package_count()
         self.release_count = package_service.get_release_count()
         self.user_count = user_service.get_user_count()
+
+        package_ids = [r.package_id for r in self.releases]
+        packages = package_service.get_packages_by_ids(package_ids)
+
+        self.package_lookup = {p.id: p for p in packages}

app/ch16_mongodb/final/pypi_org/viewmodels/packages/pagedetails_viewmodel.py

@@ -15,8 +15,11 @@ def __init__(self, package_name: str):
         self.latest_release = None
         self.is_latest = True
 
-        if self.package and self.package.releases:
-            self.latest_release = self.package.releases[0]
-            self.latest_version = self.latest_release.version_text
+        if self.package:
+            self.latest_release = package_service.get_latest_release_for_package(
+                self.package.id)
+
+            if self.latest_release:
+                self.latest_version = self.latest_release.version_text
 
         self.release_version = self.latest_release