- MandrivaUpdate:

  o faster startup if there're priority updates
  o only list priority updates if existing

2 files changed, 12 insertions, 1 deletions

diff --git a/NEWS b/NEWS
index 9b683972..ee22a94d 100644
--- a/NEWS
+++ b/NEWS
@@ -1,3 +1,7 @@
+- MandrivaUpdate:
+  o faster startup if there're priority updates
+  o only list priority updates if existing
+
 Version 4.5 - 7 March 2008, Thierry Vignaud
 
 - reimplement priority upgrade support through urpmi-5.9's
diff --git a/Rpmdrake/pkg.pm b/Rpmdrake/pkg.pm
index 365965d6..892d0968 100644
--- a/Rpmdrake/pkg.pm
+++ b/Rpmdrake/pkg.pm
@@ -451,7 +451,7 @@ sub get_pkgs {
 
     # list of pure updates (w/o those matching /etc/urpmi/skip.list but with their deps):
     my @requested_strict;
-    if ($probe_only_for_updates) {
+    if ($probe_only_for_updates && !$need_restart) {
         @requested_strict = sort map {
             urpm_name($_);
         } $urpm->resolve_requested($db, $state, $requested, callback_choices => \&Rpmdrake::gui::callback_choices);
@@ -466,6 +466,13 @@ sub get_pkgs {
         }
     }
 
+    if ($probe_only_for_updates && $need_restart) {
+        @requested_strict = map { scalar $_->fullname } @{$urpm->{depslist}}[keys %{$state->{selected}}];
+        # drop non priority updates:
+        undef @requested;
+    }
+
+
     # list updates including skiped ones + their deps in MandrivaUpdate:
     @requested = uniq(@requested, @requested_strict) if $probe_only_for_updates;