Adding scope import

app/crud/scope_info.py

@@ -0,0 +1,56 @@
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.models.scope_info import ScopeInfo
+
+
+async def get(db: AsyncSession, scope_info_id: int) -> ScopeInfo | None:
+    return await db.get(ScopeInfo, scope_info_id)
+
+
+async def get_by_augmented_part(
+    db: AsyncSession, augmented_part_id: int
+) -> ScopeInfo | None:
+    result = await db.execute(
+        select(ScopeInfo).where(ScopeInfo.augmented_part_id == augmented_part_id)
+    )
+    return result.scalar_one_or_none()
+
+
+async def create(
+    db: AsyncSession,
+    *,
+    augmented_part_id: int,
+    magnification_min: float | None = None,
+    magnification_max: float | None = None,
+    model_suffix: str | None = None,
+    objective_thread_pitch: str | None = None,
+) -> ScopeInfo:
+    info = ScopeInfo(
+        augmented_part_id=augmented_part_id,
+        magnification_min=magnification_min,
+        magnification_max=magnification_max,
+        model_suffix=model_suffix,
+        objective_thread_pitch=objective_thread_pitch,
+    )
+    db.add(info)
+    await db.commit()
+    await db.refresh(info)
+    return info
+
+
+async def upsert(db: AsyncSession, *, augmented_part_id: int, **fields) -> ScopeInfo:
+    """Create the ScopeInfo for an AugmentedPart, or update it if it exists."""
+    info = await get_by_augmented_part(db, augmented_part_id)
+    if info is None:
+        return await create(db, augmented_part_id=augmented_part_id, **fields)
+    for field, value in fields.items():
+        setattr(info, field, value)
+    await db.commit()
+    await db.refresh(info)
+    return info
+
+
+async def delete(db: AsyncSession, info: ScopeInfo) -> None:
+    await db.delete(info)
+    await db.commit()