From 9a6ec41ce624516a4467f980f2eb3b7231dddb22 Mon Sep 17 00:00:00 2001 From: Al Date: Fri, 19 Aug 2016 01:01:05 -0400 Subject: [PATCH] [points] Adding __iter__ and __len__ to point index --- scripts/geodata/points/index.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/scripts/geodata/points/index.py b/scripts/geodata/points/index.py index ea2c19ac..31258024 100644 --- a/scripts/geodata/points/index.py +++ b/scripts/geodata/points/index.py @@ -139,6 +139,14 @@ class PointIndex(object): point_index.load_properties(os.path.join(d, cls.PROPS_FILENAME)) return point_index + def __iter__(self): + for i in xrange(self.i): + lat, lon = self.points[i * 2], self.points[i * 2 + 1] + yield self.get_properties(i), lat, lon + + def __len__(self): + return self.i + def get_candidate_points(self, latitude, longitude): code = geohash.encode(latitude, longitude)[:self.precision] candidates = OrderedDict()