import os

from src.constants import PlaylistDigestMetric


ENVIRONMENT = os.environ.get("ENVIRONMENT", "prod").lower()
SENTRY_DSN = os.environ.get("SENTRY_DSN", "testing")


class Snowflake:
    USER = os.environ["SNOWFLAKE_USER"]
    PASSWORD = os.environ["SNOWFLAKE_PASSWORD"]
    ACCOUNT = os.environ["SNOWFLAKE_ACCOUNT"]
    WAREHOUSE = os.environ["SNOWFLAKE_WAREHOUSE"]
    REGION = os.environ.get("SNOWFLAKE_REGION", "us-east-1")
    ROLE = os.environ.get("SNOWFLAKE_ROLE", "APOLLO_TEAM_ROLE")
    INTEGRATION = os.environ["SNOWFLAKE_INTEGRATION"]
    PUBLIC_DATA_MAIN_DATABASE = os.environ["SNOWFLAKE_PUBLIC_DATA_MAIN_DATABASE"]
    PUBLIC_DATA_MAIN_PLAYLISTS_SCHEMA = os.environ.get("SNOWFLAKE_PUBLIC_DATA_MAIN_PLAYLISTS_SCHEMA", "PLAYLISTS")
    EXPLORATION_DATABASE = os.environ["SNOWFLAKE_EXPLORATION_DATABASE"]
    EXPLORATION_MAIN_SCHEMA = os.environ.get("SNOWFLAKE_EXPLORATION_MAIN_SCHEMA", "MAIN")
    EXPLORATION_EXP_SCHEMA = os.environ.get("SNOWFLAKE_EXPLORATION_MAIN_SCHEMA", "EXP")
    AUTHENTICATOR = os.environ.get("SNOWFLAKE_AUTHENTICATOR", "snowflake")


class MySQL:
    HOST = os.environ["MYSQL_HOST"]
    PORT = int(os.environ.get("MYSQL_PORT", 3306))
    USER = os.environ["MYSQL_USER"]
    PASSWORD = os.environ["MYSQL_PASSWORD"]
    DATABASE = os.environ["MYSQL_DATABASE"]
    SLEEP = int(os.environ.get("MYSQL_SLEEP", 3))


S3_BUCKET = os.environ["S3_BUCKET"]
AWS_DEFAULT_REGION = os.environ.get("AWS_DEFAULT_REGION", "us-east-1")

OVERRIDE_PREVIOUS_DATE = bool(os.environ.get("OVERRIDE_PREVIOUS_DATE", ""))
MAX_DAYS_COUNT = {
    PlaylistDigestMetric.STREAMS: int(os.environ.get("STREAMS_MAX_DAYS_COUNT", 7)),
    PlaylistDigestMetric.FOLLOWERS: int(os.environ.get("FOLLOWERS_MAX_DAYS_COUNT", 30)),
}

METRIC_MIN_VALUE = {
    PlaylistDigestMetric.STREAMS: int(os.environ.get("STREAMS_MIN_VALUE", 100)),
    PlaylistDigestMetric.FOLLOWERS: int(os.environ.get("FOLLOWERS_MIN_VALUE", 100)),
}