Add download url function

Author: tmorrell

caltechdata_api/__init__.py

@@ -3,4 +3,4 @@
 from .customize_schema import customize_schema
 from .decustomize_schema import decustomize_schema
 from .get_metadata import get_metadata
-from .download_file import download_file
+from .download_file import download_file, download_url

caltechdata_api/download_file.py

@@ -2,8 +2,8 @@
 from tqdm import tnrange, tqdm_notebook
 
 
-def download_file(doi, fname=None, media_type=None):
-    """Download a file  listed in the media API for a DataCite DOI"""
+def download_url(doi, media_type=None):
+    """Get a download link for a file listed in the media API for a DataCite DOI"""
     api_url = "https://api.datacite.org/dois/" + doi + "/media"
     r = requests.get(api_url).json()
     data = r["data"]
@@ -13,6 +13,12 @@ def download_file(doi, fname=None, media_type=None):
         for media in data:
             if media["attributes"]["mediaType"] == media_type:
                 url = media["attributes"]
+    return url
+
+
+def download_file(doi, fname=None, media_type=None):
+    """Download a file  listed in the media API for a DataCite DOI"""
+    url = download_url(doi, media_type)
     r = requests.get(url, stream=True)
     # Set file name
     if fname == None:

codemeta.json

@@ -6,7 +6,7 @@
   "codeRepository": "https://github.com/caltechlibrary/caltechdata_api",
   "issueTracker": "https://github.com/caltechlibrary/caltechdata_api/issues",
   "license": "https://data.caltech.edu/license",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "author": [
     {
       "@type": "Person",

setup.py

@@ -51,7 +51,7 @@ def read(fname):
 
 # What packages are required for this module to be executed?
 REQUIRED = [
-    'requests','datacite','tqdm'
+    'requests','datacite>1.1.0','tqdm'
 ]
 
 # What packages are optional?

tests/test_download.py

@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+#
+# This file is part of caltechdata_api.
+#
+# Copyright (C) 2020 Caltech.
+#
+# caltechdata_api is free software; you can redistribute it and/or modify it
+# under the terms of the Revised BSD License; see LICENSE file for
+# more details.
+
+"""Tests for format transformations."""
+
+import pytest
+
+from caltechdata_api import download_url, download_file
+
+def test_download():
+    """Test that downloads from the DataCite Media API work."""
+    example_doi = '10.22002/D1.1945'
+    expected_url = 'https://data.caltech.edu/tindfiles/serve/0cedc067-1ff0-4fc9-871a-2869f3a4957d/'
+    assert expected_url == download_url(example_doi)