1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
import html
import re
from flask import redirect, request, url_for
from werkzeug.exceptions import BadRequest
from pysite.base_route import RouteView
from pysite.decorators import csrf
from pysite.mixins import DBMixin
STRIP_REGEX = re.compile(r"<[^<]+?>")
class SearchView(RouteView, DBMixin):
path = "/search" # "path" means that it accepts slashes
name = "search"
table_name = "wiki"
revision_table_name = "wiki_revisions"
def get(self):
return self.render("wiki/search.html")
@csrf
def post(self):
given_query = request.form.get("query")
if not given_query or not given_query.strip():
raise BadRequest()
query = f"({re.escape(given_query)})"
pages = self.db.filter(
self.table_name,
lambda doc: doc["text"].match(f"(?i){query}")
)
if len(pages) == 1:
slug = pages[0]["slug"]
return redirect(url_for("wiki.page", page=slug), code=303)
for obj in pages:
text = obj["text"]
matches = re.finditer(query, text, flags=re.IGNORECASE)
snippets = []
for match in matches:
start = match.start() - 50
if start < 0:
start = 0
end = match.end() + 50
if end > len(text):
end = len(text)
match_text = text[start:end]
match_text = re.sub(query, r"<strong>\1</strong>", html.escape(match_text), flags=re.IGNORECASE)
snippets.append(match_text.replace("\n", "<br />"))
obj["matches"] = snippets
pages = sorted(pages, key=lambda d: d["title"])
return self.render("wiki/search_results.html", pages=pages, query=given_query)
|
