main.py

import argparse
from bank_parser.bank_of_america import parse_bank_of_america
from bank_parser.capital_one_bank import parse_capital_one_bank
from bank_parser.CITIBANK import parse_CITIBANK
from bank_parser.Redacted_Bank import parse_redacted_bank
from bank_parser.PNC_Bank import parse_pnc_bank

def main():
    # Set up the argument parser
    parser = argparse.ArgumentParser(description='Process bank statements.')

    # Add an argument for the bank name
    parser.add_argument('bank', nargs='?', type=str, help='Name of the bank to process')

    # Parse the command-line arguments
    args = parser.parse_args()

    # Now, you can access the bank name using args.bank
    bank = args.bank
    
    if bank == "Bank of America" or bank is None:
        parse_bank_of_america()
    if bank == "Capital One Bank" or bank is None:
        parse_capital_one_bank()
    if bank == "CITIBANK" or bank is None:
        parse_CITIBANK()
    if bank == "Redacted_Bank" or bank is None:
        parse_redacted_bank()
    if bank == "PNC_Bank" or bank is None:
        parse_pnc_bank()
    if bank == "PNC Bank" or bank is None:
        pass


if __name__ == "__main__":
    main()