1#!/usr/bin/env python3
2"""
3Coinglass Long/Short Ratio Module
4 
5Provides aggregate long/short position ratio data across exchanges.
6Ratio > 1 means more longs; < 1 means more shorts.
7"""
8 
9import sys
10import json
11import argparse
12from typing import Dict, Any, Optional, List
13 
14from ._api import cg_request
15 
16 
17def get_long_short_ratio(
18    symbol: str = "BTC",
19    interval: str = "h4"
20) -> Optional[Dict[str, Any]]:
21    """
22    Get long/short ratio data across exchanges.
23 
24    Args:
25        symbol: Coin symbol (BTC, ETH, etc.)
26        interval: Time interval (h1, h4, h12, h24).
27 
28    Returns:
29        Dict with exchange-level long/short data.
30 
31    Raises:
32        CoinglassError: On API failure.
33    """
34    data = cg_request(
35        "long_short", params={"symbol": symbol, "time_type": interval},
36        version="v2"
37    )
38 
39    # v2 returns wrapped data — may need to handle both list and dict
40    return data
41 
42 
43def get_exchange_ratio(
44    symbol: str = "BTC",
45    exchange: str = "Binance",
46    interval: str = "h4"
47) -> Optional[Dict[str, Any]]:
48    """
49    Get long/short ratio for a specific exchange.
50 
51    Args:
52        symbol: Coin symbol.
53        exchange: Exchange name.
54        interval: Time interval.
55 
56    Returns:
57        Dict with ratio, longPercent, shortPercent for the exchange.
58        None if exchange not found in data.
59    """
60    data = get_long_short_ratio(symbol, interval)
61    if not data:
62        return None
63 
64    # Data may be a list of exchange entries
65    entries = data if isinstance(data, list) else [data]
66    for entry in entries:
67        if entry.get("exchangeName", "").lower() == exchange.lower():
68            long_pct = entry.get("longRate", 0)
69            short_pct = entry.get("shortRate", 0)
70            ratio = long_pct / short_pct if short_pct else 0
71            return {
72                "symbol": symbol.upper(),
73                "exchange": entry.get("exchangeName"),
74                "long_percent": long_pct * 100,
75                "short_percent": short_pct * 100,
76                "ratio": ratio,
77            }
78    return None
79 
80 
81def get_sentiment(
82    symbol: str = "BTC",
83    interval: str = "h4"
84) -> Optional[Dict[str, Any]]:
85    """
86    Get aggregated market sentiment from long/short ratios.
87 
88    Args:
89        symbol: Coin symbol.
90        interval: Time interval.
91 
92    Returns:
93        Dict with avg ratio, sentiment label, and per-exchange breakdown.
94    """
95    data = get_long_short_ratio(symbol, interval)
96    if not data:
97        return None
98 
99    entries = data if isinstance(data, list) else [data]
100    ratios = []
101    for entry in entries:
102        long_r = entry.get("longRate", 0)
103        short_r = entry.get("shortRate", 0)
104        if short_r:
105            ratios.append({
106                "exchange": entry.get("exchangeName", ""),
107                "ratio": long_r / short_r,
108            })
109 
110    if not ratios:
111        return None
112 
113    avg_ratio = sum(r["ratio"] for r in ratios) / len(ratios)
114    if avg_ratio > 1.2:
115        sentiment = "Very Bullish"
116    elif avg_ratio > 1.0:
117        sentiment = "Bullish"
118    elif avg_ratio > 0.8:
119        sentiment = "Bearish"
120    else:
121        sentiment = "Very Bearish"
122 
123    return {
124        "symbol": symbol.upper(),
125        "avg_ratio": avg_ratio,
126        "sentiment": sentiment,
127        "exchanges": ratios,
128    }
129 
130 
131def compare_exchanges(
132    symbol: str = "BTC",
133    interval: str = "h4"
134) -> Optional[List[Dict[str, Any]]]:
135    """
136    Compare long/short ratios across all exchanges.
137 
138    Returns:
139        Sorted list of exchanges by ratio (most bullish first).
140    """
141    data = get_long_short_ratio(symbol, interval)
142    if not data:
143        return None
144 
145    entries = data if isinstance(data, list) else [data]
146    results = []
147    for entry in entries:
148        long_r = entry.get("longRate", 0)
149        short_r = entry.get("shortRate", 0)
150        ratio = long_r / short_r if short_r else 0
151        results.append({
152            "exchange": entry.get("exchangeName", ""),
153            "ratio": round(ratio, 4),
154            "long_percent": round(long_r * 100, 2),
155            "short_percent": round(short_r * 100, 2),
156        })
157 
158    results.sort(key=lambda x: x["ratio"], reverse=True)
159    return results if results else None
160 
161 
162def main():
163    """CLI entry point."""
164    parser = argparse.ArgumentParser(
165        description="Coinglass Long/Short Ratio Tools"
166    )
167    parser.add_argument("--symbol", "-s", default="BTC")
168    parser.add_argument("--exchange", "-e", default=None)
169    parser.add_argument("--interval", "-i", default="h4")
170    parser.add_argument("--sentiment", action="store_true",
171                        help="Show sentiment analysis")
172    parser.add_argument("--compare", action="store_true",
173                        help="Compare across exchanges")
174    args = parser.parse_args()
175 
176    try:
177        if args.sentiment:
178            result = get_sentiment(args.symbol, args.interval)
179        elif args.compare:
180            result = compare_exchanges(args.symbol, args.interval)
181        elif args.exchange:
182            result = get_exchange_ratio(
183                args.symbol, args.exchange, args.interval
184            )
185        else:
186            result = get_long_short_ratio(args.symbol, args.interval)
187 
188        if result:
189            print(json.dumps(result, indent=2))
190        else:
191            print("No data found", file=sys.stderr)
192    except Exception as e:
193        print(f"Error: {e}", file=sys.stderr)
194        sys.exit(1)
195 
196 
197if __name__ == "__main__":
198    main()
199