在Python的Flask框架中实现全文搜索功能

灰常不幸的是，关系型数据库对全文检索的支持没有被标准化。不同的数据库通过它们自己的方式来实现全文检索，而且SQLAlchemy在全文检索上也没有提供一个好的抽象。

我们现在使用SQLite作为我们的数据库，所以我们可以绕开SQLAlchemy而使用SQLite提供的工具来创建一个全文检索索引。但这么做不怎么好，因为如果有一天我们换用别的数据库，那么我们就得重写另一个数据库的全文检索方法。

所以我们的方案是，我们将让我们现有的数据库处理常规数据，然后我们创建一个专门的数据库来解决全文检索。

只有很少的开源的全文检索引擎。据我说知只有一个Whoosh提供了Flask的扩展，它是用Python语言写的全文检索引擎。使用纯Python引擎的优点是它可以运行在任何有Python解释器的地方。缺点就是它的搜索性能没有达到用C或者C++写的搜索引擎那么好。在我的脑子里理想的解决方案是有一个搜索引擎，它提供了Flask的扩展，能连接大多数数据库，而且还要像Flask-SQLAlchemy那样提供一个能自由使用大多数数据库的方法，但现在貌似木有这样的全文检索引擎。Django的开发者有一个非常棒的，支持大多数全文检索引擎的扩展，叫django-haystack。希望有一天某个家伙能为Flask提供一个相似的扩展。

但现在，我们将通过Whoosh实现我们自己的全文检索。我们将使用Flask-WhooshAlchemy扩展，该扩展使得Whoosh数据库和Flask-SQLAlchemy模块结合起来。

如果你还没在你的虚拟环境中安装Flask-WhooshAlchemy扩展，马上安装它。

Windows用户用以下命令安装:

from app import app
import flask.ext.whooshalchemy as whooshalchemy
 
class Post(db.Model):
  __searchable__ = ['body']
 
  id = db.Column(db.Integer, primary_key = True)
  body = db.Column(db.String(140))
  timestamp = db.Column(db.DateTime)
  user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
 
  def __repr__(self):
    return '' % (self.text)
 
whooshalchemy.whoosh_index(app, Post)

>>> from app.models import Post
>>> from app import db
>>> for post in Post.query.all():
...  db.session.delete(post)
>>> db.session.commit()

>>> from app.models import User, Post
>>> from app import db
>>> import datetime
>>> u = User.query.get(1)
>>> p = Post(body='my first post', timestamp=datetime.datetime.utcnow(), author=u)
>>> db.session.add(p)
>>> p = Post(body='my second post', timestamp=datetime.datetime.utcnow(), author=u)
>>> db.session.add(p)
>>> p = Post(body='my third and last post', timestamp=datetime.datetime.utcnow(), author=u)
>>> db.session.add(p)
>>> db.session.commit()

>>> Post.query.whoosh_search('post').all()
[, , ]
>>> Post.query.whoosh_search('second').all()
[]
>>> Post.query.whoosh_search('second OR last').all()
[, ]

MAX_SEARCH_RESULTS = 50

class SearchForm(Form):
  search = TextField('search', validators = [Required()])

@app.before_request
def before_request():
  g.user = current_user
  if g.user.is_authenticated():
    g.user.last_seen = datetime.utcnow()
    db.session.add(g.user)
    db.session.commit()
    g.search_form = SearchForm()

Microblog:
  Home
  {% if g.user.is_authenticated() %}
  | Your Profile
  | 
  | Logout
  {% endif %}

@app.route('/search', methods = ['POST'])
@login_required
def search():
  if not g.search_form.validate_on_submit():
    return redirect(url_for('index'))
  return redirect(url_for('search_results', query = g.search_form.search.data))

@app.route('/search_results/')
@login_required
def search_results(query):
  results = Post.query.whoosh_search(query, MAX_SEARCH_RESULTS).all()
  return render_template('search_results.html',
    query = query,
    results = results)

{% extends "base.html" %}
 
{% block content %}
Search results for "{{query}}":
{% for post in results %}
  {% include 'post.html' %}
{% endfor %}
{% endblock %}