from blofin.client import Client, DemoClient
from blofin.rest_trading import TradingAPI
from blofin.rest_market import MarketAPI
from blofin.rest_affiliate import AffiliateAPI
from blofin.rest_copytrading import CopyTradingAPI

# Initialize production client
client = Client(
    apiKey="your-api-key",
    apiSecret="your-api-secret",
    passphrase="your-passphrase"
)

# Or initialize demo trading client
demo_client = DemoClient(
    apiKey="your-api-key",
    apiSecret="your-api-secret",
    passphrase="your-passphrase"
)

# Initialize APIs (works with both production and demo clients)
trading = TradingAPI(client)  # or TradingAPI(demo_client)
market = MarketAPI(client)    # or MarketAPI(demo_client)
affiliate = AffiliateAPI(client)  # or AffiliateAPI(demo_client)
copytrading = CopyTradingAPI(client)  # or CopyTradingAPI(demo_client)

from blofin.websocket_client import BlofinWsClient, BlofinWsPublicClient, BlofinWsPrivateClient, BlofinWsCopytradingClient

# Public WebSocket (no authentication required)
public_client = BlofinWsPublicClient()  # For production
demo_public_client = BlofinWsPublicClient(isDemo=True)  # For demo trading

# Private WebSocket (authentication required)
private_client = BlofinWsPrivateClient(
    apiKey="your-api-key",
    secret="your-api-secret",
    passphrase="your-passphrase"
)  # For production

demo_private_client = BlofinWsPrivateClient(
    apiKey="your-api-key",
    secret="your-api-secret",
    passphrase="your-passphrase",
    isDemo=True
)  # For demo trading

# Copytrading WebSocket
copytrading_client = BlofinWsCopytradingClient(
    apiKey="your-api-key",
    secret="your-api-secret",
    passphrase="your-passphrase"
)  # For production