Made searching more advanced.

This commit is contained in:
Raoul Snyman 2010-02-18 09:02:46 +02:00
parent 12d2a96626
commit 31fc3695c3
5 changed files with 82 additions and 9 deletions

View File

@ -48,6 +48,8 @@ def make_map():
map.connect('/archive/{year}/{month}/{day}', controller='blog', action='archive')
map.connect('/archive/{year}/{month}/{day}/{url}', controller='blog', action='view')
map.connect('/search', controller='blog', action='search')
map.connect('/tag/{id}', controller='blog', action='tag')
map.connect('/{controller}/{action}')

View File

@ -118,14 +118,11 @@ class BlogController(BaseController):
if not c.querystring:
h.flash.set_message(u'You didn\'t supply anything to search for.', u'error')
h.redirect_to('/')
keywords = [or_(Post.body.contains(kw.strip()), Post.title.contains(kw.strip())) for kw in c.querystring.split(',')]
c.page_title = u'Search'
c.posts = Session.query(Post)\
.filter(
or_(
Post.body.contains(c.querystring),
Post.title.contains(c.querystring)
)
)\
.all()
c.posts = Session.query(Post)
for or_clause in keywords:
c.posts = c.posts.filter(or_clause)
c.posts = c.posts.all()
return render(u'/blog/search.mako')

View File

@ -32,6 +32,76 @@ from turbomail import Message
from scribeengine.lib.base import render, h
class KeywordProcessor(object):
"""Process user-supplied keywords, tags, or search terms.
This tries to be as flexible as possible while being efficient.
The vast majority of the work is done in the regular expression."""
def __init__(self, separators=' \t', quotes=['"', "'"], groups=[], group=False, normalize=None, sort=False, result=list):
"""Configure the processor.
separators: A list of acceptable separator characters. The first will be used for joins.
quotes: Pass a list or tuple of allowable quotes. E.g. ["\"", "'"] or None to disable.
groups: Pass a string, list, or tuple of allowable prefixes. E.g. '+-' or None to disable.
group: Pass in the type you want to group by, e.g. list, tuple, or dict.
normalize: Pass a function which will normalize the results. E.g. lambda s: s.lower().strip(' \"')
sort: Sort the resulting list (or lists) alphabeticlly.
result: The return type. One of set, tuple, list.
If groups are defined, and group is not, the result will be a list/tuple/set of tuples, e.g. [('+', "foo"), ...]
"""
separators = list(separators)
self.pattern = ''.join((
('[\s%s]*' % (''.join(separators), )), # Trap possible leading space or separators.
'(',
('[%s]%s' % (''.join([i for i in list(groups) if i is not None]), '?' if None in groups else '')) if groups else '', # Pass groups=('+','-') to handle optional leading + or -.
''.join([(r'%s[^%s]+%s|' % (i, i, i)) for i in quotes]) if quotes else '', # Match any amount of text (that isn't a quote) inside quotes.
('[^%s]+' % (''.join(separators), )), # Match any amount of text that isn't whitespace.
')',
('[%s]*' % (''.join(separators), )), # Match possible separator character.
))
self.regex = re.compile(self.pattern)
self.groups = list(groups)
self.group = dict if group is True else group
self.normalize = normalize
self.sort = sort
self.result = result
def split(self, value):
if not isinstance(value, basestring): raise TypeError("Invalid type for argument 'value'.")
matches = self.regex.findall(value)
if callable(self.normalize): matches = [self.normalize(i) for i in matches]
if self.sort: matches.sort()
if not self.groups: return self.result(matches)
groups = dict([(i, list()) for i in self.groups])
if None not in groups.iterkeys(): groups[None] = list() # To prevent errors.
for i in matches:
if i[0] in self.groups:
groups[i[0]].append(i[1:])
else:
groups[None].append(i)
if self.group is dict: return groups
if self.group is False or self.group is None:
results = []
for group in self.groups:
results.extend([(group, match) for match in groups[group]])
return self.result(results)
return self.group([[match for match in groups[group]] for group in self.groups])
def send_mail(template, mail_to, mail_from, subject, variables={}, attachments=[]):
"""
Sends an e-mail using the template ``template``.

View File

@ -1,6 +1,7 @@
<%inherit file="/base.mako"/>
<%include file="/flash.mako"/>
<h2 class="title">Searching for: &quot;${c.querystring}&quot;</h2>
% if len(c.posts) > 0:
% for post in c.posts:
<div class="post">
<h3 class="title"><a href="${h.url_for_post(post)}">${post.title}</a></h3>
@ -20,3 +21,6 @@
</p>
</div>
% endfor
% else:
<div class="post"><p>Sorry, there seem to be no results for your search query.</p></div>
% endif

View File

@ -1,7 +1,7 @@
<div id="sidebar">
<ul>
<li id="search">
<form id="searchform" method="get" action="/blog/search">
<form id="searchform" method="get" action="/search">
<div>
<input type="text" name="q" id="s" size="15" />
<br />