"""Art-Relation DB Connector."""

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

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