#!/usr/bin/env python
# -*- coding: utf-8 -*-

# Import from the standard library
from os.path import split
import argparse
import pandas as pd

# Import from proj
import proj
from proj.lib import clean_data

if __name__ == '__main__':
    usage = '%(prog)s'
    description = 'proj_description'
    # https://docs.python.org/3/library/argparse.html
    # https://docs.python.org/3/howto/argparse.html
    parser = argparse.ArgumentParser(description=description, usage=usage)
    args = parser.parse_args()
    folder_source, _ = split(proj.__file__)
    df = pd.read_csv('{}/data/data.csv.gz'.format(folder_source))
    print('  proj/data/data.csv.gz Loaded')

    res = clean_data(df)
    res.to_csv('out.csv', index=False)
    print('==> out.csv MADE ')
    print('    shape is {}'.format(res.shape))
