import os

ENV = os.getenv('ENVIRONMENT')
if ENV is None:
    ENV = 'dev'

decompressed_bucket = f"{ENV}-sme-data-decompressed"
archive_bucket = f"{ENV}-sme-data-archive"
corrupt_bucket = f"{ENV}-delphi-sme-data-corrupted"
body_read_length = 5

aws = {
    "RDS": {
        "POSTGRESQL_PORT": "5432",
        "DB_HOST":
        f"{ENV}-delphi-slz-master.cic9zqsxd2y6.us-east-1.rds.amazonaws.com",
        "DB_NAME": "profile",
        'USERNAME': 'chartmetric_user',
        "queries": {
            "INSERT_ARTIST_QUERY":
            "INSERT INTO chartmetric_file(filename,data_quality,data_movement,data_write_big_table,report_date,update_date,start_time) values('{}','{}','{}','{}','{}',CURRENT_DATE,'{}') ON CONFLICT(filename)"
            "DO UPDATE SET update_date = CURRENT_DATE ;",
            'GET_LATEST_RUN_COUNT':
            "SELECT max(schedule) from chartmetric_performance where run_date = CURRENT_DATE and process_type = '{1}';",
            'INSERT_EXECUTION_TIME_METRICS':
            "INSERT INTO chartmetric_performance (process_type,schedule,start_time,end_time,total_time,run_date,parent_id) "
            "VALUES('{0}',{1},'{2}','{3}',age('{4}','{5}'),'{6}',{7});"
        },
    }
}

if ENV == 'stage':
    aws['RDS'][
        'DB_HOST'] = 'stage-delphi-slz.cluster-cic9zqsxd2y6.us-east-1.rds.amazonaws.com'

elif ENV == 'prod':
    aws['RDS'][
        'DB_HOST'] = 'prod-delphi-slz.cluster-cjzbf0a3lwld.us-east-1.rds.amazonaws.com'
    decompressed_bucket = 'sme-data-decompressed'
    archive_bucket = f"sme-data-archive"