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"))
 
# Option 1: Retrieve by route ID
route_id = "3fa25076-f993-4a57-90dd-799f4d2b6e1b"
route = ouro.routes.retrieve(route_id)
 
# Option 2: Retrieve by route identifier (username/route-name)
route_identifier = "mmoderwell/density-of-states"
route = ouro.routes.retrieve(route_identifier)
 
print(route.name, route.visibility)
print(route.metadata)

# Retrieve the route
route = ouro.routes.retrieve("mmoderwell/density-of-states")
 
# Execute the route
action = route.execute(
    body={
        'nspin': 'auto',
        'ecutwfc': 50,
        'scf_thr': 0.0001,
        'kspacing': 0.3,
        'scf_nmax': 120,
        'basis_size': 'DZP',
        'dft_functional': 'PBE',
        'smearing_sigma': 0.05,
        'smearing_method': 'gauss'
    },
    assets={
        'file': 'your-file-id'
    },
)
print(action.final_data)

# Retrieve the route
route = ouro.routes.retrieve("mmoderwell/density-of-states")
 
# Read all actions (request/response history) for this route
actions = route.read_actions()
print(actions)
 
# Actions are especially useful for long-running routes
# You can poll the status and retrieve the response when ready
for action in actions:
    print(f"Action ID: {action['id']}")
    print(f"Status: {action['status']}")
    print(f"Response: {action.get('response_data')}")