add species query

Author: sonnenteich

app/api/tree.py

@@ -11,7 +11,8 @@
 )
 from ..dependencies import get_session
 from ..services.tree import (
-    get_tree_by_id
+    get_tree_by_id,
+    get_tree_by_species
 )
 
 route_street_tree = APIRouter(prefix='/street_tree/v1')
@@ -54,3 +55,24 @@ async def fetch_tree_by_id(
         )
 
     return rows
+
+@route_street_tree.get(
+    '/species',
+    response_model=StreetTreeResponse,
+    tags=['Strassenbaeume'],
+    description=(
+        'Retrieves street tree details based on the provided tree id.'
+    )
+)
+async def fetch_tree_by_species(
+    session: AsyncSession = Depends(get_session)
+) -> List[StreetTreeResponse]:
+    rows = await get_tree_by_species(session)
+
+    if not rows:
+        raise HTTPException(
+            status_code=status.HTTP_404_NOT_FOUND,
+            detail=f'No matches found'
+        )
+
+    return rows

app/services/tree.py

@@ -1,5 +1,8 @@
+from sqlalchemy import select, case, literal_column, func
+from sqlalchemy.sql import exists, and_, not_
+from sqlalchemy.orm import aliased
+from geoalchemy2 import functions as geofunc
 from sqlalchemy.ext.asyncio import AsyncSession
-from sqlmodel import select
 from app.models.tree import StreetTreeRegister
 
 async def get_tree_by_id(session: AsyncSession, tree_id: int):
@@ -10,3 +13,64 @@ async def get_tree_by_id(session: AsyncSession, tree_id: int):
     row = result.scalars().first()
 
     return row
+
+async def get_tree_by_species(session: AsyncSession):
+    # Alias for subquery
+    gef = aliased(StreetTreeRegister)
+
+    stmt = (
+        select(
+            StreetTreeRegister.id,
+            StreetTreeRegister.tree_number,
+            StreetTreeRegister.street,
+            StreetTreeRegister.species,
+            StreetTreeRegister.type,
+            func.round(
+                func.ST_X(func.ST_Transform(StreetTreeRegister.geom, 4326)).cast('numeric'),
+                6
+            ).label("lon"),
+            func.round(
+                func.ST_Y(func.ST_Transform(StreetTreeRegister.geom, 4326)).cast('numeric'),
+                6
+            ).label("lat"),
+            case(
+                (
+                    StreetTreeRegister.species.ilike("%Tilia%"), 1,
+                ),
+                (
+                    StreetTreeRegister.species.ilike("%Acer%"), 2,
+                ),
+                (
+                    StreetTreeRegister.species.ilike("%Quercus%"), 3,
+                ),
+                (
+                    StreetTreeRegister.species.ilike("%Fagus%"), 4,
+                ),
+                (
+                    StreetTreeRegister.species.ilike("%Betula%"), 5,
+                ),
+                (
+                    StreetTreeRegister.species.ilike("%Carpinus%"), 6,
+                ),
+                else_=0,
+            ).label("species_index")
+        )
+        .where(
+            StreetTreeRegister.type == 'bestand',
+            not_(
+                exists()
+                .where(
+                    and_(
+                        gef.type == 'gefaellt',
+                        gef.tree_number == StreetTreeRegister.tree_number,
+                        gef.street == StreetTreeRegister.street,
+                    )
+                )
+            )
+        )
+        .order_by(StreetTreeRegister.species)
+    )
+
+    result = await session.execute(stmt)
+
+    return result.scalars().all()