42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
from app import app, db
|
|
from app.forms.search import AdvancedSearchForm, SearchForm
|
|
from app.models.post import Post
|
|
from app.models.comment import Comment
|
|
from app.models.topic import Topic
|
|
from app.models.forum import Forum
|
|
from app.utils.render import render
|
|
from sqlalchemy import text, func
|
|
from flask import request
|
|
|
|
|
|
def websearch_to_tsquery_multilang(search):
|
|
return func.websearch_to_tsquery('french', search)
|
|
|
|
def to_tsvector_multilang(text):
|
|
return func.to_tsvector('french', text)
|
|
|
|
|
|
@app.route('/rechercher')
|
|
def search():
|
|
form = AdvancedSearchForm(request.args)
|
|
results = list()
|
|
if form.validate():
|
|
# Topics are sorted first in results
|
|
tsquery = websearch_to_tsquery_multilang(form.q.data)
|
|
results = db.session.query(Topic).where(
|
|
to_tsvector_multilang(Topic.title).bool_op('@@')(tsquery)
|
|
).group_by(
|
|
Topic.id,
|
|
Post.id
|
|
)
|
|
# Comments are less important than topics and programs
|
|
res = db.session.query(Comment).where(
|
|
to_tsvector_multilang(Comment.text).bool_op('@@')(tsquery)
|
|
).group_by(
|
|
Comment.id,
|
|
Post.id
|
|
)
|
|
results = list(results) + list(res)
|
|
|
|
return render('search.html', form=form, results=results)
|