[extensions-web] search: Add a simple command to test search

[Jasper St. Pierre <jstpierre src gnome org>]

commit 09c42c54816659a19b33c4e0a746f586749b2f93
Author: Jasper St. Pierre <jstpierre mecheye net>
Date:   Mon Mar 18 01:50:46 2013 -0400

    search: Add a simple command to test search

 .../management/commands/searchextensions.py        |   38 ++++++++++++++++++++
 1 files changed, 38 insertions(+), 0 deletions(-)
---
diff --git a/sweettooth/extensions/management/commands/searchextensions.py 
b/sweettooth/extensions/management/commands/searchextensions.py
new file mode 100644
index 0000000..93f32c5
--- /dev/null
+++ b/sweettooth/extensions/management/commands/searchextensions.py
@@ -0,0 +1,38 @@
+
+from django.core.management.base import BaseCommand, make_option
+from extensions.models import Extension
+from extensions.search import enquire
+
+def append_version(option, opt_str, value, parser):
+    values = parser.values
+    if getattr(values, "versions", None) is None:
+        values.versions = []
+    values.versions.append(value)
+
+class Command(BaseCommand):
+    args = '<query>'
+    help = 'Test the search engine'
+
+    option_list = BaseCommand.option_list + (
+        make_option('-V', action="callback", callback=append_version, type="string"),
+    )
+
+    def handle(self, *args, **kwargs):
+        query = ' '.join(args)
+        versions = kwargs.get('versions')
+        db, enquiry = enquire(query, versions)
+
+        mset = enquiry.get_mset(0, db.get_doccount())
+        pks = [match.document.get_data() for match in mset]
+
+        # filter doesn't guarantee an order, so we need to get all the
+        # possible models then look them up to get the ordering
+        # returned by xapian. This hits the database all at once, rather
+        # than pagesize times.
+        extension_lookup = {}
+        for extension in Extension.objects.filter(pk__in=pks):
+            extension_lookup[str(extension.pk)] = extension
+
+        extensions = [extension_lookup[pk] for pk in pks]
+        for ext in extensions:
+            print ext.name