#!/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 edgar_toolbox
import edgar_toolbox
from edgar_toolbox.lib import clean_data

if __name__ == "__main__":
    usage = "%(prog)s"
    description = "edgar_toolbox_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(edgar_toolbox.__file__)
    df = pd.read_csv("{}/data/data.csv.gz".format(folder_source))
    print("  edgar_toolbox/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))
