Add ability to create custom layers

emfcamp · May 4, 2024 · 0b01b82 · 0b01b82
1 parent 0ad959f
commit 0b01b82
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/buildmap/exporter/tegola.py b/buildmap/exporter/tegola.py
@@ -112,7 +112,7 @@ def generate_tegola_config(self):
         }
 
         if (
-            type(self.config["mapbox_vector_layer"]) is dict
+            isinstance(self.config["mapbox_vector_layer"], dict)
             and "attribution" in self.config["mapbox_vector_layer"]
         ):
             m["attribution"] = self.config["mapbox_vector_layer"]["attribution"]
@@ -150,6 +150,22 @@ def generate_tegola_config(self):
             }
         )
 
+        for layer_name, data in self.config.get("custom_layers", {}).items():
+            provider["layers"].append(
+                {
+                    "name": layer_name,
+                    "sql": data["query"],
+                    "geometry_type": data["geometry_type"],
+                }
+            )
+            m["layers"].append(
+                {
+                    "provider_layer": "%s.%s" % (self.PROVIDER_NAME, layer_name),
+                    "min_zoom": self.config["zoom_range"][0],
+                    "max_zoom": self.config["zoom_range"][1],
+                }
+            )
+
         # Construct config
         data = {
             "cache": {"type": "file", "basepath": "/tmp/tegola"},