django全文搜索学习心得（五） whoosh 精简版

今晚先睡觉，明天写一个，简单版的，不用任何框架。

今天看到有人rss阅读我这篇文章，好感动啊，赶紧写吧，不能辜负这位读者的期望！！

直接贴两端代码

settings.py 中的设置

WHOOSH_INDEX = os.path.join(os.path.dirname(__file__), 'whoosh_index2')

再app 的 __init__.py 文件中添加如下代码：

import os
from django.db.models import signals
from django.conf import settings
from whoosh import fields
from whoosh.index import create_in,open_dir
from test01.models import MyComment

WHOOSH_SCHEMA = fields.Schema(title=fields.TEXT(stored=True),
    content=fields.TEXT(stored=True),)
    #url=fields.ID(stored=True, unique=True))

def create_index(sender=None, **kwargs):
    if not os.path.exists(settings.WHOOSH_INDEX):
        os.mkdir(settings.WHOOSH_INDEX)
        ix = create_in(settings.WHOOSH_INDEX,schema=WHOOSH_SCHEMA,indexname='comment')

signals.post_syncdb.connect(create_index)


def update_index(sender, instance, created, **kwargs):
    ix = open_dir(settings.WHOOSH_INDEX, indexname='comment')
    writer = ix.writer()
    if created:
        writer.add_document(title=unicode(instance.title), content=instance.content,)
        writer.commit()
    else:
        writer.update_document(title=unicode(instance.title), content=instance.content,)
        writer.commit()

signals.post_save.connect(update_index, sender=MyComment)

这是利用django的信号signal机制,当数据保存时候，我们就往whoosh的索引文件中添加信息。

当我们添加结束之后，就可以在views.py 中使用了。whoosh的基本使用方法，我第一篇文章中介绍过了，这里直接使用了。

from django.shortcuts import render_to_response
from whoosh import index
from whoosh.qparser import QueryParser
from settings import WHOOSH_INDEX

def search(request):
    
    ix = index.open_dir(WHOOSH_INDEX, indexname='comment')
    #ix = index.open_dir('E:\whoosh_index', indexname='comment')
    hits = []
    query = request.GET.get('q', None)
    if query is not None and query != u'':
        parser = QueryParser('title', schema=ix.schema)
        try:
            qry = parser.parse(unicode(query))
        except:
            qry = None
        if qry is not None:
            s = ix.searcher()
            hits = s.search(qry)
            
    return render_to_response('sphinx/search.html',
        {'query': query, 'hits': hits})

这是template文件中主要部分，也可直接使用

<form action="" method="get">
    <input type="text" id="id_q" name="q" value="{{ query|default_if_none:"" }}" />
    <input type="submit" value="{% trans "Search" %}"/>
    {% if hits %}
        <ul>
            {% for hit in hits %}
                <li><a href="{{ hit.url }}">{{ hit.title }}</a></li>
            {% endfor %}
        </ul>
    {%else%}
        <p>meiyou</p>
    {% endif %}
</form>

ok，基本就这样，就是利用django的signals机制，简单吧。