Add option to show all search results

1 files changed, 29 insertions, 18 deletions

diff --git a/lesana/command.py b/lesana/command.py
index dd1864a..56c855a 100644
--- a/lesana/command.py
+++ b/lesana/command.py
@@ -139,27 +139,35 @@ class Index(guacamole.Command):
 
 class Search(guacamole.Command):
     arguments = [
-        (['--collection', '-c'], dict(
+        ]
+
+    def register_arguments(self, parser):
+        parser.add_argument(
+            '--collection', '-c',
             help='The collection to work on (default .)'
-            )),
-        (['--template', '-t'], dict(
+            )
+        parser.add_argument(
+            '--template', '-t',
             help='Am',
-            )),
-        (['--offset'], dict(
+            ),
+        parser.add_argument(
+            '--offset',
             type=int,
-            )),
-        (['--pagesize'], dict(
+            ),
+        parser.add_argument(
+            '--pagesize',
             type=int,
-            )),
-        (['query'], dict(
+            ),
+        parser.add_argument(
+            '--all',
+            action='store_true',
+            help='Return all available results'
+            )
+        parser.add_argument(
+            'query',
             help='Xapian query to search in the collection',
             nargs='+'
-            )),
-        ]
-
-    def register_arguments(self, parser):
-        for arg in self.arguments:
-            parser.add_argument(*arg[0], **arg[1])
+            ),
 
     def invoked(self, ctx):
         # TODO: implement "searching" for everything
@@ -177,9 +185,12 @@ class Search(guacamole.Command):
         pagesize = ctx.args.pagesize or 12
         collection = Collection(ctx.args.collection)
         collection.start_search(' '.join(ctx.args.query))
-        results = collection.get_search_results(
-            offset,
-            pagesize)
+        if ctx.args.all:
+            results = collection.get_all_search_results()
+        else:
+            results = collection.get_search_results(
+                offset,
+                pagesize)
         if ctx.args.template:
             env = jinja2.Environment(
                 loader=jinja2.FileSystemLoader(