个人网站制作 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)