hyperquant 0.2__py3-none-any.whl → 0.22__py3-none-any.whl

hyperquant/__init__.py

@@ -2,4 +2,5 @@ from .core import *
 from .draw import *
 from .logkit import *
 from .datavison import *
+from .notikit import *
 __version__ = "0.1.0"

hyperquant/broker/hyperliquid.py

@@ -0,0 +1,538 @@
+# hyperliquid_trader_optimized.py
+"""Async wrapper around Hyperliquid REST + WebSocket endpoints.
+
+Key design goals
+----------------
+* **Single point of truth** – all endpoint paths & constants live at module scope.
+* **Safety** – every public coroutine is fully‑typed and guarded with rich
+  error messages; internal state is protected by an `asyncio.Lock` where
+  necessary.
+* **Performance** – expensive metadata is fetched once and cached; price/size
+  formatting uses the `decimal` module only when needed.
+* **Ergonomics** – high‑level order helpers (`buy`, `sell`) are provided on top
+  of the generic `place_order` routine; context‑manager semantics make sure
+  network resources are cleaned up.
+"""
+from __future__ import annotations
+
+import asyncio
+import decimal
+import itertools
+import logging
+from dataclasses import dataclass
+import time
+from typing import Any, Dict, Optional
+
+import pybotters
+from yarl import URL
+
+from .lib.hpstore import MyHyperStore
+from .lib.hyper_types import AccountBalance
+
+import uuid
+
+
+__all__ = [
+    "HyperliquidTrader",
+]
+
+_API_BASE_MAIN = "https://api.hyperliquid.xyz"
+_API_BASE_TEST = "https://api.hyperliquid-testnet.xyz"
+_WSS_URL_MAIN = "wss://api.hyperliquid.xyz/ws"
+_WSS_URL_TEST = "wss://api.hyperliquid-testnet.xyz/ws"
+_INFO = "/info"
+_EXCHANGE = "/exchange"
+
+logger = logging.getLogger(__name__)
+
+
+# ╭─────────────────────────────────────────────────────────────────────────╮
+# │                                Helpers                                 │
+# ╰─────────────────────────────────────────────────────────────────────────╯
+@dataclass(frozen=True, slots=True)
+class AssetMeta:
+    """Metadata for a tradable asset."""
+
+    asset_id: int
+    name: str
+    sz_decimals: int
+
+@dataclass(frozen=True, slots=True)
+class SpotAssetMeta:
+    """Metadata for a tradable asset."""
+
+    asset_id: int # eg. 10000
+    name: str # eg. "#173"
+    sz_decimals: int # eg. 2
+    index: int # eg. 0
+    token_name: str # eg. "FEUSD"
+    mark_price: float = None
+
+@dataclass
+class OrderData():
+    o_id: str = ''
+    c_id: str = ''
+    name: str = ''
+    status: str = 'resting'
+    price: float = None
+    sz: float = None
+
+
+_DECIMAL_CTX_5 = decimal.Context(prec=5)
+
+
+def _normalize_number(number: str) -> str:
+    """Normalize a number.
+
+    e.g. "3300.0" -> "3300"
+
+    Hyperliquid API expects normalized numbers. Otherwise, `L1 error` will occur.
+    """
+    return format(decimal.Decimal(number).normalize(), "f")
+
+def _fmt_price(price: float, sz_decimals: int, *, max_decimals: int = 6) -> str:
+    """Format *price* according to Hyperliquid rules.
+
+    * For `price >= 1` keep five significant digits.
+    * For small prices keep *max_decimals - sz_decimals* after the dot.
+    """
+    if price >= 1:
+        return format(_DECIMAL_CTX_5.create_decimal(price).normalize(), "f")
+
+    decimal_places = max_decimals - sz_decimals
+    return _normalize_number(format(decimal.Decimal(price).quantize(decimal.Decimal(f"1e-{decimal_places}")), "f"))
+
+
+def _fmt_size(size: float, sz_decimals: int) -> str:
+    return _normalize_number(format(decimal.Decimal(size).quantize(decimal.Decimal(f"1e-{sz_decimals}")), "f"))
+
+
+# ╭─────────────────────────────────────────────────────────────────────────╮
+# │                           Public main class                            │
+# ╰─────────────────────────────────────────────────────────────────────────╯
+class HyperliquidTrader:
+    """High‑level async client for Hyperliquid."""
+
+    def __init__(
+        self,
+        apis: str | dict | None = None,
+        *,
+        client: Optional[pybotters.Client] = None,
+        user_address: Optional[str] = None,
+        msg_callback: Optional[callable[[dict, pybotters.ws.WebSocketAppProtocol], None]] = None,
+        testnet: bool = False,
+    ) -> None:
+        self._external_client = client is not None
+        self._client: pybotters.Client | None = client
+        self._apis = apis
+        self._user = user_address
+        self._msg_cb_user = msg_callback
+
+        self._testnet = testnet
+        self._api_base = _API_BASE_TEST if testnet else _API_BASE_MAIN
+        self._wss_url = _WSS_URL_TEST if testnet else _WSS_URL_MAIN
+
+        self._assets: dict[str, AssetMeta] = {}
+        self._spot_assets: dict[str, SpotAssetMeta] = {}
+
+        self._assets_with_name: dict[str, AssetMeta] = {}
+        self._spot_assets_with_name: dict[str, SpotAssetMeta] = {}
+
+        self._next_id = itertools.count().__next__  # fast thread‑safe counter
+        self._waiters: dict[int, asyncio.Future[dict[str, Any]]] = {}
+        self._waiter_lock = asyncio.Lock()
+
+        self._ws_app: Optional[pybotters.ws.WebSocketConnection] = None
+        self.store: MyHyperStore = MyHyperStore()
+
+
+    # ──────────────────────────────────────────────────────────────────────
+    # Lifecyle helpers
+    # ──────────────────────────────────────────────────────────────────────
+    async def __aenter__(self) -> "HyperliquidTrader":
+        if self._client is None:
+            self._client = await pybotters.Client(apis=self._apis, base_url=self._api_base).__aenter__()
+
+        await self._fetch_meta()
+        await self._fech_spot_meta()
+
+        self._ws_app:pybotters.WebSocketApp = await self._client.ws_connect(
+            self._wss_url,
+            send_json=[],
+            hdlr_json=self._dispatch_msg,
+        )
+
+        if self._user:
+            await self.store.initialize(
+                ("orders", self._client.post(_INFO, data={"type": "openOrders", "user": self._user})),
+            )
+            self._client.ws_connect(
+                self._wss_url,
+                send_json=[
+                {
+                    "method": "subscribe",
+                    "subscription": {
+                        "type": "orderUpdates",
+                        "user": self._user,
+                    },
+                }, 
+                {
+                    "method": "subscribe",
+                    "subscription": {
+                        "type": "userFills",
+                        "user": self._user,
+                    },
+                }, 
+                {
+                    "method": "subscribe",
+                    "subscription": {
+                        "type": "webData2",
+                        "user": self._user,
+                    },
+                }],
+                hdlr_json=self.store.onmessage,
+            )
+
+        return self
+
+    async def __aexit__(self, exc_type, exc, tb):  # noqa: D401
+        if not self._external_client and self._client is not None:
+            await self._client.__aexit__(exc_type, exc, tb)
+
+    # ──────────────────────────────────────────────────────────────────────
+    # Internal – metadata & formatting helpers
+    # ──────────────────────────────────────────────────────────────────────
+    async def _fetch_meta(self) -> None:
+        assert self._client is not None  # mypy
+        resp = await self._client.fetch("POST", _INFO, data={"type": "meta"})
+        if not resp.data:
+            raise RuntimeError(f"Failed to fetch meta: {resp.error}")
+
+        self._assets = {
+            d["name"]: AssetMeta(asset_id=i, name=d["name"], sz_decimals=d["szDecimals"])
+            for i, d in enumerate(resp.data["universe"])
+        }
+
+        logger.debug("Loaded %d assets", len(self._assets))
+
+    async def _fech_spot_meta(self) -> None:
+        assert self._client is not None  # mypy
+        resp = await self._client.fetch("POST", _INFO, data={"type": "spotMeta"})
+        if not resp.data:
+            raise RuntimeError(f"Failed to fetch meta: {resp.error}")
+
+        metadata = resp.data
+
+        tokens = metadata['tokens']
+
+        for u in metadata['universe']:
+            coin_name = u['name']
+            index = u['index']
+            tk_id = u['tokens'][0]
+            token_name = tokens[tk_id]['name']
+            szDecimals = tokens[tk_id]['szDecimals']
+
+            meta =  SpotAssetMeta(
+                asset_id= 10000 + index,
+                name=coin_name,
+                sz_decimals=szDecimals,
+                index=index,
+                token_name=token_name,
+                mark_price=0.0,
+            )
+            self._spot_assets[token_name] = meta
+            self._spot_assets_with_name[coin_name] = meta
+
+        
+
+    def _asset(self, symbol: str, is_spot: bool = False) -> AssetMeta:
+        try:
+            if is_spot:
+                return self._spot_assets[symbol]
+            else:
+                return self._assets[symbol]
+        except KeyError as exc:
+            raise ValueError(f"Unknown asset '{symbol}'. Have you called __aenter__()?") from exc
+        
+    def fmt_price(self, price: float, symbol: str, is_spot: bool = False) -> float:
+        asset = self._asset(symbol, is_spot=is_spot)
+        return float(_fmt_price(price, asset.sz_decimals))
+    
+    def fmt_size(self, size: float, symbol: str, is_spot: bool = False) -> float:
+        asset = self._asset(symbol, is_spot=is_spot)
+        return float(_fmt_size(size, asset.sz_decimals))
+        
+
+    def sub_l2_book(self, symbol: str, is_spot: bool = False) -> None:
+
+        asset = self._asset(symbol, is_spot=is_spot)
+        
+        self._client.ws_connect(
+            self._wss_url,
+            send_json={
+                "method": "subscribe",
+                "subscription": {
+                    "type": "l2Book",
+                    "coin": asset.name,
+                },
+            },
+            hdlr_json=self.store.onmessage
+        )
+  
+
+    # ──────────────────────────────────────────────────────────────────────
+    # Internal – WebSocket message routing
+    # ──────────────────────────────────────────────────────────────────────
+    def _dispatch_msg(self, msg: dict[str, Any], wsapp):  # noqa: ANN001
+        mid = msg.get("data", {}).get("id")
+        if mid is not None:
+            fut = self._waiters.pop(mid, None)
+            if fut and not fut.done():
+                fut.set_result(msg)
+                return
+        # fallback: hand over to user callback / buffer
+        if self._msg_cb_user is not None:
+            self._msg_cb_user(msg, wsapp)
+        else:
+            logger.debug("Unhandled WS message: %s", msg)
+
+    async def _wait_for_id(self, rid: int) -> dict[str, Any]:
+        async with self._waiter_lock:
+            fut = self._waiters[rid] = asyncio.get_event_loop().create_future()
+        return await fut
+
+    # ──────────────────────────────────────────────────────────────────────
+    # REST helpers
+    # ──────────────────────────────────────────────────────────────────────
+    async def _post(self, path: str, data: dict[str, Any]):
+        assert self._client is not None
+        info = await self._client.fetch("POST", path, data=data)
+        info.response.raise_for_status()
+        return info.data
+
+    # ──────────────────────────────────────────────────────────────────────
+    # Public API – account
+    # ──────────────────────────────────────────────────────────────────────
+    async def balances(self, user: Optional[str] = None) -> AccountBalance | None:  # todo support spot
+        try:
+            user = user or self._user
+            if user is None:
+                raise ValueError("User address required – pass it now or in constructor")
+            data = await self._post(_INFO, {"type": "clearinghouseState", "user": user})
+            match data:
+                case pybotters.NotJSONContent():
+                    print('可能参数有误')
+                    return None
+            return data    
+        except Exception as e:
+            print(f"Error fetching balances: {e}")
+            return None
+
+    async def open_orders(self, user: Optional[str] = None):
+        user = user or self._user
+        if user is None:
+            raise ValueError("User address required – pass it now or in constructor")
+        return await self._post(_INFO, {"type": "openOrders", "user": user})
+
+    async def cancel_order(
+        self,
+        asset: str,
+        order_id: str,
+        *,
+        use_ws: bool = False,
+        is_spot: bool = False,
+    ) -> dict[str, Any]:
+
+        meta = self._asset(asset, is_spot=is_spot)
+        
+        payload = {
+            "action": {
+                "type": "cancel",
+                "cancels": [{"a": meta.asset_id, "o": order_id}],
+            }
+        }
+
+
+
+        if not use_ws:
+            return await self._post(_EXCHANGE, payload)
+
+        signed = await self._ws_sign(payload)
+        assert self._ws_app is not None
+        await self._ws_app.current_ws.send_json(signed)
+        return order_id
+    
+    async def cancel_all_orders(
+        self,
+        asset: Optional[str] = None,
+        *,
+        use_ws: bool = False,
+        is_spot: bool = False,
+    ) -> dict[str, Any]:
+        # [{'coin': '@153', 'side': 'A', 'limitPx': '0.9904', 'sz': '202.33', 'oid': 93287495240, 'timestamp': 1747118448026, 'origSz': '202.33', 'cloid': '0x441f6c3b8dde4ccfb34a24ec23419f2d'}, {'coin': '@153', 'side': 'B', 'limitPx': '0.9864', 'sz': '202.76', 'oid': 93278072490, 'timestamp': 1747115006552, 'origSz': '202.76', 'cloid': '0x90fac8c56d224a20b4fd0ab6cf4eae5e'}]
+        orders = await self.open_orders()
+        
+        if asset:
+            meta = self._asset(asset, is_spot=is_spot)
+            orders = [o for o in orders if o['coin'] == meta.name]
+ 
+        if is_spot:
+            for o in orders:
+                o['asset_id'] = self._spot_assets_with_name[o['coin']].asset_id
+        else:
+            for o in orders:
+                o['asset_id'] = self._asset(o['coin']).asset_id
+
+        # 构建payload
+        payload = {
+            "action": {
+                "type": "cancel",
+                "cancels": [
+                    {"a": o['asset_id'], "o": o['oid']}
+                    for o in orders
+                ],
+            }
+        }
+        print(payload)
+
+        if not use_ws:
+            return await self._post(_EXCHANGE, payload)
+        signed = await self._ws_sign(payload)
+        assert self._ws_app is not None
+        await self._ws_app.current_ws.send_json(signed)
+        return orders
+
+    
+    async def get_mid(self, asset: str, is_spot: bool = False) -> float:
+        """Get the mid price of an asset."""
+        meta = self._asset(asset, is_spot=is_spot)
+        book = await self._post(_INFO, {"type": "l2Book", "coin": meta.name})
+        bid = float(book["levels"][0][0]["px"])
+        ask = float(book["levels"][1][0]["px"])
+        return float(_fmt_price((bid + ask) / 2, meta.sz_decimals))
+    
+    async def get_books(self, asset: str, is_spot: bool = False) -> list[float]:
+        """Get the ask prices of an asset."""
+        meta = self._asset(asset, is_spot=is_spot)
+        book = await self._post(_INFO, {"type": "l2Book", "coin": meta.name})
+        levels = book["levels"]
+        bids = levels[0]
+        asks = levels[1]
+        return {
+            "bids": bids,
+            "asks": asks,
+        }
+
+    # ──────────────────────────────────────────────────────────────────────
+    # Public API – trading
+    # ──────────────────────────────────────────────────────────────────────
+    async def place_order(
+        self,
+        asset: str,
+        *,
+        side: str = "buy",  # "buy" | "sell"
+        order_type: str = "market",  # "market" | "limit"
+        size: float,
+        slippage: float = 0.02,
+        price: Optional[float] = None,  # required for limit orders
+        use_ws: bool = False,
+        is_spot: bool = False,
+    ) -> OrderData:
+        """`place_order` 下订单
+
+        返回值:
+         'OrderData'
+        """
+        meta = self._asset(asset, is_spot=is_spot)
+        is_buy = side.lower() == "buy"
+
+        if order_type == "limit":
+            if price is None:
+                raise ValueError("price must be supplied for limit orders")
+            price_str = _fmt_price(price, meta.sz_decimals)
+            tif = "Gtc"
+        else:
+            # emulate market by crossing the spread via mid ± slippage
+            if price is None:
+                mid = await self.get_mid(asset, is_spot=is_spot)
+            else:
+                mid = price  # allow user‑supplied mid for testing
+            crossed = mid * (1 + slippage) if is_buy else mid * (1 - slippage)
+            price_str = _fmt_price(crossed, meta.sz_decimals)
+            tif = "Ioc"
+
+        size_str = _fmt_size(size, meta.sz_decimals)
+        order_payload = {
+            "action": {
+                "type": "order",
+                "orders": [
+                    {
+                        "a": meta.asset_id,
+                        "b": is_buy,
+                        "p": price_str,
+                        "s": size_str,
+                        "r": False,
+                        "t": {"limit": {"tif": tif}},
+                    }
+                ],
+                "grouping": "na",
+            }
+        }
+        # 创建时间戳cid
+        # Generate a 128-bit client order ID as a hex string (0x-prefixed, 32 hex chars)
+        cloid = uuid.uuid4().hex
+        cloid = "0x" + cloid
+        
+        order_payload["action"]["orders"][0]["c"] = cloid
+
+        # print(f"Placing order: {order_payload}")
+        print(order_payload)
+
+        if not use_ws:
+            ret = await self._post(_EXCHANGE, order_payload)
+            if 'error' in str(ret):
+                raise RuntimeError(f"Failed to place order: {ret}")
+            elif 'filled' in str(ret):
+                return OrderData(
+                    o_id=ret['response']['data']['statuses'][0]['filled']['oid'],
+                    c_id=cloid,
+                    name=asset,
+                    status='filled',
+                    price=float(ret['response']['data']['statuses'][0]['filled']['avgPx']),
+                    sz=float(ret['response']['data']['statuses'][0]['filled']['totalSz']),
+                )
+            elif 'resting' in str(ret):
+                return OrderData(
+                    o_id=ret['response']['data']['statuses'][0]['resting']['oid'],
+                    c_id=cloid,
+                    price=float(price_str),
+                    name=asset,
+                    status='resting',
+                )
+
+        # else – signed WS flow
+        signed = await self._ws_sign(order_payload)
+        assert self._ws_app is not None
+        await self._ws_app.current_ws.send_json(signed)
+        return OrderData(o_id=cloid, name=asset, price=float(price_str))
+
+    # Convenience wrappers -------------------------------------------------
+    async def buy(self, asset: str, **kw):
+        return await self.place_order(asset, side="buy", **kw)
+
+    async def sell(self, asset: str, **kw):
+        return await self.place_order(asset, side="sell", **kw)
+
+    # ──────────────────────────────────────────────────────────────────────
+    # Internal – signing helper
+    # ──────────────────────────────────────────────────────────────────────
+    async def _ws_sign(self, payload):  # noqa: ANN001 – hyperliquid internal format
+        # mimic pybotters signing util for WS‑POST messages
+        url = URL(f"{self._api_base}/abc")
+        pybotters.Auth.hyperliquid((None, url), {"data": payload, "session": self._client._session})  # type: ignore[attr-defined]
+        rid = self._next_id()
+        return {
+            "method": "post",
+            "id": rid,
+            "request": {"type": "action", "payload": payload},
+        }

