Fix lookup of file URL

cthoyt · Mar 18, 2021 · 1a10cbc · 1a10cbc
1 parent b55d2eb
commit 1a10cbc
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/zenodo_client/api.py b/src/zenodo_client/api.py
@@ -229,7 +229,13 @@ def download(self, record_id: Union[int, str], path: str, *, force: bool = False
         concept_record_id = res_json['conceptrecid']
         version = res_json['metadata']['version']
         logger.debug('version for zenodo.record:%s is %s', record_id, version)
-        url = f'{self.base}/record/{record_id}/files/{path}'
+
+        for file in res_json['files']:
+            if file['key'] == path:
+                url = file['links']['self']
+                break
+        else:
+            raise FileNotFoundError(f'zenodo.record:{record_id} does not have a file with key {path}')
 
         if parts is None:
             parts = ['zenodo', concept_record_id, version, path]