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