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,
     *,