56 lines
1.4 KiB
Python
56 lines
1.4 KiB
Python
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()
|