dakweb: add -a option for madison

[dak.git] / dakweb / queries / madison.py

diff --git a/dakweb/queries/madison.py b/dakweb/queries/madison.py
index 469f9c80d304e10c0c2d7ef614d84e62572023cf..82a3af5b31048ab1b60fab5a43303b902bc9955b 100644 (file)
--- a/dakweb/queries/madison.py
+++ b/dakweb/queries/madison.py
@@ -1,3 +1,11 @@
+""" "Madison" interface
+
+@contact: Debian FTPMaster <ftpmaster@debian.org>
+@copyright: 2014  Ansgar Burchardt <ansgar@debian.org>
+@copyright: 2014  Joerg Jaspert <joerg@debian.org>
+@license: GNU General Public License version 2 or later
+"""
+
 import bottle
 import json
 
@@ -6,11 +14,32 @@ from dakweb.webregister import QueryRegister
 
 @bottle.route('/madison')
 def madison():
+    """
+    Display information about B{package(s)}.
+
+    @since: December 2014
+
+    @keyword package: Space seperated list of packages.
+    @keyword a: only show info for specified architectures.
+    @keyword b: only show info for a binary type. I{deb/udeb/dsc}
+    @keyword c: only show info for specified component(s). I{main/contrib/non-free}
+    @keyword s: only show info for this suite.
+    @keyword S: show info for the binary children of source pkgs. I{true/false}
+    @keyword f: output json format. I{json}
+    @see: L{I{suites}<dakweb.queries.suite.suites>} on how to receive a list of valid suites.
+
+    @rtype: text/plain or application/json
+    @return: Text or Json format of the data
+    """
+
     r = bottle.request
 
     packages = r.query.get('package', '').split()
     kwargs = dict()
 
+    architectures = r.query.get('a', None)
+    if architectures is not None:
+        kwargs['architectures'] = architectures.split(",")
     binary_type = r.query.get('b', None)
     if binary_type is not None:
         kwargs['binary_types'] = [binary_type]
@@ -24,8 +53,19 @@ def madison():
         kwargs['source_and_binary'] = True
     #if 'r' in r.query:
     #    kwargs['regex'] = True
+    format = r.query.get('f', None)
+    if format is not None:
+        kwargs['format'] = 'python'
 
     result = list_packages(packages, **kwargs)
-    return "\n".join(result) + "\n"
+
+    if format is None:
+        bottle.response.content_type = 'text/plain'
+        for row in result:
+            yield row
+            yield "\n"
+    else:
+        yield json.dumps(list(result))
+
 
 QueryRegister().register_path('/madison', madison)