fix api root

src/c3nav/api/__init__.py

@@ -11,7 +11,9 @@ orig_render = JSONRenderer.render
 def nicer_renderer(self, data, accepted_media_type=None, renderer_context=None):
     if self.get_indent(accepted_media_type, renderer_context) is None:
         return orig_render(self, data, accepted_media_type, renderer_context)
-    shorten_limit = 5 if any(('geometry' in item) for item in data[:50]) else 50
+    shorten_limit = 50
+    if isinstance(data, (list, tuple)):
+        shorten_limit = 5 if any(('geometry' in item) for item in data[:50]) else 50
     shorten = isinstance(data, (list, tuple)) and len(data) > shorten_limit
     orig_len = None
     if shorten:

src/c3nav/mapdata/render/svg.py

@@ -0,0 +1,126 @@
+import xml.etree.ElementTree as ET
+from abc import ABC
+
+from shapely.affinity import scale
+
+from c3nav.mapdata.utils.misc import get_render_dimensions
+
+
+class SVGGroup:
+    def __init__(self, width: int, height: int, scale: float=1):
+        self.width = width
+        self.height = height
+        self.scale = scale
+        self.g = ET.Element('g', {
+            'transform': 'scale(1 -1) translate(0 -%d)' % (self.height),
+        })
+
+    def get_xml(self):
+        return self.g
+
+    def add_group(self):
+        group = SVGGroup(self.width, self.height, self.scale)
+        self.g.append(group)
+        return group
+
+
+class SVGImage(SVGGroup):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    def get_xml(self):
+        root = ET.Element('svg', {
+            'width': str(self.width),
+            'height': str(self.height),
+            'xmlns:svg': 'http://www.w3.org/2000/svg',
+            'xmlns': 'http://www.w3.org/2000/svg',
+            'xmlns:xlink': 'http://www.w3.org/1999/xlink',
+        })
+        root.append(self.g)
+        return root
+
+    def add_geometry(self, geometry, fill_color=None, fill_opacity=None,
+                    stroke_width=0.0, stroke_color=None, stroke_opacity=None):
+        scaled = scale(geometry, xfact=self.scale, yfact=self.scale, origin=(0, 0))
+        element = ET.fromstring(scaled.svg(0, fill_color or '#FFFFFF'))
+        if element.tag != 'g':
+            new_element = ET.Element('g')
+            new_element.append(element)
+            element = new_element
+
+        paths = element.findall('polyline')
+        if len(paths) == 0:
+            paths = element.findall('path')
+
+        for path in paths:
+            path.attrib.pop('opacity')
+            path.set('stroke-width', str(stroke_width * self.scale))
+
+            if fill_color is None and 'fill' in path.attrib:
+                path.attrib.pop('fill')
+                path.set('fill-opacity', '0')
+
+            if fill_opacity is not None:
+                path.set('fill-opacity', str(fill_opacity))
+
+            if stroke_color is not None:
+                path.set('stroke', stroke_color)
+            elif 'stroke' in path.attrib:
+                path.attrib.pop('stroke')
+
+            if stroke_opacity is not None:
+                path.set('stroke-opacity', str(stroke_opacity))
+
+        self.g.append(element)
+        return element
+
+
+class MapRenderer(ABC):
+    def __init__(self):
+        pass
+
+    def create_svg(self):
+        width, height = get_render_dimensions()
+        svg = ET.Element('svg', {
+            'width': str(width),
+            'height': str(height),
+            'xmlns:svg': 'http://www.w3.org/2000/svg',
+            'xmlns': 'http://www.w3.org/2000/svg',
+            'xmlns:xlink': 'http://www.w3.org/1999/xlink',
+        })
+        return svg
+
+    @staticmethod
+    def polygon_svg(geometry, fill_color=None, fill_opacity=None,
+                    stroke_width=0.0, stroke_color=None, stroke_opacity=None):
+        scaled = scale(geometry, xfact=settings.RENDER_SCALE, yfact=settings.RENDER_SCALE, origin=(0, 0))
+        element = ET.fromstring(scaled.svg(0, fill_color or '#FFFFFF'))
+        if element.tag != 'g':
+            new_element = ET.Element('g')
+            new_element.append(element)
+            element = new_element
+
+        paths = element.findall('polyline')
+        if len(paths) == 0:
+            paths = element.findall('path')
+
+        for path in paths:
+            path.attrib.pop('opacity')
+            path.set('stroke-width', str(stroke_width * settings.RENDER_SCALE))
+
+            if fill_color is None and 'fill' in path.attrib:
+                path.attrib.pop('fill')
+                path.set('fill-opacity', '0')
+
+            if fill_opacity is not None:
+                path.set('fill-opacity', str(fill_opacity))
+
+            if stroke_color is not None:
+                path.set('stroke', stroke_color)
+            elif 'stroke' in path.attrib:
+                path.attrib.pop('stroke')
+
+            if stroke_opacity is not None:
+                path.set('stroke-opacity', str(stroke_opacity))
+
+        return element