Refaktoring

src/graphql_schema/entities/resolvers/photo.py

@@ -0,0 +1,40 @@
+from sqlalchemy import update
+from sqlalchemy.ext.asyncio import AsyncSession
+from database import models
+from database.transaction import get_session
+from graphql_schema.entities.helpers.combobox import handle_combobox_save
+from graphql_schema.entities.resolvers.base import BaseMutationResolver
+from graphql_schema.entities.types.mutation_input import EditPhotoInput
+from graphql_schema.entities.types.types import Photo
+
+
+class PhotoMutationResolver(BaseMutationResolver):
+    def __init__(self):
+        super().__init__(Photo, models.Photo)
+
+    async def reset_flight_cover(self, db: AsyncSession, flight_id: int):
+        (await db.execute(
+            update(models.Photo)
+            .filter(models.Photo.flight_id == flight_id)
+            .filter(models.Photo.id != id).values(is_flight_cover=False))
+         )
+
+    async def update(self, id: int, input: EditPhotoInput, user_id: int) -> Photo:
+        data = input.to_dict()
+
+        async with get_session() as db:
+            photo = await self._get_one(db, id, created_by_id=user_id)
+            if input.point_of_interest:
+                data['point_of_interest_id'] = await handle_combobox_save(
+                    db,
+                    models.PointOfInterest,
+                    input.point_of_interest,
+                    user_id,
+                    extra_data={"description": ""}
+                )
+
+            if input.is_flight_cover:
+                # reset other covers
+                await self.reset_flight_cover(db, photo.flight_id)
+
+            return await self._do_update(db, obj=photo, data=data)