diff --git a/src/backend/app/central/central_crud.py b/src/backend/app/central/central_crud.py index f28e506021..84953e9e38 100644 --- a/src/backend/app/central/central_crud.py +++ b/src/backend/app/central/central_crud.py @@ -330,12 +330,23 @@ def list_submissions(project_id: int, odk_central: project_schemas.ODKCentral = def get_form_list(db: Session, skip: int, limit: int): """Returns the list of id and title of xforms from the database.""" try: - return ( + forms = ( db.query(db_models.DbXForm.id, db_models.DbXForm.title) .offset(skip) .limit(limit) .all() ) + + result_dict = [] + for form in forms: + form_dict = { + 'id': form[0], # Assuming the first element is the ID + 'title': form[1] # Assuming the second element is the title + } + result_dict.append(form_dict) + + return result_dict + except Exception as e: log.error(e) raise HTTPException(e) from e diff --git a/src/backend/app/organization/organization_crud.py b/src/backend/app/organization/organization_crud.py index 38159ed89f..df297dd14b 100644 --- a/src/backend/app/organization/organization_crud.py +++ b/src/backend/app/organization/organization_crud.py @@ -22,7 +22,7 @@ import string from fastapi import HTTPException, File,UploadFile import re - +from sqlalchemy import func from sqlalchemy.orm import Session from ..db import db_models @@ -44,15 +44,12 @@ def generate_slug(text: str) -> str: async def get_organisation_by_name(db: Session, name: str): - - # Construct the SQL query with the case-insensitive search - query = f"SELECT * FROM organisations WHERE LOWER(name) LIKE LOWER('%{name}%') LIMIT 1" - - # Execute the query and retrieve the result - result = db.execute(query) - - # Fetch the first row of the result - db_organisation = result.fetchone() + # Use SQLAlchemy's query-building capabilities + db_organisation = ( + db.query(db_models.DbOrganisation) + .filter(func.lower(db_models.DbOrganisation.name).like(func.lower(f'%{name}%'))) + .first() + ) return db_organisation diff --git a/src/backend/app/projects/project_crud.py b/src/backend/app/projects/project_crud.py index d8551eb05c..60ce9b177e 100644 --- a/src/backend/app/projects/project_crud.py +++ b/src/backend/app/projects/project_crud.py @@ -51,6 +51,7 @@ from sqlalchemy import and_, column, func, inspect, select, table from sqlalchemy.dialects.postgresql import insert from sqlalchemy.orm import Session +from sqlalchemy import text from sqlalchemy.sql import text from cpuinfo import get_cpu_info from ..db import database @@ -437,14 +438,13 @@ def remove_z_dimension(coord): ) db.commit() + + # Generate project outline from tasks - # query = f'''SELECT ST_AsText(ST_Buffer(ST_Union(outline), 0.5, 'endcap=round')) as oval_envelope - # FROM tasks - # where project_id={project_id};''' + query = text(f"""SELECT ST_AsText(ST_ConvexHull(ST_Collect(outline))) + FROM tasks + WHERE project_id={project_id};""") - query = f"""SELECT ST_AsText(ST_ConvexHull(ST_Collect(outline))) - FROM tasks - WHERE project_id={project_id};""" log.debug("Generating project outline from tasks") result = db.execute(query) data = result.fetchone() @@ -701,15 +701,15 @@ def process_polygon(db:Session, project_id:uuid.UUID, boundary_data:str, no_of_b db.commit() else: # Remove the polygons outside of the project AOI using a parameterized query - query = f""" + query = text(f""" DELETE FROM ways_poly WHERE NOT ST_Within(ST_Centroid(ways_poly.geom), (SELECT geom FROM project_aoi WHERE project_id = '{project_id}')); - """ + """) result = db.execute(query) db.commit() with open('app/db/split_algorithm.sql', 'r') as sql_file: query = sql_file.read() - result = db.execute(query, params={'num_buildings': no_of_buildings}) + result = db.execute(text(query), params={'num_buildings': no_of_buildings}) result = result.fetchall() db.query(db_models.DbBuildings).delete() db.query(db_models.DbOsmLines).delete() @@ -1214,7 +1214,7 @@ def generate_task_files( # Get the features for this task. # Postgis query to filter task inside this task outline and of this project # Update those features and set task_id - query = f"""UPDATE features + query = text(f"""UPDATE features SET task_id={task_id} WHERE id IN ( SELECT id @@ -1223,12 +1223,12 @@ def generate_task_files( AND ST_IsValid(geometry) AND ST_IsValid('{task.outline}'::Geometry) AND ST_Contains('{task.outline}'::Geometry, ST_Centroid(geometry)) - )""" + )""") result = db.execute(query) # Get the geojson of those features for this task. - query = f"""SELECT jsonb_build_object( + query = text(f"""SELECT jsonb_build_object( 'type', 'FeatureCollection', 'features', jsonb_agg(feature) ) @@ -1241,9 +1241,10 @@ def generate_task_files( ) AS feature FROM features WHERE project_id={project_id} and task_id={task_id} - ) features;""" + ) features;""") result = db.execute(query) + features = result.fetchone()[0] upload_media = False if features['features'] is None else True diff --git a/src/backend/app/projects/project_schemas.py b/src/backend/app/projects/project_schemas.py index 0ec20a2709..77b60c662d 100644 --- a/src/backend/app/projects/project_schemas.py +++ b/src/backend/app/projects/project_schemas.py @@ -16,7 +16,7 @@ # along with FMTM. If not, see . # -from typing import List, Union +from typing import List, Union, Optional from geojson_pydantic import Feature from pydantic import BaseModel @@ -50,26 +50,33 @@ class BETAProjectUpload(BaseModel): xform_title: Union[str, None] odk_central: ODKCentral hashtags: Union[List[str], None] - organisation_id: int = None + organisation_id: Optional[int] = None # city: str # country: str +class Feature(BaseModel): + id: int + project_id: int + task_id: Optional[int] = None + geometry: Optional[Feature] = None + + class ProjectSummary(BaseModel): id: int = -1 priority: ProjectPriority = ProjectPriority.MEDIUM priority_str: str = priority.name - title: str = None - location_str: str = None - description: str = None - num_contributors: int = None - total_tasks: int = None - tasks_mapped: int = None - tasks_validated: int = None - tasks_bad: int = None - hashtags: List[str] = None - organisation_id: int = None - organisation_logo: str = None + title: Optional[str] = None + location_str: Optional[str] = None + description: Optional[str] = None + total_tasks: Optional[int] = None + tasks_mapped: Optional[int] = None + num_contributors: Optional[int] = None + tasks_validated: Optional[int] = None + tasks_bad: Optional[int] = None + hashtags: Optional[List[str]] = None + organisation_id: Optional[int] = None + organisation_logo: Optional[str] = None class ProjectBase(BaseModel): @@ -79,11 +86,11 @@ class ProjectBase(BaseModel): project_info: List[ProjectInfo] status: ProjectStatus # location_str: str - outline_geojson: Feature = None - project_tasks: List[tasks_schemas.Task] = None - xform_title: str = None - hashtags: List[str] = None - organisation_id: int = None + # outline_geojson: Optional[Feature] + project_tasks: Optional[List[tasks_schemas.Task]] + xform_title: Optional[str] = None + hashtags: Optional[List[str]] = None + organisation_id: Optional[int] = None class ProjectOut(ProjectBase): @@ -91,8 +98,3 @@ class ProjectOut(ProjectBase): -class Feature(BaseModel): - id: int - project_id: int - task_id: int = None - geometry: Feature diff --git a/src/backend/app/tasks/tasks_crud.py b/src/backend/app/tasks/tasks_crud.py index fd4066945c..a1cc26c9e3 100644 --- a/src/backend/app/tasks/tasks_crud.py +++ b/src/backend/app/tasks/tasks_crud.py @@ -50,10 +50,18 @@ async def get_task_count_in_project(db: Session, project_id: int): def get_task_lists(db: Session, project_id: int): """Get a list of tasks for a project.""" - query = f"""select id from tasks where project_id = {project_id}""" - result = db.execute(query) - tasks = [task[0] for task in result.fetchall()] - return tasks + query = text(""" + SELECT id + FROM tasks + WHERE project_id = :project_id + """) + + # Then execute the query with the desired parameter + result = db.execute(query, {"project_id": project_id}) + + # Fetch the result + task_ids = [row.id for row in result] + return task_ids def get_tasks( diff --git a/src/backend/app/tasks/tasks_routes.py b/src/backend/app/tasks/tasks_routes.py index e97a3dbed3..f6e08d851a 100644 --- a/src/backend/app/tasks/tasks_routes.py +++ b/src/backend/app/tasks/tasks_routes.py @@ -29,6 +29,7 @@ from . import tasks_crud, tasks_schemas from ..projects import project_crud, project_schemas from ..central import central_crud +from sqlalchemy.sql import text router = APIRouter( @@ -89,16 +90,15 @@ async def get_point_on_surface( List[Tuple[int, str]]: A list of tuples containing the task ID and the centroid as a string. """ - query = f""" + query = text(f""" SELECT id, ARRAY_AGG(ARRAY[ST_X(ST_PointOnSurface(outline)), ST_Y(ST_PointOnSurface(outline))]) AS point FROM tasks WHERE project_id = {project_id} - GROUP BY id; - """ + GROUP BY id; """) result = db.execute(query) - result = result.fetchall() - return result + result_dict_list = [{"id": row[0], "point": row[1]} for row in result.fetchall()] + return result_dict_list @router.post("/near_me", response_model=tasks_schemas.TaskOut) @@ -178,9 +178,9 @@ async def task_features_count( ) def process_task(task): - feature_count_query = f""" - select count(*) from features where project_id = {project_id} and task_id = {task} - """ + feature_count_query = text(f""" + select count(*)from features where project_id = {project_id} and task_id = {task} + """) result = db.execute(feature_count_query) feature_count = result.fetchone() @@ -190,7 +190,7 @@ def process_task(task): # form_details = central_crud.get_form_full_details(project.odkid, task, odk_credentials) return { "task_id": task, - "feature_count": feature_count["count"], + "feature_count": feature_count[0], # 'submission_count': form_details['submissions'], "submission_count": len(submission_list) if isinstance(submission_list, list)