when a country is in the results, restrict further searches to places

[nominatim.git] / nominatim / utils / centroid.py

diff --git a/nominatim/utils/centroid.py b/nominatim/utils/centroid.py
index c2bd61927ef8a57bdd79f02762233fa07f1cdfa1..218231768fb2294b213feb49543cdf35dcdc378f 100644 (file)
--- a/nominatim/utils/centroid.py
+++ b/nominatim/utils/centroid.py
@@ -7,6 +7,7 @@
 """
 Functions for computation of centroids.
 """
 """
 Functions for computation of centroids.
 """

+from typing import Tuple, Any

 from collections.abc import Collection
 
 class PointsCentroid:
 from collections.abc import Collection
 
 class PointsCentroid:


@@ -17,12 +18,12 @@ class PointsCentroid:
         (i.e. in OSM style).
     """
 
         (i.e. in OSM style).
     """
 

-    def __init__(self):
+    def __init__(self) -> None:

         self.sum_x = 0
         self.sum_y = 0
         self.count = 0
 
         self.sum_x = 0
         self.sum_y = 0
         self.count = 0
 

-    def centroid(self):
+    def centroid(self) -> Tuple[float, float]:

         """ Return the centroid of all points collected so far.
         """
         if self.count == 0:
         """ Return the centroid of all points collected so far.
         """
         if self.count == 0:


@@ -32,11 +33,11 @@ class PointsCentroid:
                 float(self.sum_y/self.count)/10000000)
 
 
                 float(self.sum_y/self.count)/10000000)
 
 

-    def __len__(self):
+    def __len__(self) -> int:

         return self.count
 
 
         return self.count
 
 

-    def __iadd__(self, other):
+    def __iadd__(self, other: Any) -> 'PointsCentroid':

         if isinstance(other, Collection) and len(other) == 2:
             if all(isinstance(p, (float, int)) for p in other):
                 x, y = other
         if isinstance(other, Collection) and len(other) == 2:
             if all(isinstance(p, (float, int)) for p in other):
                 x, y = other