feat: improved search results in command palette

lms/sqlite.py

@@ -0,0 +1,122 @@
+from contextlib import suppress
+
+import frappe
+from frappe.search.sqlite_search import SQLiteSearch, SQLiteSearchIndexMissingError
+from frappe.utils import update_progress_bar
+from redis.exceptions import ResponseError
+
+
+class LearningSearch(SQLiteSearch):
+	INDEX_NAME = "learning.db"
+
+	INDEX_SCHEMA = {
+		"metadata_fields": ["category", "owner", "published"],
+		"tokenizer": "unicode61 remove_diacritics 2 tokenchars '-_'",
+	}
+
+	INDEXABLE_DOCTYPES = {
+		"LMS Course": {
+			"fields": [
+				"name",
+				"title",
+				{"content": "description"},
+				"short_introduction",
+				"published",
+				"category",
+				"owner",
+				{"modified": "published_on"},
+			],
+		},
+		"LMS Batch": {
+			"fields": [
+				"name",
+				"title",
+				"description",
+				{"content": "batch_details"},
+				"published",
+				"category",
+				"owner",
+				{"modified": "start_date"},
+			],
+		},
+	}
+
+	DOCTYPE_FIELDS = {
+		"LMS Course": [
+			"name",
+			"title",
+			"description",
+			"short_introduction",
+			"category",
+			"creation",
+			"modified",
+			"owner",
+		],
+		"LMS Batch": [
+			"name",
+			"title",
+			"description",
+			"batch_details",
+			"category",
+			"creation",
+			"modified",
+			"owner",
+		],
+	}
+
+	def can_create_course(self, roles):
+		return "Course Creator" in roles or "Moderator" in roles
+
+	def can_create_batch(self, roles):
+		return "Batch Evaluator" in roles or "Moderator" in roles
+
+	def get_records(self, doctype):
+		records = []
+		roles = frappe.get_roles()
+		filters = {}
+
+		if doctype == "LMS Course":
+			if not self.can_create_course(roles):
+				filters = {"published": 1}
+
+		if doctype == "LMS Batch":
+			if not self.can_create_batch(roles):
+				filters = {"published": 1}
+
+		records = frappe.db.get_all(doctype, filters=filters, fields=self.DOCTYPE_FIELDS[doctype])
+		for record in records:
+			record["doctype"] = doctype
+
+		return records
+
+	def build_index(self):
+		try:
+			super().build_index()
+		except Exception as e:
+			frappe.throw(e)
+
+	def get_search_filters(self):
+		roles = frappe.get_roles()
+		if not (self.can_create_course(roles) and self.can_create_batch(roles)):
+			return {"published": 1}
+		return {}
+
+
+class LearningSearchIndexMissingError(SQLiteSearchIndexMissingError):
+	pass
+
+
+def build_index():
+	search = LearningSearch()
+	search.build_index()
+
+
+def build_index_in_background():
+	if not frappe.cache().get_value("learning_search_indexing_in_progress"):
+		frappe.enqueue(build_index, queue="long")
+
+
+def build_index_if_not_exists():
+	search = LearningSearch()
+	if not search.index_exists():
+		build_index()