def run(sql, warehouse_name=None): """ Runs sql queries against warehouse_name defined in ~/.whale/config/connections.yaml. If no warehouse_name is given, the first is used. """ connection_dict = get_connection(warehouse_name=warehouse_name) connection = ConnectionConfigSchema(**connection_dict) engine, conf = configure_unscoped_sqlalchemy_engine(connection) engine.init(conf) result = engine.execute(sql, has_header=True) headers = next(result) table = list(result) df = pd.DataFrame(table, columns=headers) return df
def run(sql, warehouse_name=None) -> pd.DataFrame: """ Runs sql queries against warehouse_name defined in ~/.whale/config/connections.yaml. If no warehouse_name is given, the first is used. """ connection = get_connection(warehouse_name=warehouse_name) engine, conf = configure_unscoped_sqlalchemy_engine(connection) engine.init(conf) sql = template_query(sql, connection_name=connection.name) LOGGER.info(f"Templated query:\n{sql}") result = engine.execute(sql, has_header=True) headers = next(result) table = list(result) df = pd.DataFrame(table, columns=headers) return df