X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/15beeef6ce762b7c9b7e360e4367738406b28a63..9b636fdc1004f9da3215d9fda350fcd995cf089a:/nominatim/indexer/runners.py?ds=sidebyside diff --git a/nominatim/indexer/runners.py b/nominatim/indexer/runners.py index ac7a0015..c8495ee4 100644 --- a/nominatim/indexer/runners.py +++ b/nominatim/indexer/runners.py @@ -11,14 +11,17 @@ tasks. import functools from psycopg2 import sql as pysql +import psycopg2.extras -from nominatim.indexer.place_info import PlaceInfo +from nominatim.data.place_info import PlaceInfo # pylint: disable=C0111 def _mk_valuelist(template, num): return pysql.SQL(',').join([pysql.SQL(template)] * num) +def _analyze_place(place, analyzer): + return psycopg2.extras.Json(analyzer.process_place(PlaceInfo(place))) class AbstractPlacexRunner: """ Returns SQL commands for indexing of the placex table. @@ -56,7 +59,7 @@ class AbstractPlacexRunner: for place in places: for field in ('place_id', 'name', 'address', 'linked_place_id'): values.append(place[field]) - values.append(PlaceInfo(place).analyze(self.analyzer)) + values.append(_analyze_place(place, self.analyzer)) worker.perform(self._index_sql(len(places)), values) @@ -66,7 +69,7 @@ class RankRunner(AbstractPlacexRunner): """ def name(self): - return "rank {}".format(self.rank) + return f"rank {self.rank}" def sql_count_objects(self): return pysql.SQL("""SELECT count(*) FROM placex @@ -86,7 +89,7 @@ class BoundaryRunner(AbstractPlacexRunner): """ def name(self): - return "boundaries rank {}".format(self.rank) + return f"boundaries rank {self.rank}" def sql_count_objects(self): return pysql.SQL("""SELECT count(*) FROM placex @@ -150,7 +153,7 @@ class InterpolationRunner: values = [] for place in places: values.extend((place[x] for x in ('place_id', 'address'))) - values.append(PlaceInfo(place).analyze(self.analyzer)) + values.append(_analyze_place(place, self.analyzer)) worker.perform(self._index_sql(len(places)), values)