From 8d3d05c7cb81611537aa34fa68e98e5e22b74847 Mon Sep 17 00:00:00 2001
From: Dan McGee <dan@archlinux.org>
Date: Tue, 15 Feb 2011 20:00:33 -0600
Subject: Allow for optional info in required by display

We need to make our root object the PackageDepend rather than the Package to
get at this, so do a slight refactor on get_requiredby().

Signed-off-by: Dan McGee <dan@archlinux.org>
---
 main/models.py | 31 ++++++++++++++++++-------------
 1 file changed, 18 insertions(+), 13 deletions(-)

(limited to 'main')

diff --git a/main/models.py b/main/models.py
index dafbb1e..73a7620 100644
--- a/main/models.py
+++ b/main/models.py
@@ -7,7 +7,6 @@ from packages.models import PackageRelation
 
 from itertools import groupby
 import pytz
-from operator import attrgetter
 
 class UserProfile(models.Model):
     notify = models.BooleanField(
@@ -173,10 +172,15 @@ class Package(models.Model):
         list slim by including the corresponding package in the same testing
         category as this package if that check makes sense.
         """
-        requiredby = Package.objects.select_related('arch', 'repo').filter(
-                packagedepend__depname=self.pkgname,
-                arch__in=self.applicable_arches()
-                ).distinct().order_by('pkgname')
+        requiredby = PackageDepend.objects.select_related('pkg',
+                'pkg__arch', 'pkg__repo').filter(
+                pkg__arch__in=self.applicable_arches(),
+                depname=self.pkgname).order_by(
+                'pkg__pkgname', 'pkg__id')
+        # sort out duplicate packages; this happens if something has a double
+        # versioned dep such as a kernel module
+        requiredby = [list(vals)[0] for k, vals in
+                groupby(requiredby, lambda x: x.pkg.id)]
 
         # find another package by this name in the opposite testing setup
         if not Package.objects.filter(pkgname=self.pkgname,
@@ -189,14 +193,15 @@ class Package(models.Model):
         # for each unique package name, try to screen our package list down to
         # those packages in the same testing category (yes or no) iff there is
         # a package in the same testing category.
-        for name, pkgs in groupby(requiredby, attrgetter('pkgname')):
-            pkgs = list(pkgs)
-            pkg = pkgs[0]
-            if len(pkgs) > 1:
-                pkgs = [p for p in pkgs if p.repo.testing == self.repo.testing]
-                if len(pkgs) > 0:
-                    pkg = pkgs[0]
-            trimmed.append(pkg)
+        for name, dep_pkgs in groupby(requiredby, lambda x: x.pkg.pkgname):
+            dep_pkgs = list(dep_pkgs)
+            dep = dep_pkgs[0]
+            if len(dep_pkgs) > 1:
+                dep_pkgs = [d for d in dep_pkgs
+                        if d.pkg.repo.testing == self.repo.testing]
+                if len(dep_pkgs) > 0:
+                    dep = dep_pkgs[0]
+            trimmed.append(dep)
         return trimmed
 
     @cache_function(300)
-- 
cgit v1.2.3-24-g4f1b