tommy/libpostal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[fix] save_index

Browse Source
This commit is contained in:
Al
2015-10-31 17:59:44 -04:00
parent 0227d9335f
commit c54fb412a3
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/geodata/polygons/index.py
View File

@@ -162,7 +162,7 @@ class PolygonIndex(object):
def save(self, polys_filename=DEFAULT_POLYS_FILENAME): def save(self, polys_filename=DEFAULT_POLYS_FILENAME):
self.save_polygons(os.path.join(self.save_dir, polys_filename)) self.save_polygons(os.path.join(self.save_dir, polys_filename))
self.save_index(index_filename=self.index_path) self.save_index()
def save_polygons(self, out_filename): def save_polygons(self, out_filename):
out = open(out_filename, 'w') out = open(out_filename, 'w')
@@ -295,6 +295,8 @@ class GeohashPolygonIndex(PolygonIndex):
return candidates return candidates
def save_index(self): def save_index(self):
if not self.index_path:
self.index_path = os.path.join(self.save_dir or '.', self.INDEX_FILENAME)
json.dump(self.index, open(self.index_path, 'w')) json.dump(self.index, open(self.index_path, 'w'))
@classmethod @classmethod