Adding filtering to tables

app/crud/augmented_part.py

@@ -24,6 +24,55 @@ async def list_all(
     return list(result.scalars().all())
 
 
+async def list_filtered(
+    db: AsyncSession,
+    description: str = "",
+    colour_id: int | None = None,
+    manufacturer_id: int | None = None,
+    scope_model_id: int | None = None,
+    part_type_id: int | None = None,
+    sub_part_type_id: int | None = None,
+    skip: int = 0,
+    limit: int = 100,
+) -> list[AugmentedPart]:
+    q = select(AugmentedPart)
+    if description.strip():
+        q = q.where(AugmentedPart.description.ilike(f"%{description.strip()}%"))
+    if colour_id is not None:
+        q = q.where(AugmentedPart.colour_id == colour_id)
+    if manufacturer_id is not None:
+        q = q.where(AugmentedPart.manufacturer_id == manufacturer_id)
+    if scope_model_id is not None:
+        q = q.where(AugmentedPart.scope_model_id == scope_model_id)
+    if part_type_id is not None:
+        q = q.where(AugmentedPart.part_type_id == part_type_id)
+    if sub_part_type_id is not None:
+        q = q.where(AugmentedPart.sub_part_type_id == sub_part_type_id)
+    result = await db.execute(q.order_by(AugmentedPart.description).offset(skip).limit(limit))
+    return list(result.scalars().all())
+
+
+async def get_unlinked_imported_parts_filtered(
+    db: AsyncSession,
+    description: str = "",
+    colour: str = "",
+) -> list[ImportedPart]:
+    q = (
+        select(ImportedPart)
+        .where(
+            ImportedPart.InventoryID.not_in(
+                select(augmented_part_imported_parts.c.imported_part_id)
+            )
+        )
+    )
+    if description.strip():
+        q = q.where(ImportedPart.Description.ilike(f"%{description.strip()}%"))
+    if colour.strip():
+        q = q.where(ImportedPart.TBXColour.ilike(f"%{colour.strip()}%"))
+    result = await db.execute(q.order_by(ImportedPart.InventoryID))
+    return list(result.scalars().all())
+
+
 async def create(
     db: AsyncSession,
     *,

app/crud/imported_part.py

@@ -1,4 +1,4 @@
-from sqlalchemy import insert, select, update
+from sqlalchemy import distinct, insert, select, update
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from app.models.imported_part import ImportedPart
@@ -23,6 +23,28 @@ async def list_all(
     return list(result.scalars().all())
 
 
+async def list_filtered(
+    db: AsyncSession,
+    description: str = "",
+    skip: int = 0,
+    limit: int = 500,
+) -> list[ImportedPart]:
+    q = select(ImportedPart)
+    if description.strip():
+        q = q.where(ImportedPart.Description.ilike(f"%{description.strip()}%"))
+    result = await db.execute(q.order_by(ImportedPart.InventoryID).offset(skip).limit(limit))
+    return list(result.scalars().all())
+
+
+async def distinct_colours(db: AsyncSession) -> list[str]:
+    result = await db.execute(
+        select(distinct(ImportedPart.TBXColour))
+        .where(ImportedPart.TBXColour.isnot(None))
+        .order_by(ImportedPart.TBXColour)
+    )
+    return [row[0] for row in result.all()]
+
+
 async def count(db: AsyncSession) -> int:
     from sqlalchemy import func
     result = await db.execute(select(func.count()).select_from(ImportedPart))