Fix get_test_polygons script
This commit is contained in:
parent
342167f9c9
commit
322a976f7a
2
.gitignore
vendored
2
.gitignore
vendored
@ -11,4 +11,4 @@ app/start.sh
|
|||||||
etl/cache/*
|
etl/cache/*
|
||||||
etl/images/*
|
etl/images/*
|
||||||
etl/*.geojson
|
etl/*.geojson
|
||||||
etl/*.txt
|
etl/*.csv
|
||||||
|
@ -8,6 +8,8 @@ Then run:
|
|||||||
"""
|
"""
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
import os
|
import os
|
||||||
|
import subprocess
|
||||||
|
|
||||||
import osmnx
|
import osmnx
|
||||||
|
|
||||||
# configure logging/caching
|
# configure logging/caching
|
||||||
@ -28,7 +30,9 @@ fig, ax = osmnx.plot_buildings(gdf_proj, bgcolor='#333333', color='w', figsize=(
|
|||||||
filename='test_buildings_preview', dpi=600)
|
filename='test_buildings_preview', dpi=600)
|
||||||
|
|
||||||
# save
|
# save
|
||||||
test_data_file = os.path.join(os.path.dirname(__file__), 'test_buildings.csv')
|
test_dir = os.path.dirname(__file__)
|
||||||
|
test_data_geojson = str(os.path.join(test_dir, 'test_buildings.geojson'))
|
||||||
|
subprocess.run(["rm", test_data_geojson])
|
||||||
|
|
||||||
gdf_to_save = gdf_proj.reset_index(
|
gdf_to_save = gdf_proj.reset_index(
|
||||||
)[
|
)[
|
||||||
@ -38,5 +42,13 @@ gdf_to_save = gdf_proj.reset_index(
|
|||||||
gdf_to_save.rename(
|
gdf_to_save.rename(
|
||||||
columns={'index': 'fid'}
|
columns={'index': 'fid'}
|
||||||
).to_file(
|
).to_file(
|
||||||
test_data_file, driver='CSV'
|
test_data_geojson, driver='GeoJSON'
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# convert to CSV
|
||||||
|
test_data_csv = str(os.path.join(test_dir, 'test_buildings.csv'))
|
||||||
|
subprocess.run(["rm", test_data_csv])
|
||||||
|
subprocess.run(["ogr2ogr", "-f", "CSV", test_data_csv, test_data_geojson, "-lco", "GEOMETRY=AS_WKT"])
|
||||||
|
|
||||||
|
# add SRID for ease of loading to PostgreSQL
|
||||||
|
subprocess.run(["sed", "-i", "s/^\"POLYGON/\"SRID=3857;POLYGON/", test_data_csv])
|
||||||
|
Loading…
Reference in New Issue
Block a user