hyperquant/datavison/okx.py

@@ -0,0 +1,177 @@
+import aiohttp
+import asyncio
+import pandas as pd
+import time
+from datetime import datetime, timezone
+from ._util import _to_milliseconds  # 确保时间转换函数可用
+
+class OKX:
+    def __init__(self) -> None:
+        self.session = aiohttp.ClientSession()
+        self.base_url = "https://www.okx.com/api/v5/market"
+
+    async def get_klines(self, symbol: str, interval: str, start_time, end_time=None, limit: int = 100):
+        """
+        获取 OKX 永续合约 K 线数据，带时间过滤，从 end_time 向 start_time 方向翻页。
+
+        :param symbol: 交易对, 如 'BTC-USDT'
+        :param interval: K 线间隔, 如 '1m', '15m', '1H', '4H', '1D'
+        :param start_time: 开始时间(datetime 或 毫秒)
+        :param end_time: 结束时间(datetime 或 毫秒), 可选
+        :param limit: 每次请求最大数量(OKX 最大 300)
+        :return: DataFrame 格式的 K 线数据，按时间升序
+        """
+        if 'h' in interval or 'd' in interval:
+            interval = interval.upper()  # 确保间隔是大写格式
+
+        url = f"{self.base_url}/history-candles"
+        all_rows = []
+        # 转换 start_time 和 end_time 到毫秒时间戳
+        if isinstance(start_time, (int, float)):
+            start_ts = int(start_time)
+        else:
+            # 处理 datetime 对象
+            start_ts = int(start_time.timestamp() * 1000)
+        if end_time:
+            if isinstance(end_time, (int, float)):
+                end_ts = int(end_time)
+            else:
+                end_ts = int(end_time.timestamp() * 1000)
+        else:
+            # 如果没有指定结束时间，就用当前时间戳
+            end_ts = int(time.time() * 1000)
+
+        # 每次请求最多返回 limit=300
+        batch_limit = min(limit, 300)
+        # 初始 after 参数为 end_ts，向过去翻页
+        current_after = end_ts
+
+        while True:
+            params = {
+                "instId": symbol,
+                "bar": interval,
+                "limit": str(batch_limit),
+                "after": str(current_after)
+            }
+            # 发送请求
+            async with self.session.get(url, params=params) as resp:
+                data = await resp.json()
+                if not data or data.get("code") != "0" or not data.get("data"):
+                    # 返回错误或无数据，结束循环
+                    break
+                buf = data["data"]  # 每条是 [ts, o, h, l, c, vol, volCcy, volCcyQuote, confirm]
+
+            # 本批数据按时间从新到旧排列, 最后一条是最旧的
+            rows_this_batch = []
+            for item in buf:
+                ts = int(item[0])
+                # 如果已经早于 start_ts，就跳过，并认为后面更旧，也可以结束循环
+                if ts < start_ts:
+                    continue
+                # 如果某些条目时间超出 end_ts，也跳过
+                if ts > end_ts:
+                    continue
+                # 解析数值字段
+                dt = pd.to_datetime(ts, unit='ms', utc=True)
+                o = float(item[1]); h = float(item[2]); l = float(item[3]); c = float(item[4]); vol = float(item[5])
+                # 按需把每个 K 线封装为字典，后续转换为 DataFrame
+                rows_this_batch.append({
+                    "symbol": symbol,
+                    "open_time": dt,
+                    "open": o,
+                    "high": h,
+                    "low": l,
+                    "close": c,
+                    "volume": vol,
+                    "interval": interval,
+                    "confirm": item[8]
+                })
+
+            if not rows_this_batch:
+                # 本批没有符合时间范围的数据，直接结束
+                break
+
+            # 累积本批符合条件的行
+            all_rows.extend(rows_this_batch)
+
+            # 检查是否到达 start_ts 之前：buf 最后一项是最旧
+            oldest_ts = int(buf[-1][0])
+            if oldest_ts < start_ts:
+                # 已经翻到 start_time 范围之前，结束循环
+                break
+
+            # 否则，更新 after = oldest_ts，继续向过去翻页
+            current_after = oldest_ts
+            # 为了不触发速率限制，稍做休眠（根据需要可以调整或删除）
+
+        # 如果累积到数据，则转换为 DataFrame；否则返回空 DataFrame
+        if all_rows:
+            df = pd.DataFrame(all_rows)
+            # 去重、按时间排序
+            df.drop_duplicates(subset=["open_time"], inplace=True)
+            df.sort_values("open_time", inplace=True)
+            df.reset_index(drop=True, inplace=True)
+            return df
+        else:
+            return pd.DataFrame()
+
+    async def get_index_klines(self, pair: str, interval: str, start_time, end_time=None, limit: int = 100):
+        """
+        获取OKX指数K线数据（自动分批）
+
+        :param pair: 指数名称, 如 'BTC-USD'
+        :param interval: K线间隔, 如 '1m', '1H', '1D'
+        :param start_time: 开始时间(毫秒时间戳/datetime/date)
+        :param end_time: 结束时间(毫秒时间戳/datetime/date)
+        :param limit: 每次请求最大数量(OKX最大300)
+        :return: DataFrame格式的指数K线
+        """
+        url = f"{self.base_url}/index-candles"
+        all_klines = []
+        ms_start = _to_milliseconds(start_time)
+        ms_end = _to_milliseconds(end_time) if end_time else None
+
+        params = {
+            "instId": pair,
+            "bar": interval,
+            "limit": min(limit, 300),
+            "after": ms_start
+        }
+        if ms_end:
+            params["before"] = ms_end
+
+        while True:
+            async with self.session.get(url, params=params) as resp:
+                data = await resp.json()
+                if data['code'] != "0":
+                    raise Exception(f"OKX API Error: {data['msg']} (Code {data['code']})")
+
+                klines = data['data']
+                if not klines:
+                    break
+
+                all_klines.extend(klines)
+
+                if len(klines) < params["limit"]:
+                    break
+
+                last_ts = int(klines[-1][0])
+                params["after"] = last_ts
+
+        # 数据转换
+        columns = ["open_time", "open", "high", "low", "close", "confirm"]
+        df = pd.DataFrame(all_klines, columns=columns)
+
+        df["open_time"] = pd.to_datetime(df["open_time"].astype(int), unit="ms")
+        num_cols = ["open", "high", "low", "close"]
+        df[num_cols] = df[num_cols].apply(pd.to_numeric, errors="coerce")
+
+        return df.sort_values("open_time").reset_index(drop=True)
+
+    async def close(self):
+        """关闭会话"""
+        await self.session.close()
+
+    # 使用示例
+    # async with OKXSwap() as okx:
+    #     df = await okx.get_klines("BTC-USDT", "1H", datetime(2023,1,1))

