add station count to facility API

hiddewie · Dec 15, 2024 · e8627bd · e8627bd
1 parent 6cf7f54
commit e8627bd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/api/openrailwaymap_api/facility_api.py b/api/openrailwaymap_api/facility_api.py
@@ -66,7 +66,7 @@ async def search_by_name(self, q, limit):
                 {fields}, latitude, longitude, rank
               FROM (
                 SELECT
-                    {fields}, ST_X(ST_Transform(geom, 4326)) AS latitude, ST_Y(ST_Transform(geom, 4326)) AS longitude, openrailwaymap_name_rank(phraseto_tsquery('simple', unaccent(openrailwaymap_hyphen_to_space($1))), terms, route_count, railway, station) AS rank
+                    {fields}, ST_X(ST_Transform(geom, 4326)) AS latitude, ST_Y(ST_Transform(geom, 4326)) AS longitude, openrailwaymap_name_rank(phraseto_tsquery('simple', unaccent(openrailwaymap_hyphen_to_space($1))), terms, route_count, station_count, railway, station) AS rank
                   FROM openrailwaymap_facilities_for_search
                   WHERE terms @@ phraseto_tsquery('simple', unaccent(openrailwaymap_hyphen_to_space($1)))
                 ) AS a

diff --git a/api/prepare_facilities.sql b/api/prepare_facilities.sql
@@ -41,6 +41,7 @@ CREATE TABLE openrailwaymap_facilities_for_search AS
       station,
       railway_ref,
       route_count,
+      station_count,
       geom
     FROM (
       SELECT DISTINCT ON (osm_id, key, value, name, railway, station, railway_ref, route_count, geom)
@@ -52,6 +53,7 @@ CREATE TABLE openrailwaymap_facilities_for_search AS
           station,
           railway_ref,
           route_count,
+          station_count,
           geom
         FROM (
           SELECT
@@ -62,7 +64,8 @@ CREATE TABLE openrailwaymap_facilities_for_search AS
               railway_ref,
               name_tags,
               route_count,
-              way AS geom
+              ST_NumGeometries(way) as station_count,
+              ST_Centroid(way) AS geom
             FROM stations_with_route_counts
             WHERE
               railway IN ('station', 'halt', 'tram_stop', 'service_station', 'yard', 'junction', 'spur_junction', 'crossover', 'site')