import os
from kaggle.api.kaggle_api_extended import KaggleApi
import sys


# Initialize
api = KaggleApi()
api.authenticate()

# Download dataset to current dir
def kaggleDataset(url,query):
    try:
        print(f"{url} started downloading")
        os.makedirs("./downloads/"+query, exist_ok=True)
        api.dataset_download_files(url, path='./downloads/'+query, unzip=True)
    except Exception as e:
        print("Dataset not found")

# kaggleDataset('zynicide/wine-reviews')