From 5e3a69d1f3366f0272333542afc8b963dd5cf871 Mon Sep 17 00:00:00 2001
From: Colm McDonald <55803987+G21-Goose@users.noreply.github.com>
Date: Sat, 28 Aug 2021 23:41:42 +0100
Subject: [PATCH] Add retain_all option to get_streets()

Add retain_all option so that separate graphs can be kept instead of only keeping the largest.
---
 prettymaps/fetch.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/prettymaps/fetch.py b/prettymaps/fetch.py
index fd092f5..6e0fa6c 100644
--- a/prettymaps/fetch.py
+++ b/prettymaps/fetch.py
@@ -80,7 +80,7 @@ def get_geometries(perimeter = None, point = None, radius = None, tags = {}, per
     return geometries
 
 # Get streets
-def get_streets(perimeter = None, point = None, radius = None, layer = 'streets', width = 6, custom_filter = None, buffer = 0,circle = True, dilate = 0):
+def get_streets(perimeter = None, point = None, radius = None, layer = 'streets', width = 6, custom_filter = None, buffer = 0, retain_all = False, circle = True, dilate = 0):
 
     if layer == 'streets':
         layer = 'highway'
@@ -94,7 +94,7 @@ def get_streets(perimeter = None, point = None, radius = None, layer = 'streets'
     # Boundary defined by polygon (perimeter)
     elif (point is not None) and (radius is not None):
         # Fetch streets data, save CRS & project
-        streets = ox.graph_from_point(point, dist = radius+dilate+buffer, custom_filter = custom_filter)
+        streets = ox.graph_from_point(point, dist = radius+dilate+buffer, retain_all = retain_all, custom_filter = custom_filter)
         crs = ox.graph_to_gdfs(streets, nodes = False).crs
         streets = ox.project_graph(streets)
         # Compute perimeter from point & CRS