Allow json format for madison

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

diff --git a/dakweb/queries/madison.py b/dakweb/queries/madison.py
index d27d4784864caa1fbf9b2696dfbf1bcbe6d93eba..eb8c45cd6ffaac27891d840bd624486125369fed 100644 (file)
--- a/dakweb/queries/madison.py
+++ b/dakweb/queries/madison.py
@@ -24,12 +24,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)
 
-    bottle.response.content_type = 'text/plain'
-    for row in result:
-        yield row
-        yield "\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)