hyperquant/notikit.py

@@ -0,0 +1,104 @@
+"""
+通知类的功能简单封装，非必要别修改 :)
+只要知道怎么使用以下函数：
+- send_wecom_msg
+- send_wecom_img
+
+Binance期现套利 | 邢不行 | 2024分享会
+author: 邢不行
+微信: xbx6660
+"""
+import base64
+import hashlib
+import os.path
+import requests
+import json
+import traceback
+from datetime import datetime
+
+from hyperquant.logkit import get_logger
+logger = get_logger('notikit', './data/logs/notikit.log', show_time=True)
+
+
+
+proxy = {}
+
+
+def handle_exception(e: Exception, msg: str = '') -> None:
+    logger.error(f"{msg}:{e}")
+    logger.error(e)
+    logger.error(traceback.format_exc())
+
+
+# 企业微信通知
+def send_wecom_msg(content, webhook_url):
+    if not webhook_url:
+        logger.warning('未配置wecom_webhook_url，不发送信息')
+        return
+    if not content:
+        logger.warning('未配置content，不发送信息')
+        return
+    try:
+        data = {
+            "msgtype": "text",
+            "text": {
+                "content": content + '\n' + datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+            }
+        }
+        r = requests.post(webhook_url, data=json.dumps(data), timeout=10, proxies=proxy)
+        logger.info(f'调用企业微信接口返回： {r.text}')
+        logger.ok('成功发送企业微信')
+    except Exception as e:
+        handle_exception(e, '发送企业微信失败')
+
+
+# 上传图片，解析bytes
+class MyEncoder(json.JSONEncoder):
+    def default(self, obj):
+        """
+        只要检查到了是bytes类型的数据就把它转为str类型
+        :param obj:
+        :return:
+        """
+        if isinstance(obj, bytes):
+            return str(obj, encoding='utf-8')
+        return json.JSONEncoder.default(self, obj)
+
+
+# 企业微信发送图片
+def send_wecom_img(file_path, webhook_url):
+    """
+    企业微信发送图片
+    :param file_path: 图片地址
+    :param webhook_url: 企业微信webhook网址
+    :return:
+    """
+    if not os.path.exists(file_path):
+        logger.warning('找不到图片')
+        return
+    if not webhook_url:
+        logger.warning('未配置wecom_webhook_url，不发送信息')
+        return
+    try:
+        with open(file_path, 'rb') as f:
+            image_content = f.read()
+        image_base64 = base64.b64encode(image_content).decode('utf-8')
+        md5 = hashlib.md5()
+        md5.update(image_content)
+        image_md5 = md5.hexdigest()
+        data = {
+            'msgtype': 'image',
+            'image': {
+                'base64': image_base64,
+                'md5': image_md5
+            }
+        }
+        # 服务器上传bytes图片的时候，json.dumps解析会出错，需要自己手动去转一下
+        r = requests.post(webhook_url, data=json.dumps(data, cls=MyEncoder, indent=4), timeout=10, proxies=proxy)
+        logger.info(f'调用企业微信接口返回： {r.text}')
+        logger.ok('成功发送企业微信图片')
+    except Exception as e:
+        handle_exception(e, '发送企业微信图片失败')
+    finally:
+        if os.path.exists(file_path):
+            os.remove(file_path)

