feat: Add filtering, searching, and sorting to user extractions retrieval
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
"""Extraction repository for database operations."""
|
||||
|
||||
from sqlalchemy import desc
|
||||
from sqlalchemy import asc, desc, or_
|
||||
from sqlmodel import select
|
||||
from sqlmodel.ext.asyncio.session import AsyncSession
|
||||
|
||||
@@ -55,3 +55,39 @@ class ExtractionRepository(BaseRepository[Extraction]):
|
||||
.order_by(desc(Extraction.created_at)),
|
||||
)
|
||||
return list(result.all())
|
||||
|
||||
async def get_user_extractions_filtered(
|
||||
self,
|
||||
user_id: int,
|
||||
search: str | None = None,
|
||||
sort_by: str = "created_at",
|
||||
sort_order: str = "desc",
|
||||
status_filter: str | None = None,
|
||||
) -> list[Extraction]:
|
||||
"""Get extractions for a user with filtering, search, and sorting."""
|
||||
query = select(Extraction).where(Extraction.user_id == user_id)
|
||||
|
||||
# Apply search filter
|
||||
if search:
|
||||
search_pattern = f"%{search}%"
|
||||
query = query.where(
|
||||
or_(
|
||||
Extraction.title.ilike(search_pattern),
|
||||
Extraction.url.ilike(search_pattern),
|
||||
Extraction.service.ilike(search_pattern),
|
||||
)
|
||||
)
|
||||
|
||||
# Apply status filter
|
||||
if status_filter:
|
||||
query = query.where(Extraction.status == status_filter)
|
||||
|
||||
# Apply sorting
|
||||
sort_column = getattr(Extraction, sort_by, Extraction.created_at)
|
||||
if sort_order.lower() == "asc":
|
||||
query = query.order_by(asc(sort_column))
|
||||
else:
|
||||
query = query.order_by(desc(sort_column))
|
||||
|
||||
result = await self.session.exec(query)
|
||||
return list(result.all())
|
||||
|
||||
Reference in New Issue
Block a user