surround block with driver.connect() to keep connection opened

DHI · Aug 9, 2023 · 2cda97e · 2cda97e
1 parent 354a915
commit 2cda97e
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/terracotta/handlers/metadata.py b/terracotta/handlers/metadata.py
@@ -46,9 +46,10 @@ def multiple_metadata(
     key_names = driver.key_names
 
     out = []
-    for dataset in datasets:
-        metadata = filter_metadata(driver.get_metadata(dataset), columns)
-        metadata["keys"] = OrderedDict(zip(key_names, dataset))
-        out.append(metadata)
+    with driver.connect():
+        for dataset in datasets:
+            metadata = filter_metadata(driver.get_metadata(dataset), columns)
+            metadata["keys"] = OrderedDict(zip(key_names, dataset))
+            out.append(metadata)
 
     return out