Merge lp:~brian-murray/ubuntu-archive-tools/package-sub-json into lp:ubuntu-archive-tools

[Brian Murray]

I've modified unsubscribed-packages to display the subscribers for packages in main and restricted and to write to a json file (although it can still print). Additionally, I renamed it to package-subscribers as it does more than just show the unsubscribed packages.

[Colin Watson (cjwatson)]

Approved in principle, but I'd like this to dump out information about all packages, as discussed on IRC. I'll make those changes and merge.

review: Approve

=== renamed file 'unsubscribed-packages' => 'package-subscribers'
--- unsubscribed-packages	2013-08-21 18:24:15 +0000
+++ package-subscribers	2013-10-30 19:39:18 +0000
@@ -4,9 +4,11 @@
 
 import atexit
 import bz2
+from collections import defaultdict
 from contextlib import closing
 from optparse import OptionParser
 import shutil
+import simplejson as json
 import sys
 import tempfile
 try:
@@ -68,9 +70,17 @@
 def main():
     parser = OptionParser(
         description="Check for source packages in main or restricted in "
-            "active distro series without a team subscribed.")
+            "active distro series and return a json file of the teams "
+            "to which they map.")
     parser.add_option(
         "-l", "--launchpad", dest="launchpad_instance", default="production")
+    parser.add_option(
+        "-u", "--unsubscribed", action="store_true", default=False,
+        help="Only return packages which have no subscriber")
+    parser.add_option(
+        "-p", "--print", action="store_true", default=False,
+        dest="display",
+        help="Print results to screen instead of a json file")
     options, _ = parser.parse_args()
     options.suite = None
     options.distribution = "ubuntu"
@@ -78,6 +88,7 @@
         "unsubscribed-packages", options.launchpad_instance)
     launchpad = options.launchpad
     ubuntu = launchpad.distributions[options.distribution]
+    json_filename = "m-r-package-team-mapping.json"
     options.suites = []
     for series in ubuntu.series:
         # very few lucid packages are supported
@@ -110,18 +121,34 @@
         'xubuntu-bugs',
         ]
 
-    subscriptions = set()
+    data = { "unsubscribed": [] }
+    subscriptions = defaultdict(list)
     for team_name in team_names:
+        data[team_name] = []
         team = launchpad.people[team_name]
         team_subs = team.getBugSubscriberPackages()
         for src_pkg in team_subs:
-            subscriptions.add(src_pkg.name)
+            subscriptions[src_pkg.name].append(team_name)
 
     source_packages = source_names(options)
-
     for source_package in source_packages:
         if source_package not in subscriptions:
-            print("No team is subscribed to: %s" % source_package)
+            data["unsubscribed"].append(source_package)
+            if options.display:
+                print("No team is subscribed to: %s" %
+                      source_package)
+        else:
+            if not options.unsubscribed:
+                for team_name in subscriptions[source_package]:
+                    data[team_name].append(source_package)
+                if options.display:
+                    print("%s is subscribed to: %s" %
+                          (team_name, source_package))
+
+    if not options.display:
+        with open(json_filename, 'w') as json_file:
+            json_file.write(json.dumps(data, indent=4))
+
 
 if __name__ == '__main__':
     main()