from banking_breakdown import document_builder
from banking_breakdown import statement_parser
from banking_breakdown import ui

import argparse


def categorize_func(args):
    from banking_breakdown.statement_parser import get_stripped_statement

    df = None
    if args.i is not None:
        df = get_stripped_statement(args.i)

    ui.show_main_window(df=df)


def report_func(args):
    print("Report")


#
# Define CLI
#


def main():
    parser = argparse.ArgumentParser(prog='banking_breakdown',
                                     description='Visualize bank statements')

    subparsers = parser.add_subparsers()

    categorize_parser = subparsers.add_parser("categorize")
    categorize_parser.set_defaults(func=categorize_func)
    categorize_parser.add_argument('-i', required=False,
                                   help="Bank statement CSV")
    categorize_parser.add_argument('-f', required=False,
                                   help="JSON file containing regexes to"
                                        " pre-categorize statement entries")

    report_parser = subparsers.add_parser("report")
    report_parser.set_defaults(func=report_func)
    report_parser.add_argument('-i', required=True,
                               help="CSV output file from categorization step")

    args = parser.parse_args()

    if hasattr(args, 'func'):
        args.func(args)
    else:
        parser.print_help()


if __name__ == "__main__":
    main()