From 1e8f0918135361662d1ac5b8dca2aa7a8eb1b808 Mon Sep 17 00:00:00 2001
From: Anshul Gupta <ansg191@anshulg.com>
Date: Fri, 7 Jun 2024 15:24:23 -0700
Subject: Make query search over title and content

---
 index.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

(limited to 'index.py')

diff --git a/index.py b/index.py
index 96b7e9d..280320d 100644
--- a/index.py
+++ b/index.py
@@ -70,6 +70,7 @@ class Index:
             doc.add(StoredField("url", rows[i][4]))
             writer.addDocument(doc)
 
+        print("Finished Indexing...")
         writer.close()
 
     def query(self, query):
@@ -77,10 +78,12 @@ class Index:
         lucene.getVMEnv().attachCurrentThread()
 
         reader = DirectoryReader.open(self.dir)
-        print(reader.numDocs())
         searcher = IndexSearcher(reader)
-        parser = QueryParser("content", self.analyzer)
-        q = parser.parse(query)
+
+        parser = QueryParser("<default field>", self.analyzer)
+
+        new_query = "content:" + query + " OR title:" + query
+        q = parser.parse(new_query)
         hits = searcher.search(q, 10)
 
         results = []
@@ -99,4 +102,5 @@ class Index:
                 }
             )
 
+        lucene.getVMEnv().detachCurrentThread()
         return results
-- 
cgit v1.2.3