Merge pull request #576 from Police-Data-Accessibility-Project/mc_506_pending_agency_edit

Add agency proposal edit logic

Author: maxachis

…s/proposals/agencies/approve/__init__.py …nts/proposals/agencies/by_id/__init__.pysrc/api/endpoints/proposals/agencies/approve/__init__.py renamed to src/api/endpoints/proposals/agencies/by_id/__init__.py src/api/endpoints/proposals/agencies/approve/__init__.py renamed to src/api/endpoints/proposals/agencies/by_id/__init__.py

@@ -3,7 +3,7 @@
 from sqlalchemy.exc import NoResultFound
 from sqlalchemy.ext.asyncio import AsyncSession
 
-from src.api.endpoints.proposals.agencies.approve.response import ProposalAgencyApproveResponse
+from src.api.endpoints.proposals.agencies.by_id.approve.response import ProposalAgencyApproveResponse
 from src.db.models.impl.agency.enums import JurisdictionType, AgencyType
 from src.db.models.impl.agency.sqlalchemy import Agency
 from src.db.models.impl.link.agency_location.sqlalchemy import LinkAgencyLocation

…oints/proposals/agencies/get/__init__.py …osals/agencies/by_id/approve/__init__.pysrc/api/endpoints/proposals/agencies/get/__init__.py renamed to src/api/endpoints/proposals/agencies/by_id/approve/__init__.py src/api/endpoints/proposals/agencies/get/__init__.py renamed to src/api/endpoints/proposals/agencies/by_id/approve/__init__.py

@@ -0,0 +1,30 @@
+from sqlalchemy import delete
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from src.db.models.impl.link.agency_location.sqlalchemy import LinkAgencyLocation
+from src.db.models.impl.proposals.agency_.link__location import ProposalLinkAgencyLocation
+from src.db.queries.base.builder import QueryBuilderBase
+
+
+class DeleteProposalAgencyLocationQueryBuilder(QueryBuilderBase):
+
+    def __init__(
+            self,
+            agency_id: int,
+            location_id: int,
+    ):
+        super().__init__()
+        self.agency_id = agency_id
+        self.location_id = location_id
+
+    async def run(self, session: AsyncSession) -> None:
+        statement = (
+            delete(ProposalLinkAgencyLocation)
+            .where(
+                (ProposalLinkAgencyLocation.proposal_agency_id == self.agency_id)
+                & (ProposalLinkAgencyLocation.location_id == self.location_id)
+            )
+        )
+
+        await session.execute(statement)
+

…ints/proposals/agencies/approve/query.py …roposals/agencies/by_id/approve/query.pysrc/api/endpoints/proposals/agencies/approve/query.py renamed to src/api/endpoints/proposals/agencies/by_id/approve/query.py src/api/endpoints/proposals/agencies/approve/query.py renamed to src/api/endpoints/proposals/agencies/by_id/approve/query.py

@@ -0,0 +1,41 @@
+from typing import Sequence
+
+from sqlalchemy import select, RowMapping
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from src.api.endpoints.agencies.by_id.locations.get.response import AgencyGetLocationsResponse
+from src.api.endpoints.proposals.agencies.by_id.locations.get.response import ProposalAgencyGetLocationsOuterResponse
+from src.db.models.impl.link.agency_location.sqlalchemy import LinkAgencyLocation
+from src.db.models.impl.proposals.agency_.link__location import ProposalLinkAgencyLocation
+from src.db.models.views.location_expanded import LocationExpandedView
+from src.db.queries.base.builder import QueryBuilderBase
+
+
+class GetProposalAgencyLocationsQueryBuilder(QueryBuilderBase):
+
+    def __init__(
+            self,
+            agency_id: int,
+    ):
+        super().__init__()
+        self.agency_id = agency_id
+
+    async def run(self, session: AsyncSession) -> ProposalAgencyGetLocationsOuterResponse:
+        query = (
+            select(
+                ProposalLinkAgencyLocation.location_id,
+                LocationExpandedView.full_display_name
+            )
+            .where(
+                ProposalLinkAgencyLocation.proposal_agency_id == self.agency_id
+            )
+            .join(
+                LocationExpandedView,
+                LocationExpandedView.id == ProposalLinkAgencyLocation.location_id
+            )
+        )
+
+        result: Sequence[RowMapping] = await self.sh.mappings(session, query=query)
+        return ProposalAgencyGetLocationsOuterResponse(
+            results=[AgencyGetLocationsResponse(**row) for row in result]
+        )

…s/proposals/agencies/approve/response.py …osals/agencies/by_id/approve/response.pysrc/api/endpoints/proposals/agencies/approve/response.py renamed to src/api/endpoints/proposals/agencies/by_id/approve/response.py src/api/endpoints/proposals/agencies/approve/response.py renamed to src/api/endpoints/proposals/agencies/by_id/approve/response.py

@@ -0,0 +1,7 @@
+from pydantic import BaseModel
+
+from src.api.endpoints.agencies.by_id.locations.get.response import AgencyGetLocationsResponse
+
+
+class ProposalAgencyGetLocationsOuterResponse(BaseModel):
+    results: list[AgencyGetLocationsResponse]