54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
from fastapi import APIRouter, Depends, Query
|
|
from sqlalchemy import select, text
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.database import get_db
|
|
from app.models import Bill, Member
|
|
from app.schemas.schemas import BillSchema, MemberSchema
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("")
|
|
async def search(
|
|
q: str = Query(..., min_length=2),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
# Bill ID direct match
|
|
id_results = await db.execute(
|
|
select(Bill).where(Bill.bill_id.ilike(f"%{q}%")).limit(20)
|
|
)
|
|
id_bills = id_results.scalars().all()
|
|
|
|
# Full-text search on title/content via tsvector
|
|
fts_results = await db.execute(
|
|
select(Bill)
|
|
.where(text("search_vector @@ plainto_tsquery('english', :q)"))
|
|
.order_by(text("ts_rank(search_vector, plainto_tsquery('english', :q)) DESC"))
|
|
.limit(20)
|
|
.params(q=q)
|
|
)
|
|
fts_bills = fts_results.scalars().all()
|
|
|
|
# Merge, dedup, preserve order (ID matches first)
|
|
seen = set()
|
|
bills = []
|
|
for b in id_bills + fts_bills:
|
|
if b.bill_id not in seen:
|
|
seen.add(b.bill_id)
|
|
bills.append(b)
|
|
|
|
# Fuzzy member search
|
|
member_results = await db.execute(
|
|
select(Member)
|
|
.where(Member.name.ilike(f"%{q}%"))
|
|
.order_by(Member.last_name)
|
|
.limit(10)
|
|
)
|
|
members = member_results.scalars().all()
|
|
|
|
return {
|
|
"bills": [BillSchema.model_validate(b) for b in bills],
|
|
"members": [MemberSchema.model_validate(m) for m in members],
|
|
}
|