"""Snowflake Client for ows-accounting testing.."""
import snowflake.connector
from snowflake.connector.cursor import DictCursor


class SnowflakeClient:
    """Snowflake Client for ows-accounting testing."""

    def __init__(
            self, account, host, role,
            warehouse, database, schema, user, password):
        """Initialize SnowflakeClient class."""
        self.account = account
        self.host = host
        self.role = role
        self.warehouse = warehouse
        self.database = database
        self.schema = schema
        self.user = user
        self.password = password

    def execute_query(self, query_string):
        """Execute a query, return results."""
        con = snowflake.connector.connect(
            account=self.account,
            host=self.host,
            role=self.role,
            warehouse=self.warehouse,
            database=self.database,
            schema=self.schema,
            user=self.user,
            password=self.password
        )

        cursor = con.cursor(DictCursor)
        cursor.execute(query_string)
        data = cursor.fetchall()
        cursor.close()
        con.close()

        return data