import psycopg2
import sqlalchemy as sa

DB_NAME = "kroton_analytics_prod"
USERNAME = "last_mile"
HOST = "kroton-analytics-datalake-prod.cnzkefry7wsx.us-east-1.redshift.amazonaws.com"
PASSWORD = "*******"
PORT = "******"


class DatasetConnection:
    @staticmethod
    def get_connector():
        return psycopg2.connect("""
            dbname='{db_name}'
            user='{username}'
            host='{host}'
            port={port}
            password='{password}'""".format(
                db_name=DB_NAME,
                username=USERNAME,
                host=HOST,
                port=PORT,
                password=PASSWORD))

    def get_engine():
        return sa.create_engine(
            'redshift+psycopg2://last_mile:##(Kroton2018)!@kroton-analytics-datalake-prod.cnzkefry7wsx.us-east-1.redshift.amazonaws.com:5439/kroton_analytics_prod',
            pool_pre_ping=True)