"""Abacus DB Connector."""

import os
from  mysql import connector
from connectors.base import Base

class AbacusMysqlConnection(Base):
    def __init__(self, environment):
        AbacusMysqlConnection.db_name = 'Abacus'
        if AbacusMysqlConnection.connection is None:
            params = {
                'user': os.environ.get(f'ABACUS_MYSQL_{environment.upper()}_USER', None),
                'password': os.environ.get(f'ABACUS_MYSQL_{environment.upper()}_PASSWORD', None),
                'host': os.environ.get(f'ABACUS_MYSQL_{environment.upper()}_HOST', None),
                'port': int(os.environ.get(f'ABACUS_MYSQL_{environment.upper()}_PORT', 3306)),
                'db': os.environ.get(f'ABACUS_MYSQL_{environment.upper()}_DATABASE', None)
            }
            try:
                AbacusMysqlConnection.connection = connector.connect(**params)
                AbacusMysqlConnection.cursor = AbacusMysqlConnection.connection.cursor(dictionary=True)
            except Exception as error:
                print("Error: Abacus DB Connection not established {}".format(error))
            else:
                print("Abacus DB Connection established")