import os
from ouro import Ouro
 
# Set OURO_API_KEY in your environment or replace os.environ.get("OURO_API_KEY")
ouro = Ouro(api_key=os.environ.get("OURO_API_KEY"))
 
dataset_id = "019fbf58-45c3-7b70-9394-2420e70a831c"
 
# Retrieve dataset metadata
dataset = ouro.datasets.retrieve(dataset_id)
print(dataset.name, dataset.visibility)
print(dataset.metadata)

# Get column definitions for the underlying table
columns = ouro.datasets.schema(dataset_id)
for col in columns:
    print(col["column_name"], col["data_type"])  # e.g., age integer, name text

# Option 1: All rows as a Pandas DataFrame
df = ouro.datasets.query(dataset_id)
print(df.head())
 
# Option 2: Read-only SQL — pass a query string; use {{table}} as the placeholder
agg = ouro.datasets.query(
    dataset_id,
    "SELECT col, count(*) AS n FROM {{table}} GROUP BY col ORDER BY n DESC",
)

import pandas as pd
 
# Update dataset metadata
updated = ouro.datasets.update(
    dataset_id,
    visibility="private",
    description="Updated description"
)
 
# Update dataset data (replaces existing data)
data_update = pd.DataFrame([
    {"name": "Charlie", "age": 33},
    {"name": "Diana", "age": 28},
])
updated = ouro.datasets.update(dataset_id, data=data_update)