Cartography/prettymaps
1
0
Fork 0
mirror of https://github.com/marceloprates/prettymaps.git synced 2025-08-07 23:26:46 +02:00
Code Issues Projects Releases Wiki Activity

Merge pull request #57 from sacovo/patch-2

Browse Source 
catch empty response instead of failing
This commit is contained in:
Marcelo de Oliveira Rosa Prates
2021-09-13 16:45:46 -03:00
committed by GitHub
parent 660019b60b 74b58711e3
commit 18c4dd704c
1 changed files with 30 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
prettymaps/fetch.py
View File

@@ -6,6 +6,7 @@ from xmlrpc.client import Boolean
import osmnx as ox import osmnx as ox
from osmnx import utils_geo from osmnx import utils_geo
from osmnx._errors import EmptyOverpassResponse
import numpy as np import numpy as np
from shapely.geometry import Point, Polygon, MultiPolygon, LineString, MultiLineString from shapely.geometry import Point, Polygon, MultiPolygon, LineString, MultiLineString
from shapely.ops import unary_union from shapely.ops import unary_union
@@ -266,19 +267,21 @@ def get_streets(
# Boundary defined by polygon (perimeter) # Boundary defined by polygon (perimeter)
if perimeter is not None: if perimeter is not None:
# Fetch streets data, project & convert to GDF # Fetch streets data, project & convert to GDF
try:
streets = ox.graph_from_polygon( streets = ox.graph_from_polygon(
unary_union(perimeter.to_crs(3174).buffer(buffer).to_crs(4326).geometry) unary_union(perimeter.geometry).buffer(buffer)
if buffer > 0 if buffer > 0
else unary_union(perimeter.geometry), else unary_union(perimeter.geometry),
retain_all=retain_all,
custom_filter=custom_filter, custom_filter=custom_filter,
) )
streets = ox.project_graph(streets) streets = ox.project_graph(streets)
streets = ox.graph_to_gdfs(streets, nodes=False) streets = ox.graph_to_gdfs(streets, nodes=False)
perimeter = unary_union(ox.project_gdf(perimeter).geometry) except EmptyOverpassResponse:
return MultiLineString()
# Boundary defined by polygon (perimeter) # Boundary defined by polygon (perimeter)
elif (point is not None) and (radius is not None): elif (point is not None) and (radius is not None):
# Fetch streets data, save CRS & project # Fetch streets data, save CRS & project
try:
streets = ox.graph_from_point( streets = ox.graph_from_point(
point, point,
dist=radius + dilate + buffer, dist=radius + dilate + buffer,
@@ -291,10 +294,11 @@ def get_streets(
perimeter = get_boundary(point, radius, crs, circle=circle, dilate=dilate) perimeter = get_boundary(point, radius, crs, circle=circle, dilate=dilate)
# Convert to GDF # Convert to GDF
streets = ox.graph_to_gdfs(streets, nodes=False) streets = ox.graph_to_gdfs(streets, nodes=False)
# Intersect with perimeter & filter empty elements # Intersect with perimeter & filter empty elements
streets.geometry = streets.geometry.intersection(perimeter) streets.geometry = streets.geometry.intersection(perimeter)
streets = streets[~streets.geometry.is_empty] streets = streets[~streets.geometry.is_empty]
except EmptyOverpassResponse:
return MultiLineString()
if type(width) == dict: if type(width) == dict:
streets = unary_union( streets = unary_union(