import os
from os.path import join
from pathlib import Path
import urllib.request
import zipfile


def download_sipakmed(data_dest=None):
    """ Download the SIPaKMeD dataset, http://cs.uoi.gr/~marina
    Args:
      data_dest (str): The path of dir to store data. If noe will default to package cache
    Returns:
      None
    """
    if data_dest is None:
        data_dest = join(get_data_cache(), 'sipakmed_raw')
    if os.path.exists(data_dest):
        raise ValueError('Data already downloaded.')
    os.mkdir(data_dest)

    file_urls = [
        'http://cs.uoi.gr/~marina/SIPAKMED/im_Metaplastic.7z',
        'http://cs.uoi.gr/~marina/SIPAKMED/im_Dyskeratotic.7z'
        'http://cs.uoi.gr/~marina/SIPAKMED/im_Koilocytotic.7z',
        'http://cs.uoi.gr/~marina/SIPAKMED/im_Parabasal.7z',
        'http://cs.uoi.gr/~marina/SIPAKMED/im_Superficial-Intermediate.7z',
    ]
    cell_types = [
        'metaplastic',
        'dyskeratotic',
        'koilocytotic'
        'parabasal',
        'superficial-Intermediate',
    ]

    print(f'Downloading SIPaKMed to: {data_dest}')
    for url, cell_type in zip(file_urls, cell_types):
        print(f'Downloading {cell_type}...')
        file_name = f'{cell_type}.zip'
        urllib.request.urlretrieve(url, join(data_dest, file_name))

    for file_name in data_dest:
        print(f'Extracting {file_name}...')
        with zipfile.ZipFile(file_urls) as f:
            f.extract(file_name, dir)
        os.remove(file_name)

    print('Finished downloading.')
    

def download_smear(data_dest=None):
    """ Download the Hervel smear dataset
    Args:
      data_dest (str): The path of dir to store data. If noe will default to package cache
    Returns:
      None
    """
    raise NotImplementedError('Get it yourself, not implemented.')


def get_data_cache():
    """ Get the cache where datasets are stored. Defaults to ~/.mediaug
    Args:
      None
    Returns:
      path (str): The path to the data cache on the system
    """
    p = Path(join(Path.home(), '.mediaug'))
    p.mkdir(exist_ok=True)
    return str(p)