fixed usage of dag-pb codec

Since go-ipfs v0.10.0, data in dag-pb is encoded differently when retrieved as JSON. This change accepts both variants (old and new).
fsspec · Oct 7, 2021 · 7b592e8 · 7b592e8
1 parent 7f53a46
commit 7b592e8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/ipfsspec/core.py b/ipfsspec/core.py
@@ -231,7 +231,11 @@ def req(endpoint):
         else:
             dag = req("dag/get")
             data = UnixFSData()
-            data.ParseFromString(base64.b64decode(dag["data"]))
+            if "data" in dag:
+                data.ParseFromString(base64.b64decode(dag["data"]))
+            else:
+                rawdata = dag["Data"]["/"]["bytes"]
+                data.ParseFromString(base64.b64decode(rawdata + "=" * (-len(rawdata) % 4)))
 
             size = data.filesize
             if data.Type == data.File: