72 lines
3 KiB
Python
72 lines
3 KiB
Python
from flask import request, redirect, url_for, render_template
|
|
from sqlalchemy import or_
|
|
from sqlalchemy.orm import aliased
|
|
|
|
from . import main
|
|
from .helpers import inventory_headers, user_headers, worklog_headers
|
|
from .. import db
|
|
from ..models import Inventory, User, WorkLog
|
|
from ..utils.load import eager_load_inventory_relationships, eager_load_user_relationships, eager_load_worklog_relationships
|
|
|
|
@main.route("/search")
|
|
def search():
|
|
query = request.args.get('q', '').strip()
|
|
|
|
if not query:
|
|
return redirect(url_for('main.index'))
|
|
|
|
InventoryAlias = aliased(Inventory)
|
|
UserAlias = aliased(User)
|
|
|
|
inventory_query = eager_load_inventory_relationships(db.session.query(Inventory).join(UserAlias, Inventory.owner)).filter(
|
|
or_(
|
|
Inventory.name.ilike(f"%{query}%"),
|
|
Inventory.serial.ilike(f"%{query}%"),
|
|
Inventory.barcode.ilike(f"%{query}%"),
|
|
Inventory.notes.ilike(f"%{query}%"),
|
|
UserAlias.first_name.ilike(f"%{query}%"),
|
|
UserAlias.last_name.ilike(f"%{query}%"),
|
|
UserAlias.title.ilike(f"%{query}%")
|
|
))
|
|
inventory_results = inventory_query.all()
|
|
user_query = eager_load_user_relationships(db.session.query(User).join(UserAlias, User.supervisor)).filter(
|
|
or_(
|
|
User.first_name.ilike(f"%{query}%"),
|
|
User.last_name.ilike(f"%{query}%"),
|
|
User.title.ilike(f"%{query}%"),
|
|
UserAlias.first_name.ilike(f"%{query}%"),
|
|
UserAlias.last_name.ilike(f"%{query}%"),
|
|
UserAlias.title.ilike(f"%{query}%")
|
|
))
|
|
user_results = user_query.all()
|
|
worklog_query = eager_load_worklog_relationships(db.session.query(WorkLog).join(UserAlias, WorkLog.contact).join(InventoryAlias, WorkLog.work_item)).filter(
|
|
or_(
|
|
WorkLog.notes.ilike(f"%{query}%"),
|
|
UserAlias.first_name.ilike(f"%{query}%"),
|
|
UserAlias.last_name.ilike(f"%{query}%"),
|
|
UserAlias.title.ilike(f"%{query}%"),
|
|
InventoryAlias.name.ilike(f"%{query}%"),
|
|
InventoryAlias.serial.ilike(f"%{query}%"),
|
|
InventoryAlias.barcode.ilike(f"%{query}%")
|
|
))
|
|
worklog_results = worklog_query.all()
|
|
|
|
results = {
|
|
'inventory': {
|
|
'results': inventory_query,
|
|
'headers': inventory_headers,
|
|
'rows': [{"id": item.id, "cells": [fn(item) for fn in inventory_headers.values()]} for item in inventory_results]
|
|
},
|
|
'users': {
|
|
'results': user_query,
|
|
'headers': user_headers,
|
|
'rows': [{"id": user.id, "cells": [fn(user) for fn in user_headers.values()]} for user in user_results]
|
|
},
|
|
'worklog': {
|
|
'results': worklog_query,
|
|
'headers': worklog_headers,
|
|
'rows': [{"id": log.id, "cells": [fn(log) for fn in worklog_headers.values()]} for log in worklog_results]
|
|
}
|
|
}
|
|
|
|
return render_template('search.html', title=f"Database Search ({query})" if query else "Database Search", results=results, query=query)
|