Support read_vineyard_dataframe with ObjectID. (#149)

Signed-off-by: Tao He <linzhu.ht@alibaba-inc.com>
v6d-io · Jan 19, 2021 · 0d959c5 · 0d959c5
1 parent a7b0727
commit 0d959c5
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/modules/io/adaptors/read_vineyard_dataframe.py b/modules/io/adaptors/read_vineyard_dataframe.py
@@ -44,7 +44,11 @@ def read_vineyard_dataframe(
     print(json.dumps(ret), flush=True)
 
     name = urlparse(path).netloc
-    df_id = client.get_name(name)
+    # the "name" part in URL can be a name, or an ObjectID for convenience.
+    try:
+        df_id = client.get_name(name)
+    except:
+        df_id = vineyard.ObjectID(name)
     dataframes = client.get(df_id)
 
     writer = stream.open_writer(client)