{hyperquant-0.2.dist-info → hyperquant-0.22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperquant
-Version: 0.2
+Version: 0.22
 Summary: A minimal yet hyper-efficient backtesting framework for quantitative trading
 Project-URL: Homepage, https://github.com/yourusername/hyperquant
 Project-URL: Issues, https://github.com/yourusername/hyperquant/issues
@@ -19,6 +19,7 @@ Requires-Dist: cryptography>=44.0.2
 Requires-Dist: duckdb>=1.2.2
 Requires-Dist: numpy>=1.21.0
 Requires-Dist: pandas>=2.2.3
+Requires-Dist: pybotters>=1.8.2
 Requires-Dist: pyecharts>=2.0.8
 Description-Content-Type: text/markdown
 

{hyperquant-0.2.dist-info → hyperquant-0.22.dist-info}/RECORD

@@ -1,11 +1,14 @@
-hyperquant/__init__.py,sha256=ogwPnRZdsCG5VqMpb_BOtmsKW64G9YmZ7t4AMGttLAE,108
+hyperquant/__init__.py,sha256=gUuAPVpg5k8X_dpda5OpqmMyZ-ZXNQq-xwx-6JR5Jr4,131
 hyperquant/core.py,sha256=vKv8KElo1eGhr_aw0I-j6ZxPOneDx86KqAoOI-wbq0A,18838
 hyperquant/db.py,sha256=i2TjkCbmH4Uxo7UTDvOYBfy973gLcGexdzuT_YcSeIE,6678
 hyperquant/draw.py,sha256=up_lQ3pHeVLoNOyh9vPjgNwjD0M-6_IetSGviQUgjhY,54624
 hyperquant/logkit.py,sha256=WALpXpIA3Ywr5DxKKK3k5EKubZ2h-ISGfc5dUReQUBQ,7795
+hyperquant/notikit.py,sha256=IztRn_qSBPf9IbDeQaruW_QSqs7TMB3h-V6obIuA1Mo,3090
+hyperquant/broker/hyperliquid.py,sha256=mSBVfmjcv6ciI1vWrmHYwBOTHrg-NQrwcyVFUXYEgVw,21998
 hyperquant/datavison/_util.py,sha256=92qk4vO856RqycO0YqEIHJlEg-W9XKapDVqAMxe6rbw,533
 hyperquant/datavison/binance.py,sha256=3yNKTqvt_vUQcxzeX4ocMsI5k6Q6gLZrvgXxAEad6Kc,5001
 hyperquant/datavison/coinglass.py,sha256=PEjdjISP9QUKD_xzXNzhJ9WFDTlkBrRQlVL-5pxD5mo,10482
-hyperquant-0.2.dist-info/METADATA,sha256=6p03HReifvxASDUM2W-lYx1GfkrrBEzf4E2n9UsaoK8,4284
-hyperquant-0.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-hyperquant-0.2.dist-info/RECORD,,
+hyperquant/datavison/okx.py,sha256=yg8WrdQ7wgWHNAInIgsWPM47N3Wkfr253169IPAycAY,6898
+hyperquant-0.22.dist-info/METADATA,sha256=7Q9d1SaurAVVr9nOv-vnESvINj2pjqjVu_kboAqM1Ug,4317
+hyperquant-0.22.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hyperquant-0.22.dist-info/RECORD,,