From 58915ac9684af8fcbed70cfa6d239769e313ccbb Mon Sep 17 00:00:00 2001 From: Toby Lorne Date: Fri, 31 Jul 2020 13:07:07 +0100 Subject: [PATCH] broadcast-areas: plot script works with groups Signed-off-by: Toby Lorne --- notifications_utils/broadcast_areas/plot-areas.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/notifications_utils/broadcast_areas/plot-areas.py b/notifications_utils/broadcast_areas/plot-areas.py index 77db88357..123424995 100755 --- a/notifications_utils/broadcast_areas/plot-areas.py +++ b/notifications_utils/broadcast_areas/plot-areas.py @@ -1,5 +1,6 @@ #!/usr/bin/env python +import geojson import cartopy.crs as ccrs import cartopy.feature as cfeature import matplotlib.pyplot as plt @@ -32,21 +33,21 @@ def main(): all_areas = BroadcastAreasRepository().get_all_areas_for_library(lid) some_areas = sample(all_areas, min(len(all_areas), 25)) for area in some_areas: - print(" ", area[1]) # noqa: T001 + print(" ", area[0], area[1]) # noqa: T001 inp = input("> ") if inp == "": break - aid = make_string_safe_for_id(inp) + aid = inp.strip() area = BroadcastAreasRepository().get_areas([aid])[0] feature = area[-2] - feature_shape = sgeom.shape(feature["geometry"]) + feature_shape = sgeom.shape(geojson.loads(feature)["geometry"]) features.append(feature_shape) simple_feature = area[-1] - simple_feature_shape = sgeom.shape(simple_feature["geometry"]) + simple_feature_shape = sgeom.shape(geojson.loads(simple_feature)["geometry"]) simple_features.append(simple_feature_shape) print() # noqa: T001