def main():
arg_parser = argparse.ArgumentParser()
arg_parser.add_argument('-w', '--workbook')
args = arg_parser.parse_args()
statements = glob.glob(DOCUMENTS_PATH)
sp = StatementProcessor()
all_transactions = []
for file in statements:
raw_data = parser.from_file(file)
raw_data = raw_data['content']
statement_type = sp.determine_statement_type(raw_data)
if statement_type == 'savings':
transactions = sp.extract_transactions_savings(raw_data)
all_transactions.extend(transactions)
elif statement_type == 'mastercard':
transactions = sp.extract_transactions_mastercard(raw_data)
all_transactions.extend(transactions)
elif statement_type == 'checking':
transactions = sp.extract_transactions_checking(raw_data)
all_transactions.extend(transactions)
ew = ExcelWriter(all_transactions, args.workbook)
ew.write_to_file()
print('done :)')
