Make it possible to exclude packages by name

In some situations some packages may need to be excluded, e.g. if we're
expecting some odd situation. In this case it would be nice to exclude
the packages by name.

Signed-off-by: Gabriel Filion <lelutin@torproject.org>

Author: Gabriel Filion

apt_info.py

@@ -24,6 +24,7 @@
 
 import apt
 import apt_pkg
+import argparse
 import collections
 import logging
 import os
@@ -60,9 +61,11 @@ def _convert_candidates_to_upgrade_infos(candidates):
     return changes_list
 
 
-def _write_pending_upgrades(registry, cache, logger):
+def _write_pending_upgrades(registry, cache, logger, cli_args):
     candidates = {
-        p.candidate for p in cache if p.is_upgradable and not p.phasing_applied
+        p.candidate
+        for p in cache
+        if p.is_upgradable and not p.phasing_applied and p.name not in cli_args.exclude
     }
     for candidate in candidates:
         logger.debug(
@@ -79,13 +82,14 @@ def _write_pending_upgrades(registry, cache, logger):
             g.labels(change.labels['origin'], change.labels['arch']).set(change.count)
 
 
-def _write_held_upgrades(registry, cache, logger):
+def _write_held_upgrades(registry, cache, logger, cli_args):
     held_candidates = {
         p.candidate for p in cache
         if (
             p.is_upgradable
             and p._pkg.selected_state == apt_pkg.SELSTATE_HOLD
             and not p.phasing_applied
+            and p.name not in cli_args.exclude
         )
     }
     for candidate in held_candidates:
@@ -103,13 +107,14 @@ def _write_held_upgrades(registry, cache, logger):
             g.labels(change.labels['origin'], change.labels['arch']).set(change.count)
 
 
-def _write_obsolete_packages(registry, cache, logger):
+def _write_obsolete_packages(registry, cache, logger, cli_args):
     # This corresponds to the apt filter "?obsolete"
     obsoletes = [p for p in cache if p.is_installed and (
                   p.candidate is None or
                   not p.candidate.origins or
                   (len(p.candidate.origins) == 1 and
                    p.candidate.origins[0].origin in ['', "/var/lib/dpkg/status"])
+                  and p.name not in cli_args.exclude
                 )]
     for package in obsoletes:
         if package.candidate is None:
@@ -126,8 +131,12 @@ def _write_obsolete_packages(registry, cache, logger):
     g.set(len(obsoletes))
 
 
-def _write_autoremove_pending(registry, cache, logger):
-    autoremovable_packages = {p.candidate for p in cache if p.is_auto_removable}
+def _write_autoremove_pending(registry, cache, logger, cli_args):
+    autoremovable_packages = {
+        p.candidate
+        for p in cache
+        if p.is_auto_removable and p.name not in cli_args.exclude
+    }
     for candidate in autoremovable_packages:
         logger.debug(
             "autoremovable package: %s / %s",
@@ -181,13 +190,17 @@ def _main():
     if os.environ.get("DEBUG"):
         logger.setLevel(logging.DEBUG)
 
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--exclude", nargs='*', default=[])
+    args = parser.parse_args(sys.argv[1:])
+
     cache = apt.cache.Cache()
 
     registry = CollectorRegistry()
-    _write_pending_upgrades(registry, cache, logger)
-    _write_held_upgrades(registry, cache, logger)
-    _write_obsolete_packages(registry, cache, logger)
-    _write_autoremove_pending(registry, cache, logger)
+    _write_pending_upgrades(registry, cache, logger, args)
+    _write_held_upgrades(registry, cache, logger, args)
+    _write_obsolete_packages(registry, cache, logger, args)
+    _write_autoremove_pending(registry, cache, logger, args)
     _write_installed_packages_per_origin(registry, cache)
     _write_cache_timestamps(registry)
     _write_reboot_required(registry)