from enum import StrEnum, unique


@unique
class PayloadKeys(StrEnum):
    """Payload keys constants."""

    SQL = "sql"
    COMPRESSION = "compression"
    OUTPUT_FORMAT = "output_format"
    STAGED = "staged"


@unique
class OutputFormat(StrEnum):
    """Output format constants."""

    MSGPACK = "msgpack"
    JSON = "json"
    CSV = "csv"
    DF = "df"


@unique
class SFClientSettings(StrEnum):
    """Snowflake client settings constants."""

    USER = "user"
    ACCOUNT = "account"
    WAREHOUSE = "warehouse"
    DATABASE = "database"
    SCHEMA = "schema"
    PRIVATE_KEY = "private_key"