个人网站制作 Part 13 添加搜索功能[Elasticsearch] | Web开发项目

# 导入必要的模块
from flask import Flask, render_template, request, redirect, url_for
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate
from flask_bootstrap import Bootstrap
from flask_moment import Moment
from elasticsearch import Elasticsearch
 
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///db.sqlite'
db = SQLAlchemy(app)
migrate = Migrate(app, db)
bootstrap = Bootstrap(app)
moment = Moment(app)
 
# 设定Elasticsearch客户端
app.config['ELASTICSEARCH_URL'] = 'http://localhost:9200/'
es = Elasticsearch([app.config['ELASTICSEARCH_URL']])
 
# 定义搜索函数
def search(query):
    results = es.search(index="posts", doc_type="post", q=query)
    ids = [int(result['_id']) for result in results['hits']['hits']]
    return Post.query.filter(Post.id.in_(ids)).all()
 
# 定义模型
class Post(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(50))
    body = db.Column(db.String(2000))
 
    def __repr__(self):
        return '<Post %r>' % self.title
 
# 定义主页视图函数
@app.route('/', methods=['GET', 'POST'])
def index():
    if request.method == 'POST':
        query = request.form['search']
        results = search(query)
        return render_template('index.html', posts=results)
    return render_template('index.html')
 
if __name__ == '__main__':
    app.run(debug=True)

这段代码示例展示了如何在Flask应用中集成Elasticsearch，并实现了一个简单的搜索功能。首先，我们配置了Elasticsearch客户端并指定了Elasticsearch服务的URL。然后，我们定义了一个搜索函数，它使用Elasticsearch客户端来查询索引为"posts"的文档类型为"post"的数据。最后，我们修改了主页的视图函数，使其能在接收到POST请求时处理搜索逻辑，并返回匹配搜索查询的文章列表。