Skip to content

Commit 4257ee0

Browse files
committed
use enumerate in geojson functions, avoid bare except in mapswipe worker check input #252
1 parent 297409d commit 4257ee0

File tree

2 files changed

+6
-7
lines changed

2 files changed

+6
-7
lines changed

mapswipe_workers/mapswipe_workers/mapswipe_workers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ class PythonLiteralOption(click.Option):
2929
def type_cast_value(self, ctx, value):
3030
try:
3131
return ast.literal_eval(value)
32-
except Exception as e:
32+
except ValueError as e:
3333
logger.exception(e)
3434
raise click.BadParameter(value)
3535

mapswipe_workers/mapswipe_workers/utils/geojson_functions.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -178,24 +178,23 @@ def create_geojson_file(geometries, outfile):
178178
layer = dataSource.CreateLayer(outfile, srs, geom_type=ogr.wkbPolygon)
179179

180180
# create fields
181-
field_id = ogr.FieldDefn("id", ogr.OFTString)
181+
field_id = ogr.FieldDefn("id", ogr.OFTInteger)
182182
layer.CreateField(field_id)
183183

184-
counter = 0
185184
if not geometries:
186185
logger.info("there are no geometries to save")
187186
else:
188-
for geom in geometries:
189-
counter += 1
187+
for counter, geom in enumerate(geometries):
190188
# init feature
191189
featureDefn = layer.GetLayerDefn()
192190
feature = ogr.Feature(featureDefn)
193191
# create polygon from wkt and set geometry
194192
feature.SetGeometry(geom)
195193
# set other attributes
196-
feature.SetField("id", counter)
194+
# set first id to 1 instead of 0
195+
feature.SetField("id", counter + 1)
197196
# add feature to layer
198197
layer.CreateFeature(feature)
199198

200199
layer = None
201-
logger.info("created outifle: %s." % outfile)
200+
logger.info("created outfile: %s." % outfile)

0 commit comments

Comments
 (0)