95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
from flask import Blueprint, render_template, request, redirect, url_for, jsonify
|
|
from flask_login import login_required, current_user
|
|
from app import db
|
|
from sqlalchemy import or_
|
|
from plugins.plant.models import Plant, PlantCommonName, PlantScientificName, Tag
|
|
from flask_wtf import FlaskForm
|
|
from wtforms import StringField, SelectMultipleField, SubmitField
|
|
from wtforms.validators import Optional, Length, Regexp
|
|
|
|
bp = Blueprint(
|
|
'search',
|
|
__name__,
|
|
url_prefix='/search',
|
|
template_folder='templates/search'
|
|
)
|
|
|
|
class SearchForm(FlaskForm):
|
|
query = StringField(
|
|
'Search',
|
|
validators=[
|
|
Optional(),
|
|
Length(min=2, max=100, message="Search term must be between 2 and 100 characters."),
|
|
Regexp(r'^[\w\s\-]+$', message="Search can only include letters, numbers, spaces, and dashes.")
|
|
]
|
|
)
|
|
tags = SelectMultipleField('Tags', coerce=int)
|
|
submit = SubmitField('Search')
|
|
|
|
@bp.route('', methods=['GET', 'POST'])
|
|
@bp.route('/', methods=['GET', 'POST'])
|
|
@login_required
|
|
def search():
|
|
form = SearchForm()
|
|
form.tags.choices = [(t.id, t.name) for t in Tag.query.order_by(Tag.name).all()]
|
|
if form.validate_on_submit():
|
|
q = form.query.data or ''
|
|
selected = form.tags.data or []
|
|
tags_param = ','.join(map(str, selected)) if selected else ''
|
|
return redirect(url_for('search.results', q=q, tags=tags_param))
|
|
return render_template('search/search.html', form=form)
|
|
|
|
@bp.route('/results', methods=['GET'])
|
|
@login_required
|
|
def results():
|
|
q = request.args.get('q', '').strip()
|
|
tags_param = request.args.get('tags', '')
|
|
tags = [int(t) for t in tags_param.split(',') if t] if tags_param else []
|
|
like_term = f"%{q}%"
|
|
|
|
# Base query, joining name tables for text search
|
|
query = (
|
|
db.session.query(Plant)
|
|
.join(Plant.common_name)
|
|
.join(Plant.scientific_name)
|
|
)
|
|
|
|
if q:
|
|
query = query.filter(
|
|
or_(
|
|
PlantCommonName.name.ilike(like_term),
|
|
PlantScientificName.name.ilike(like_term),
|
|
Plant.plant_type.ilike(like_term),
|
|
)
|
|
)
|
|
|
|
if tags:
|
|
query = query.filter(Plant.tags.any(Tag.id.in_(tags)))
|
|
|
|
# Only include active plants…
|
|
query = query.filter(Plant.is_active.is_(True))
|
|
|
|
# …and either public plants or those owned by the current user
|
|
query = query.filter(
|
|
or_(
|
|
Plant.is_public.is_(True),
|
|
Plant.owner_id == current_user.id
|
|
)
|
|
)
|
|
|
|
results = query.all()
|
|
return render_template(
|
|
'search/results.html',
|
|
results=results,
|
|
query=q,
|
|
tags=tags
|
|
)
|
|
|
|
@bp.route('/tags')
|
|
@login_required
|
|
def search_tags():
|
|
term = request.args.get('term', '')
|
|
like_term = f"%{term}%"
|
|
matches = Tag.query.filter(Tag.name.ilike(like_term)).limit(10).all()
|
|
return jsonify([t.name for t in matches])
|