zele-backend/routes.py

import os
from flask import jsonify, request
from sqlalchemy import inspect, func, text
from sqlalchemy.orm import sessionmaker
from geoalchemy2 import Geometry
from shapely import wkb

def register_route(app,db):
    @app.route("/")
    def hello_world():
        return "<p>Hello, World!</p>"

    @app.route('/table')
    def list_table():
        inspector = inspect(db.engine)
        tables = inspector.get_table_names()
        return jsonify(tables)
    
    @app.route('/find_points/', methods=['POST'])
    def find_points():
        data = request.json
        coordinates = data.get('coordinates', [])
        polygon_wkt = 'POLYGON(({}))'.format(', '.join(f'{lat} {lng}' for lat, lng in coordinates))
    
        # Construct the query string
        query = f"""
        SELECT id, geom FROM agents
        WHERE ST_Contains(
        ST_Transform(ST_GeomFromText('{polygon_wkt}', 4326), 4326),
        geom
        )
        """
    
        with db.engine.connect() as connection:
            results = connection.execute(text(query)).fetchall()
    
        # Convert results to a list of dictionaries for JSON serialization
        results_list = [{"id": result.id, "geom": [wkb.loads(bytes.fromhex(result.geom)).x, wkb.loads(bytes.fromhex(result.geom)).y]} for result in results]

        return jsonify(results_list)