brotherton-erpnext/erpnext/templates/pages/search_help.py

from __future__ import unicode_literals
import frappe, requests
from frappe import _
from jinja2 import utils
from html2text import html2text
from six import text_type
from frappe.utils import sanitize_html
from frappe.utils.global_search import search

def get_context(context):
	context.no_cache = 1
	if frappe.form_dict.q:
		query = str(utils.escape(sanitize_html(frappe.form_dict.q)))
		context.title = _('Help Results for')
		context.query = query

		context.route = '/search_help'
		d = frappe._dict()
		d.results_sections = get_help_results_sections(query)
		context.update(d)
	else:
		context.title = _('Docs Search')

@frappe.whitelist(allow_guest = True)
def get_help_results_sections(text):
	out = []
	settings = frappe.get_doc("Support Settings", "Support Settings")

	for api in settings.search_apis:
		results = []
		if api.source_type == "API":
			response_json = get_response(api, text)
			topics_data = get_topics_data(api, response_json)
			results = prepare_api_results(api, topics_data)
		else:
			# Source type is Doctype
			doctype = api.source_doctype
			raw = search(text, 0, 20, doctype)
			results = prepare_doctype_results(api, raw)

		if results:
			# Add section
			out.append({
				"title": api.source_name,
				"results": results
			})

	return out

def get_response(api, text):
	response = requests.get(api.base_url + '/' + api.query_route, data={
		api.search_term_param_name: text
	})

	response.raise_for_status()
	return response.json()

def get_topics_data(api, response_json):
	if not response_json:
		response_json = {}
	topics_data = {} # it will actually be an array
	key_list = api.response_result_key_path.split(',')

	for key in key_list:
		topics_data = response_json.get(key) if not topics_data else topics_data.get(key)

	return topics_data or []

def prepare_api_results(api, topics_data):
	if not topics_data:
		topics_data = []

	results = []
	for topic in topics_data:
		route = api.base_url + '/' + (api.post_route  + '/' if api.post_route else "")
		for key in api.post_route_key_list.split(','):
			route += text_type(topic[key])

		results.append(frappe._dict({
			'title': topic[api.post_title_key],
			'preview': html2text(topic[api.post_description_key]),
			'route': route
		}))
	return results[:5]

def prepare_doctype_results(api, raw):
	results = []
	for r in raw:
		prepared_result = {}
		parts = r["content"].split(' ||| ')

		for part in parts:
			pair = part.split(' : ', 1)
			prepared_result[pair[0]] = pair[1]

		results.append(frappe._dict({
			'title': prepared_result[api.result_title_field],
			'preview': prepared_result[api.result_preview_field],
			'route': prepared_result[api.result_route_field]
		}))

	return results