fix flake8 linting errors

Author: felixevers

backend/api/context.py

@@ -84,7 +84,7 @@ async def get_context(
 
         logger = logging.getLogger(__name__)
         organizations_raw = user_payload.get("organization")
-        
+
         if organizations_raw is None:
             scope = user_payload.get("scope", "")
             has_org_scope = "organization" in scope.split() if scope else False
@@ -94,7 +94,7 @@ async def get_context(
                 f"Token scope: {scope}. "
                 f"Available claims: {sorted(user_payload.keys())}"
             )
-        
+
         organizations = None
         if organizations_raw:
             if isinstance(organizations_raw, list):
@@ -157,7 +157,7 @@ async def get_context(
 
             if db_user:
                 try:
-                    
+
                     await _update_user_root_locations(
                         session,
                         db_user,
@@ -185,17 +185,9 @@ async def _update_user_root_locations(
             if org_id.strip()
         ]
 
-    logger = logging.getLogger(__name__)
-
     root_location_ids: list[str] = []
 
     if organization_ids:
-        org_check = await session.execute(
-            select(location_organizations.c.organization_id, location_organizations.c.location_id)
-            .where(location_organizations.c.organization_id.in_(organization_ids))
-        )
-        org_entries = org_check.all()
-        
         result = await session.execute(
             select(LocationNode)
             .join(
@@ -239,8 +231,7 @@ async def _update_user_root_locations(
                     ),
                 )
                 root_location_ids.append(new_location.id)
-    
-    
+
     if not root_location_ids:
         personal_org_title = f"{user.username}'s Organization"
         result = await session.execute(

backend/api/resolvers/location.py

@@ -20,10 +20,10 @@ async def location_roots(self, info: Info) -> list[LocationNodeType]:
         accessible_location_ids = await auth_service.get_user_accessible_location_ids(
             info.context.user, info.context
         )
-        
+
         if not accessible_location_ids:
             return []
-        
+
         result = await info.context.db.execute(
             select(models.LocationNode)
             .where(

backend/api/resolvers/patient.py

@@ -68,11 +68,11 @@ async def patients(
         accessible_location_ids = await auth_service.get_user_accessible_location_ids(
             info.context.user, info.context
         )
-        
+
         # If user has no accessible locations, return empty list
         if not accessible_location_ids:
             return []
-        
+
         query = auth_service.filter_patients_by_access(
             info.context.user, query, accessible_location_ids
         )
@@ -109,7 +109,7 @@ async def patients(
         if filter_cte is not None:
             patient_locations_filter = aliased(models.patient_locations)
             patient_teams_filter = aliased(models.patient_teams)
-            
+
             query = (
                 query.outerjoin(
                     patient_locations_filter,

backend/api/resolvers/task.py

@@ -82,7 +82,7 @@ async def tasks(
             cte, models.LocationNode.parent_id == cte.c.id
         )
         cte = cte.union_all(children)
-        
+
         if root_location_ids:
             invalid_ids = [lid for lid in root_location_ids if lid not in accessible_location_ids]
             if invalid_ids:
@@ -101,7 +101,7 @@ async def tasks(
             root_cte = root_cte.union_all(root_children)
         else:
             root_cte = cte
-        
+
         query = (
             select(models.Task)
             .options(

backend/api/types/user.py

@@ -3,7 +3,6 @@
 import strawberry
 from database import models
 from sqlalchemy import select
-from sqlalchemy.orm import aliased, selectinload
 
 if TYPE_CHECKING:
     from api.types.location import LocationNodeType
@@ -37,31 +36,31 @@ async def tasks(
         info,
     ) -> list[Annotated["TaskType", strawberry.lazy("api.types.task")]]:
         from api.services.authorization import AuthorizationService
-        
+
         auth_service = AuthorizationService(info.context.db)
         accessible_location_ids = await auth_service.get_user_accessible_location_ids(
             info.context.user, info.context
         )
-        
+
         if not accessible_location_ids:
             return []
-        
+
         from sqlalchemy.orm import aliased
         patient_locations = aliased(models.patient_locations)
         patient_teams = aliased(models.patient_teams)
-        
+
         from sqlalchemy import select
         cte = (
             select(models.LocationNode.id)
             .where(models.LocationNode.id.in_(accessible_location_ids))
             .cte(name="accessible_locations", recursive=True)
         )
-        
+
         children = select(models.LocationNode.id).join(
             cte, models.LocationNode.parent_id == cte.c.id
         )
         cte = cte.union_all(children)
-        
+
         query = (
             select(models.Task)
             .join(models.Patient, models.Task.patient_id == models.Patient.id)
@@ -91,7 +90,7 @@ async def tasks(
             )
             .distinct()
         )
-        
+
         result = await info.context.db.execute(query)
         return result.scalars().all()
 
@@ -102,7 +101,7 @@ async def root_locations(
     ) -> list[Annotated["LocationNodeType", strawberry.lazy("api.types.location")]]:
         import logging
         logger = logging.getLogger(__name__)
-        
+
         # First check what's in user_root_locations table
         user_root_check = await info.context.db.execute(
             select(models.user_root_locations.c.location_id).where(
@@ -111,7 +110,7 @@ async def root_locations(
         )
         user_root_location_ids = [row[0] for row in user_root_check.all()]
         logger.info(f"User {self.id} has {len(user_root_location_ids)} entries in user_root_locations: {user_root_location_ids}")
-        
+
         result = await info.context.db.execute(
             select(models.LocationNode)
             .join(
@@ -123,7 +122,7 @@ async def root_locations(
         )
         locations = result.scalars().all()
         logger.info(f"User {self.id} root_locations query returned {len(locations)} locations: {[loc.id for loc in locations]}")
-        
+
         # If we have user_root_locations entries but no locations returned, check if locations exist
         if user_root_location_ids and not locations:
             location_check = await info.context.db.execute(
@@ -137,5 +136,5 @@ async def root_locations(
                 f"Checking if locations exist: {[loc.id for loc in existing_locations]} "
                 f"with parent_ids: {[loc.parent_id for loc in existing_locations]}"
             )
-        
+
         return locations