import os
from ..utils import ZenodoException, check_response, download_files, get_host_and_id
from .downloader import Downloader
[docs]
class ZenodoDownloader(Downloader):
command = "zenodo"
command_help = "Download study from Zenodo"
[docs]
def download(self, args):
main(args)
[docs]
def populate_arg_parser(self, parser):
parser.description = self.command_help
parser.add_argument("identifier", help="URL or DOI of the study to download")
parser.add_argument(
"destination",
nargs="?",
default="",
help="Path to the destination folder. If not specified, defaults to the study numeric ID.",
)
parser.add_argument(
"--only-metadata",
help="Only download metadata (overrides destination directory's metadata!)",
action="store_true",
)
[docs]
def main(args):
"""Download content from Zenodo"""
import requests
from ..scripts.init import create_solidipes_directory
from ..utils import set_study_metadata
try:
host, study_id = get_host_and_id(args.identifier)
url = f"https://{host}/api/records/{study_id}"
# Scan record
response = requests.get(url)
check_response(response, 200, "retrieve record")
record = response.json()
print(f"Retrieving study {study_id} from {host}...")
# Create destination folder if it does not exist
if not args.destination:
args.destination = study_id
if not os.path.exists(args.destination):
os.makedirs(args.destination)
# Create Solidipes directory if it does not exist
try:
create_solidipes_directory(args.destination)
except FileExistsError:
pass
# Save metadata in YAML file
print("Saving metadata...")
metadata = process_metadata(record["metadata"])
set_study_metadata(metadata, initial_path=args.destination)
if args.only_metadata:
return
download_files(record, destination=args.destination, progressbar=True)
except Exception as e:
if type(e) is not ZenodoException:
raise e
print(e)
return