"""Ingest Adjustments."""

import argparse
import time
from dotenv import load_dotenv

from adjustment.adjustment import Adjustment
from connectors.abacus_db import AbacusMysqlConnection
from connectors.art_relation_db import ARMysqlConnection

load_dotenv()

if __name__ == '__main__':
    # args "python ingest_adjustments.py -p 286 -e qa -f files/ACC-5288/2022Q3/MCPS/Adjustment/valid_test_file.xlsx"
    parser = argparse.ArgumentParser()
    parser.add_argument('-p', '--p', help='Statement Period')
    parser.add_argument('-e', '--e', help='qa or prod', choices={'qa', 'prod'})
    parser.add_argument('-f', '--f', help='File path')
    args = vars(parser.parse_args())

    start_time = time.time()
    accounting_period = args.get('p', None)
    file_path = args.get('f', None)
    environment = args.get('e', None)

    try:
        if accounting_period and file_path:
            AbacusMysqlConnection(environment)
            ARMysqlConnection(environment)

            adjustment = Adjustment(accounting_period, file_path)
            adjustment.process()
        else:
            print('Command line arguments are missing...')
    finally:
        AbacusMysqlConnection.connection_close()
        ARMysqlConnection.connection_close()

    print("--- %s seconds ---" % (time.time() - start_time))