setup

#!/usr/bin/env python3

import os
import argparse

DATA_DIR = os.path.join('research', 'data')

parser = argparse.ArgumentParser(
    description='Setup serenata-de-amor using serenata-toolbox.')

parser.add_argument('--all', dest='all', action='store_true',
                    help='Force all datasets from serenata-toolbox \
                    to be downloaded')

args = parser.parse_args()

print('Downloading datasets (this might take several minutes \
    depending on your internet connection)')

from serenata_toolbox.datasets import fetch_latest_backup
os.makedirs(DATA_DIR, exist_ok=True)
fetch_latest_backup(DATA_DIR, force_all=args.all)