from dflow import DFlowClient
 
client = DFlowClient()
orderbook_api = client.orderbook

def get_orderbook(
    market_id: str,
    depth: int | None = None
) -> Orderbook

orderbook = client.orderbook.get_orderbook("BTCUSD-25JAN-100000")
 
print("Best Bid:", orderbook.bids[0] if orderbook.bids else "None")
print("Best Ask:", orderbook.asks[0] if orderbook.asks else "None")
 
# With depth limit
orderbook = client.orderbook.get_orderbook("BTCUSD-25JAN-100000", depth=5)

class Orderbook(BaseModel):
    market_id: str
    bids: list[OrderbookLevel]
    asks: list[OrderbookLevel]
    timestamp: str | None

class OrderbookLevel(BaseModel):
    price: str
    quantity: str

def display_orderbook(orderbook: Orderbook):
    print(f"Orderbook for {orderbook.market_id}")
    print("-" * 40)
    
    print("ASKS (Sell Orders):")
    for level in reversed(orderbook.asks[:5]):
        print(f"  ${level.price} - {level.quantity} contracts")
    
    print("-" * 40)
    
    print("BIDS (Buy Orders):")
    for level in orderbook.bids[:5]:
        print(f"  ${level.price} - {level.quantity} contracts")
 
# Usage
orderbook = client.orderbook.get_orderbook("market-id")
display_orderbook(orderbook)