Update location's location_group fk on location_group create

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"""
 
-    location_ids: list[int] = Field(min_length=1)
+    location_ids: list[UUID] = Field(min_length=1)
 
 
 class LocationGroupRead(LocationGroupBase):

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

@@ -9,6 +9,7 @@
     LocationGroupCreate,
     LocationGroupUpdate,
 )
+from app.models.location import Location
 
 
 class LocationGroupService:
@@ -30,8 +31,18 @@ async def create_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
+            # Update each locations location_group_id foreign key
+            for location_id in location_ids:
+                statement = select(Location).where(Location.location_id == location_id)
+                result = await session.execute(statement)
+                location = result.scalars().first()
+
+                if location:
+                    location.location_group_id = new_location_group.location_group_id
+                else:
+                    self.logger.warning(f"Location with id {location_id} not found")
+
+            await session.commit()
 
             return new_location_group
         except Exception as error: