add a route_geojson function

mapbox/services/directions.py

@@ -26,6 +26,33 @@ def route(self,
         waypoints = encode_waypoints(features, precision=6,
                                      min_limit=2, max_limit=30)
 
+        # TODO add optional args as url params
         uri = URITemplate('%s/{profile}/{waypoints}.json' % self.baseuri).expand(
             profile=self.profile, waypoints=waypoints)
         return self.session.get(uri)
+
+    def route_geojson(self, *args, **kwargs):
+        res = self.route(*args, **kwargs)
+        data = res.json()
+        fc = {
+            'type': 'FeatureCollection',
+            'features': []}
+
+        for route in data['routes']:
+
+            feature = {
+                'properties': {
+                    # TODO handle these nested structures
+                    # Flatten or ???
+                    # 'destination': data['destination'],
+                    # 'origin': data['origin'],
+                    # 'waypoints': data['waypoints'],
+                    # 'steps': route['steps']
+                    'distance': route['distance'],
+                    'duration': route['duration'],
+                    'summary': route['summary']}}
+
+            feature['geometry'] = route['geometry']
+            fc['features'].append(feature)
+
+        return fc

tests/test_directions.py

@@ -21,8 +21,6 @@
 
 @responses.activate
 def test_directions():
-    """Forward geocoding works"""
-
     with open('tests/moors.json') as fh:
         body = fh.read()
 
@@ -39,4 +37,21 @@ def test_directions():
     assert sorted(res.json().keys()) == ['destination', 'origin', 'routes', 'waypoints']
 
 
+@responses.activate
+def test_directions_geojson():
+    with open('tests/moors.json') as fh:
+        body = fh.read()
+
+    responses.add(
+        responses.GET,
+        'https://api.mapbox.com/v4/directions/mapbox.driving/-87.337875%2C36.539157%3B-88.247681%2C36.922175.json?access_token=pk.test',
+        match_querystring=True,
+        body=body, status=200,
+        content_type='application/json')
+
+    fc = mapbox.Directions(access_token='pk.test').route_geojson(points)
+    assert fc['type'] == 'FeatureCollection'
+    assert sorted(fc['features'][0]['properties'].keys()) == ['distance', 'duration', 'summary']
+    assert fc['features'][0]['geometry']['type'] == "LineString"
+
 # TODO test as_featurecollection=True