From 96465262f72b1b4230513e978d8412d0affc75a1 Mon Sep 17 00:00:00 2001
From: Elena ``of Valhalla'' Grandi <valhalla@trueelena.org>
Date: Sun, 4 Jun 2017 22:49:54 +0200
Subject: Search for all entries

---
 lesana/collection.py     |  7 +++++++
 lesana/command.py        | 15 +++++++++------
 tests/test_collection.py |  8 ++++++++
 3 files changed, 24 insertions(+), 6 deletions(-)

diff --git a/lesana/collection.py b/lesana/collection.py
index d3715d1..da001ae 100644
--- a/lesana/collection.py
+++ b/lesana/collection.py
@@ -263,6 +263,13 @@ class Collection(object):
                 yield self._match_to_entry(match)
             offset += pagesize
 
+    def get_all_documents(self):
+        cache = self._get_cache()
+        postlist = cache.postlist("")
+        for post in postlist:
+            doc = cache.get_document(post.docid)
+            yield self._doc_to_entry(doc)
+
     def _match_to_entry(self, match):
         return self._doc_to_entry(match.document)
 
diff --git a/lesana/command.py b/lesana/command.py
index 41448b3..2915d30 100644
--- a/lesana/command.py
+++ b/lesana/command.py
@@ -198,13 +198,16 @@ class Search(guacamole.Command):
         offset = ctx.args.offset or 0
         pagesize = ctx.args.pagesize or 12
         collection = Collection(ctx.args.collection)
-        collection.start_search(' '.join(ctx.args.query))
-        if ctx.args.all:
-            results = collection.get_all_search_results()
+        if ctx.args.query == ['*']:
+            results = collection.get_all_documents()
         else:
-            results = collection.get_search_results(
-                offset,
-                pagesize)
+            collection.start_search(' '.join(ctx.args.query))
+            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(
diff --git a/tests/test_collection.py b/tests/test_collection.py
index 891a439..5c326e3 100644
--- a/tests/test_collection.py
+++ b/tests/test_collection.py
@@ -87,6 +87,14 @@ class testCollection(unittest.TestCase):
         matches = list(self.collection.get_all_search_results())
         self.assertEqual(matches, [])
 
+    def test_all_entries(self):
+        self.collection = lesana.Collection('tests/data/simple')
+        res = self.collection.get_all_documents()
+        matches = list(res)
+        self.assertEqual(len(matches), 3)
+        for m in matches:
+            self.assertIsInstance(m, lesana.Entry)
+
     def test_entry_from_uid(self):
         self.collection = lesana.Collection('tests/data/simple')
         entry = self.collection.entry_from_uid(
-- 
cgit v1.2.3