Add POST /location-groups endpoint

- todo: link locations to group, blocked on location model

Author: Jam-Cai

backend/python/app/models/location_group.py

@@ -24,7 +24,7 @@ class LocationGroup(LocationGroupBase, BaseModel, table=True):
 class LocationGroupCreate(LocationGroupBase):
     """Location group creation request"""
 
-    pass
+    location_ids: list[int] = Field(min_length=1)
 
 
 class LocationGroupRead(LocationGroupBase):

backend/python/app/routers/location_group_routes.py

@@ -6,7 +6,11 @@
 
 from app.dependencies.auth import require_user_or_admin
 from app.models import get_session
-from app.models.location_group import LocationGroupRead, LocationGroupUpdate
+from app.models.location_group import (
+    LocationGroupCreate,
+    LocationGroupRead,
+    LocationGroupUpdate,
+)
 from app.services.implementations.location_group_service import LocationGroupService
 
 # Initialize service
@@ -16,6 +20,26 @@
 router = APIRouter(prefix="/location-groups", tags=["location-groups"])
 
 
+@router.post("/", response_model=LocationGroupRead, status_code=status.HTTP_201_CREATED)
+async def create_location_group(
+    location_group: LocationGroupCreate,
+    session: AsyncSession = Depends(get_session),
+    _: bool = Depends(require_user_or_admin),
+) -> LocationGroupRead:
+    """
+    Create a new location group
+    """
+    try:
+        new_location_group = await location_group_service.create_location_group(
+            session, location_group
+        )
+        return LocationGroupRead.model_validate(new_location_group)
+    except Exception as e:
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=str(e)
+        ) from e
+
+
 @router.patch("/{location_group_id}", response_model=LocationGroupRead)
 async def update_location_group(
     location_group_id: UUID,

backend/python/app/services/implementations/location_group_service.py

@@ -4,13 +4,41 @@
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlmodel import select
 
-from app.models.location_group import LocationGroup, LocationGroupUpdate
+from app.models.location_group import (
+    LocationGroup,
+    LocationGroupCreate,
+    LocationGroupUpdate,
+)
 
 
 class LocationGroupService:
     def __init__(self, logger: logging.Logger):
         self.logger = logger
 
+    async def create_location_group(
+        self,
+        session: AsyncSession,
+        location_group_data: LocationGroupCreate,
+    ) -> LocationGroup:
+        """Create a new location group"""
+        try:
+            data = location_group_data.model_dump()
+            location_ids = data.pop("location_ids")
+
+            new_location_group = LocationGroup(**data, num_locations=len(location_ids))
+            session.add(new_location_group)
+            await session.commit()
+            await session.refresh(new_location_group)
+
+            # blocked on location model
+            # TODO: update each locations location_group_id foreign key using location_ids
+
+            return new_location_group
+        except Exception as error:
+            self.logger.error(f"Failed to create location group: {error!s}")
+            await session.rollback()
+            raise error
+
     async def update_location_group(
         self,
         session: AsyncSession,