Adding augmented parts

app/crud/augmented_part.py

@@ -0,0 +1,63 @@
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.models.augmented_part import AugmentedPart
+
+_RELATIONSHIPS = ["colour", "manufacturer", "scope_model", "part_type", "sub_part_type"]
+
+
+async def get(db: AsyncSession, augmented_part_id: int) -> AugmentedPart | None:
+    return await db.get(AugmentedPart, augmented_part_id)
+
+
+async def list_all(
+    db: AsyncSession, skip: int = 0, limit: int = 100
+) -> list[AugmentedPart]:
+    result = await db.execute(
+        select(AugmentedPart)
+        .order_by(AugmentedPart.description)
+        .offset(skip)
+        .limit(limit)
+    )
+    return list(result.scalars().all())
+
+
+async def create(
+    db: AsyncSession,
+    *,
+    sku: str | None,
+    description: str,
+    colour_id: int,
+    manufacturer_id: int,
+    scope_model_id: int | None,
+    part_type_id: int,
+    sub_part_type_id: int | None,
+) -> AugmentedPart:
+    ap = AugmentedPart(
+        sku=sku,
+        description=description,
+        colour_id=colour_id,
+        manufacturer_id=manufacturer_id,
+        scope_model_id=scope_model_id,
+        part_type_id=part_type_id,
+        sub_part_type_id=sub_part_type_id,
+    )
+    db.add(ap)
+    await db.commit()
+    await db.refresh(ap, attribute_names=_RELATIONSHIPS)
+    return ap
+
+
+async def update(
+    db: AsyncSession, ap: AugmentedPart, data: dict
+) -> AugmentedPart:
+    for field, value in data.items():
+        setattr(ap, field, value)
+    await db.commit()
+    await db.refresh(ap, attribute_names=_RELATIONSHIPS)
+    return ap
+
+
+async def delete(db: AsyncSession, ap: AugmentedPart) -> None:
+    await db.delete(ap)
+    await db.commit()