WIP: add prefer_local option

Author: keller-mark

src/vitessce/config.py

@@ -242,12 +242,12 @@ def get_artifacts(self):
 
         return artifacts
 
-    def get_stores(self, base_url=None):
+    def get_stores(self, base_url=None, prefer_local=False):
         stores = {}
         for obj in self.objs:
             stores = {
                 **stores,
-                **obj.get_stores(base_url)
+                **obj.get_stores(base_url, prefer_local=prefer_local)
             }
 
         return stores
@@ -1615,7 +1615,7 @@ def get_artifacts(self):
             artifacts.update(d.get_artifacts())
         return artifacts
 
-    def get_stores(self, base_url=None):
+    def get_stores(self, base_url=None, prefer_local=False):
         """
         Convert the routes for this view config from the datasets.
 
@@ -1626,7 +1626,7 @@ def get_stores(self, base_url=None):
         for d in self.config["datasets"]:
             stores = {
                 **stores,
-                **d.get_stores(base_url)
+                **d.get_stores(base_url, prefer_local=prefer_local)
             }
         return stores
 

src/vitessce/widget.py

@@ -472,7 +472,7 @@ class VitessceWidget(anywidget.AnyWidget):
 
     store_urls = List(trait=Unicode(''), default_value=[]).tag(sync=True)
 
-    def __init__(self, config, height=600, theme='auto', uid=None, port=None, proxy=False, js_package_version='3.5.4', js_dev_mode=False, custom_js_url='', plugins=None, remount_on_uid_change=True, invoke_timeout=30000):
+    def __init__(self, config, height=600, theme='auto', uid=None, port=None, proxy=False, js_package_version='3.5.4', js_dev_mode=False, custom_js_url='', plugins=None, remount_on_uid_change=True, prefer_local=True, invoke_timeout=30000):
         """
         Construct a new Vitessce widget.
 
@@ -506,7 +506,7 @@ def __init__(self, config, height=600, theme='auto', uid=None, port=None, proxy=
         config_dict = config.to_dict(base_url=base_url)
         routes = config.get_routes()
 
-        self._stores = config.get_stores(base_url=base_url)
+        self._stores = config.get_stores(base_url=base_url, prefer_local=prefer_local)
         self._plugins = plugins or []
 
         plugin_esm = [p.plugin_esm for p in self._plugins]

src/vitessce/wrappers.py

@@ -126,8 +126,17 @@ def get_artifacts(self):
         :rtype: dict[str, lamindb.Artifact]
         """
         return self.artifacts
-
-    def get_stores(self, base_url):
+    
+    def try_getting_artifact_stores(self):
+        artifact_stores = dict()
+        for artifact_url, artifact in self.artifacts.items():
+            local_path = artifact._local_filepath
+            if os.path.isdir(local_path):
+                store = zarr.DirectoryStore(local_path)
+                artifact_stores[artifact_url] = store
+        return artifact_stores
+
+    def get_stores(self, base_url, prefer_local=False):
         """
         Obtain the stores that have been created for this wrapper class.
 
@@ -139,6 +148,8 @@ def get_stores(self, base_url):
         for relative_url, store in relative_stores.items():
             absolute_url = base_url + relative_url
             absolute_stores[absolute_url] = store
+        if prefer_local:
+            absolute_stores.update(self.try_getting_artifact_stores())
         return absolute_stores
 
     def get_file_defs(self, base_url):