hyperquant 0.1.1__tar.gz → 0.1.3__tar.gz

{hyperquant-0.1.1 → hyperquant-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperquant
-Version: 0.1.1
+Version: 0.1.3
 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
@@ -12,8 +12,11 @@ Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 Classifier: Topic :: Office/Business :: Financial :: Investment
-Requires-Python: >=3.8
+Requires-Python: >=3.9
+Requires-Dist: aiohttp>=3.11.16
 Requires-Dist: colorama>=0.4.6
+Requires-Dist: cryptography>=44.0.2
+Requires-Dist: numpy>=1.21.0
 Requires-Dist: pandas>=2.2.3
 Requires-Dist: pyecharts>=2.0.8
 Description-Content-Type: text/markdown

{hyperquant-0.1.1 → hyperquant-0.1.3}/pub.sh

@@ -3,11 +3,11 @@ echo "清理旧的构建文件..."
 rm -rf dist/* build/*
 
 echo "重新构建包..."
-rye build
+uv build
 
 echo "发布到 PyPI..."
 # 从 .pypirc 文件读取 token
 TOKEN=$(grep "password" .pypirc | cut -d "=" -f2 | tr -d ' ')
 
 # 使用 token 发布
-TWINE_USERNAME=__token__ TWINE_PASSWORD=$TOKEN rye run twine upload dist/*
+TWINE_USERNAME=__token__ TWINE_PASSWORD=$TOKEN uv run twine upload dist/*

{hyperquant-0.1.1 → hyperquant-0.1.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "hyperquant"
-version = "0.1.1"
+version = "0.1.3"
 description = "A minimal yet hyper-efficient backtesting framework for quantitative trading"
 authors = [
     { name = "MissinA", email = "1421329142@qq.com" }
@@ -9,9 +9,12 @@ dependencies = [
     "pyecharts>=2.0.8",
     "pandas>=2.2.3",
     "colorama>=0.4.6",
+    "aiohttp>=3.11.16",
+    "cryptography>=44.0.2",
+    "numpy>=1.21.0",  # Added numpy as a new dependency
 ]
 readme = "README.md"
-requires-python = ">= 3.8"
+requires-python = ">= 3.9"
 keywords = ["quant", "backtesting", "trading", "hyperquant"]
 license = {text = "MIT"}
 classifiers = [
@@ -30,12 +33,13 @@ Issues = "https://github.com/yourusername/hyperquant/issues"
 requires = ["hatchling"]
 build-backend = "hatchling.build"
 
-[tool.rye]
-managed = true
-dev-dependencies = ["twine>=6.1.0"]
-
 [tool.hatch.metadata]
 allow-direct-references = true
 
 [tool.hatch.build.targets.wheel]
-packages = ["src/hyperquant"]
+packages = ["src/hyperquant"]
+
+[dependency-groups]
+dev = [
+    "twine>=6.1.0",
+]

{hyperquant-0.1.1 → hyperquant-0.1.3}/requirements-dev.lock

@@ -10,18 +10,34 @@
 #   universal: false
 
 -e file:.
+aiohappyeyeballs==2.6.1
+    # via aiohttp
+aiohttp==3.11.16
+    # via hyperquant
+aiosignal==1.3.2
+    # via aiohttp
+attrs==25.3.0
+    # via aiohttp
 certifi==2025.1.31
     # via requests
+cffi==1.17.1
+    # via cryptography
 charset-normalizer==3.4.1
     # via requests
 colorama==0.4.6
-    # via minbacktest
+    # via hyperquant
+cryptography==44.0.2
+    # via hyperquant
 docutils==0.21.2
     # via readme-renderer
+frozenlist==1.6.0
+    # via aiohttp
+    # via aiosignal
 id==1.5.0
     # via twine
 idna==3.10
     # via requests
+    # via yarl
 jaraco-classes==3.4.0
     # via keyring
 jaraco-context==6.0.1
@@ -41,6 +57,9 @@ mdurl==0.1.2
 more-itertools==10.6.0
     # via jaraco-classes
     # via jaraco-functools
+multidict==6.4.3
+    # via aiohttp
+    # via yarl
 nh3==0.2.21
     # via readme-renderer
 numpy==2.2.4
@@ -48,11 +67,16 @@ numpy==2.2.4
 packaging==24.2
     # via twine
 pandas==2.2.3
-    # via minbacktest
+    # via hyperquant
 prettytable==3.15.1
     # via pyecharts
+propcache==0.3.1
+    # via aiohttp
+    # via yarl
+pycparser==2.22
+    # via cffi
 pyecharts==2.0.8
-    # via minbacktest
+    # via hyperquant
 pygments==2.19.1
     # via readme-renderer
     # via rich
@@ -84,3 +108,5 @@ urllib3==2.3.0
     # via twine
 wcwidth==0.2.13
     # via prettytable
+yarl==1.20.0
+    # via aiohttp

{hyperquant-0.1.1 → hyperquant-0.1.3}/requirements.lock

@@ -10,20 +10,45 @@
 #   universal: false
 
 -e file:.
+aiohappyeyeballs==2.6.1
+    # via aiohttp
+aiohttp==3.11.16
+    # via hyperquant
+aiosignal==1.3.2
+    # via aiohttp
+attrs==25.3.0
+    # via aiohttp
+cffi==1.17.1
+    # via cryptography
 colorama==0.4.6
-    # via minbacktest
+    # via hyperquant
+cryptography==44.0.2
+    # via hyperquant
+frozenlist==1.6.0
+    # via aiohttp
+    # via aiosignal
+idna==3.10
+    # via yarl
 jinja2==3.1.6
     # via pyecharts
 markupsafe==3.0.2
     # via jinja2
+multidict==6.4.3
+    # via aiohttp
+    # via yarl
 numpy==2.2.4
     # via pandas
 pandas==2.2.3
-    # via minbacktest
+    # via hyperquant
 prettytable==3.15.1
     # via pyecharts
+propcache==0.3.1
+    # via aiohttp
+    # via yarl
+pycparser==2.22
+    # via cffi
 pyecharts==2.0.8
-    # via minbacktest
+    # via hyperquant
 python-dateutil==2.9.0.post0
     # via pandas
 pytz==2025.1
@@ -36,3 +61,5 @@ tzdata==2025.1
     # via pandas
 wcwidth==0.2.13
     # via prettytable
+yarl==1.20.0
+    # via aiohttp

{hyperquant-0.1.1 → hyperquant-0.1.3}/src/hyperquant/__init__.py

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

hyperquant-0.1.3/src/hyperquant/datavison/_util.py

@@ -0,0 +1,18 @@
+from datetime import date, datetime
+
+
+def _to_milliseconds( t):
+    """
+    支持毫秒时间戳或datetime/date类型，返回毫秒时间戳
+    """
+    if t is None:
+        return None
+    if isinstance(t, int):
+        return t
+    if isinstance(t, float):
+        return int(t * 1000)
+    if isinstance(t, datetime):
+        return int(t.timestamp() * 1000)
+    if isinstance(t, date):
+        return int(datetime.combine(t, datetime.min.time()).timestamp() * 1000)
+    raise ValueError(f"不支持的时间类型: {type(t)}")

hyperquant-0.1.3/src/hyperquant/datavison/binance.py

@@ -0,0 +1,62 @@
+import aiohttp
+import asyncio
+from datetime import datetime, date, timedelta
+# from _util import _to_milliseconds
+from ._util import _to_milliseconds
+import pandas as pd
+
+class BinanceSwap:
+    def __init__(self) -> None:
+        self.session = aiohttp.ClientSession()
+
+    async def get_klines(self, symbol: str, interval: str, start_time, end_time = None, limit: int = 1500):
+        """
+        获取U本位合约K线数据，支持获取任意长度（自动分批）
+
+        :param symbol: 交易对, 如 'BTCUSDT'
+        :param interval: K线间隔, 如 '1m', '5m', '1h', '1d'
+        :param start_time: 开始时间, 毫秒时间戳或datetime/date类型
+        :param end_time: 结束时间, 毫秒时间戳或datetime/date类型, 默认为None表示最新
+        :param limit: 每次请求最大K线数量, 最大1500
+        :return: K线数据DataFrame
+        """
+        url = "https://fapi.binance.com/fapi/v1/klines"
+        all_klines = []
+        fetch_start = _to_milliseconds(start_time)
+        ms_end_time = _to_milliseconds(end_time) if end_time else None
+        while True:
+            params = {
+                "symbol": symbol.upper(),
+                "interval": interval,
+                "startTime": fetch_start,
+                "limit": limit
+            }
+            if ms_end_time:
+                params["endTime"] = ms_end_time
+            async with self.session.get(url, params=params) as resp:
+                resp.raise_for_status()
+                data = await resp.json()
+                if not data:
+                    break
+                all_klines.extend(data)
+                if len(data) < limit:
+                    break
+                last_open_time = data[-1][0]
+                fetch_start = last_open_time + 1
+                if ms_end_time and fetch_start >= ms_end_time:
+                    break
+        # 转为DataFrame
+        columns = [
+            "open_time", "open", "high", "low", "close", "volume",
+            "close_time", "quote_asset_volume", "number_of_trades",
+            "taker_buy_base_asset_volume", "taker_buy_quote_asset_volume", "ignore"
+        ]
+        df = pd.DataFrame(all_klines, columns=columns)
+        # 类型转换
+        df["open_time"] = pd.to_datetime(df["open_time"], unit="ms")
+        df["close_time"] = pd.to_datetime(df["close_time"], unit="ms")
+        for col in ["open", "high", "low", "close", "volume", "quote_asset_volume", "taker_buy_base_asset_volume", "taker_buy_quote_asset_volume"]:
+            df[col] = pd.to_numeric(df[col], errors="coerce")
+        df["number_of_trades"] = df["number_of_trades"].astype(int)
+        return df
+

hyperquant-0.1.3/src/hyperquant/datavison/coinglass.py

@@ -0,0 +1,237 @@
+"""
+Coinglass API 数据解密与数据获取模块
+优化：
+- 只保留必要依赖，整理import顺序
+- 增加类型注解和文档
+- 统一异常处理和日志输出
+- 精简冗余代码
+"""
+import asyncio
+import base64
+import json
+import struct
+import time
+import zlib
+import hmac
+import hashlib
+from datetime import datetime, timedelta
+from typing import Any, Dict, List, Optional, Union
+import pandas as pd # type: ignore
+from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
+from cryptography.hazmat.primitives import padding as crypto_padding
+from cryptography.hazmat.backends import default_backend
+import aiohttp
+
+# ------------------ 工具函数 ------------------
+class CustomParser:
+    @staticmethod
+    def parse(data: str) -> Dict[str, Union[List[int], int]]:
+        """
+        将字符串转换为数字数组，兼容原始加密逻辑。
+        """
+        length = len(data)
+        n = [0] * ((length + 3) // 4)
+        for r in range(length):
+            n[r >> 2] |= (ord(data[r]) & 255) << (24 - (r % 4) * 8)
+        return {"n": n, "e": length}
+
+def convert_words_to_bytes(words: List[int]) -> bytes:
+    """
+    将整数数组转换为字节数组。
+    """
+    return b"".join(struct.pack(">I", word) for word in words)
+
+def decrypt_and_clean(t: str, e: Dict[str, Any]) -> str:
+    """
+    解密、解压缩并清理输入字符串。
+    """
+    aes_key = convert_words_to_bytes(e['n'])
+    cipher = Cipher(algorithms.AES(aes_key), modes.ECB(), backend=default_backend())
+    decryptor = cipher.decryptor()
+    encrypted_data = base64.b64decode(t)
+    decrypted_data = decryptor.update(encrypted_data) + decryptor.finalize()
+    unpadder = crypto_padding.PKCS7(128).unpadder()
+    unpadded_data = unpadder.update(decrypted_data) + unpadder.finalize()
+    decompressed_data = zlib.decompress(unpadded_data, wbits=16 + zlib.MAX_WBITS).decode('utf-8')
+    return decompressed_data
+
+def generate_totp(secret: str, for_time: int, interval: int = 30, digits: int = 6, digest=hashlib.sha1) -> str:
+    """
+    基于标准库的TOTP实现。
+    """
+    key = base64.b32decode(secret, casefold=True)
+    counter = int(for_time // interval)
+    msg = counter.to_bytes(8, 'big')
+    h = hmac.new(key, msg, digest).digest()
+    o = h[-1] & 0x0F
+    code = (struct.unpack('>I', h[o:o+4])[0] & 0x7fffffff) % (10 ** digits)
+    return str(code).zfill(digits)
+
+def generate_encrypted_token() -> str:
+    """
+    生成加密token，用于API请求。
+    """
+    current_time = int(time.time())
+    secret_key = "I65VU7K5ZQL7WB4E"
+    otp = generate_totp(secret_key, current_time)
+    combined_string = f"{current_time},{otp}"
+    aes_key = "1f68efd73f8d4921acc0dead41dd39bc"
+    aes_key_bytes = CustomParser.parse(aes_key)
+    final_key = convert_words_to_bytes(aes_key_bytes['n'])
+    cipher = Cipher(algorithms.AES(final_key), modes.ECB(), backend=default_backend())
+    encryptor = cipher.encryptor()
+    padder = crypto_padding.PKCS7(128).padder()
+    padded_data = padder.update(combined_string.encode('utf-8')) + padder.finalize()
+    encrypted_bytes = encryptor.update(padded_data) + encryptor.finalize()
+    return base64.b64encode(encrypted_bytes).decode('utf-8')
+
+def decrypt_coinglass(data: str, user_header: str, url: str) -> str:
+    """
+    解密 Coinglass API 的响应数据。
+    """
+    base_key = base64.b64encode(f"coinglass{url}coinglass".encode("utf-8")).decode("utf-8")[:16]
+    processed_key = CustomParser.parse(base_key)
+    decrypted_key = decrypt_and_clean(user_header, processed_key)
+    session_key = decrypt_and_clean(data, CustomParser.parse(decrypted_key))
+    return session_key
+
+# ------------------ API类 ------------------
+HEADERS = {
+    'accept': 'application/json',
+    'accept-language': 'en-US,en;q=0.9',
+    'cache-ts': str(int(time.time() * 1000)),
+    'dnt': '1',
+    'encryption': 'true',
+    'language': 'en',
+    'origin': 'https://www.coinglass.com',
+    'priority': 'u=1, i',
+    'referer': 'https://www.coinglass.com/',
+    'sec-ch-ua': '"Google Chrome";v="131", "Chromium";v="131", "Not_A Brand";v="24"',
+    'sec-ch-ua-mobile': '?0',
+    'sec-ch-ua-platform': '"Windows"',
+    'sec-fetch-dest': 'empty',
+    'sec-fetch-mode': 'cors',
+    'sec-fetch-site': 'same-site',
+    'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36',
+}
+
+class CoinglassApi:
+    def __init__(self) -> None:
+        self.session = aiohttp.ClientSession()
+
+    async def connect(self):
+        pass
+
+    async def dec_data(self, response: aiohttp.ClientResponse) -> Optional[Any]:
+        try:
+            encrypted_data = (await response.json())['data']
+            requests_url = response.url.path
+            encrypted_user_header = response.headers.get("user", "HEADERNOTFOUND")
+            decrypted = decrypt_coinglass(encrypted_data, encrypted_user_header, requests_url)
+            return json.loads(decrypted)
+        except Exception as e:
+            print(f"解密失败: {e}")
+            return None
+
+    async def fetch_base_klines(self, symbol: str, start_time: datetime, end_time: Optional[datetime] = None, ktype: str = '#coin#oi_kline', interval: str = 'm1') -> Any:
+        start_ts = int(start_time.timestamp())
+        end_ts = int(end_time.timestamp()) if end_time else int(time.time())
+        url = 'https://fapi.coinglass.com/api/v2/kline'
+        params = {
+            'symbol': f'{symbol}{ktype}',
+            'interval': interval,
+            'endTime': end_ts,
+            'startTime': start_ts,
+            'minLimit': 'false',
+        }
+        async with self.session.get(url, params=params, headers=HEADERS) as response:
+            if response.status == 200:
+                return await self.dec_data(response)
+            print(f"请求失败，状态码: {response.status}: {await response.text()}")
+            return pd.DataFrame()
+
+    async def fetch_price_klines(self, symbol: str, start_time: datetime, end_time: Optional[datetime] = None, interval: str = 'm5') -> pd.DataFrame:
+        data = await self.fetch_base_klines(symbol, start_time, end_time, '#kline', interval)
+        df = pd.DataFrame(data, columns=['timestamp', 'open', 'high', 'low', 'close', 'volume'])
+        df[['open', 'high', 'low', 'close', 'volume']] = df[['open', 'high', 'low', 'close', 'volume']].astype(float)
+        df['timestamp'] = pd.to_datetime(df['timestamp'], unit="s", utc=True)
+        df['symbol'] = symbol
+        return df
+
+    async def fetch_top_account_klines(self, symbol: str, start_time: datetime, end_time: Optional[datetime] = None, interval: str = 'm5') -> Optional[pd.DataFrame]:
+        end_ts = int(end_time.timestamp()) if end_time else int(time.time())
+        start_ts = int(start_time.timestamp())
+        url = 'https://fapi.coinglass.com/api/v2/kline'
+        params = {
+            'symbol': f'{symbol}#top_account_kline',
+            'interval': interval,
+            'endTime': end_ts,
+            'startTime': start_ts,
+            'minLimit': 'false',
+        }
+        async with self.session.get(url, params=params, headers=HEADERS) as response:
+            if response.status == 200:
+                data = await self.dec_data(response)
+                columns = ['timestamp', 'ratio', 'long_ratio', 'short_ratio']
+                df = pd.DataFrame(data, columns=columns)
+                df['timestamp'] = pd.to_datetime(df['timestamp'], unit="s", utc=True)
+                df[['ratio', 'long_ratio', 'short_ratio']] = df[['ratio', 'long_ratio', 'short_ratio']].astype(float)
+                df['symbol'] = symbol
+                return df
+            print(f"请求失败，状态码: {response.status}: {await response.text()}")
+            return None
+
+    async def fetch_oi_klines(self, symbol: str, start_time: datetime, end_time: Optional[datetime] = None, interval: str = 'm5') -> pd.DataFrame:
+        data = await self.fetch_base_klines(symbol, start_time, end_time, '#coin#oi_kline', interval)
+        df = pd.DataFrame(data, columns=['timestamp', 'open', 'high', 'low', 'close'])
+        df[['open', 'high', 'low', 'close']] = df[['open', 'high', 'low', 'close']].astype(float)
+        df['timestamp'] = pd.to_datetime(df['timestamp'], unit="s", utc=True)
+        df['symbol'] = symbol
+        return df
+
+    async def fetch_liq_klines(self, symbol: str, start_time: datetime, end_time: Optional[datetime] = None, interval: str = 'm5') -> pd.DataFrame:
+        data = await self.fetch_base_klines(symbol, start_time, end_time, '#aggregated_liq_kline', interval)
+        df = pd.DataFrame(data, columns=['timestamp', 'short_amount', 'long_amount'])
+        df[['short_amount', 'long_amount']] = df[['short_amount', 'long_amount']].astype(float)
+        df['timestamp'] = pd.to_datetime(df['timestamp'], unit="s", utc=True)
+        df['symbol'] = symbol
+        return df
+
+    async def fetch_hyperliquid_top_positions(self) -> Optional[Any]:
+        url = 'https://capi.coinglass.com/api/hyperliquid/topPosition'
+        async with self.session.get(url, headers=HEADERS) as response:
+            if response.status == 200:
+                return await self.dec_data(response)
+            print(f"请求失败，状态码: {response.status}: {await response.text()}")
+            return None
+
+    async def fetch_tickers(self) -> Optional[Any]:
+        url = 'https://fapi.coinglass.com/api/select/coins/tickers'
+        params = {'exName': 'Binance'}
+        async with self.session.get(url, params=params, headers=HEADERS) as response:
+            if response.status == 200:
+                return await self.dec_data(response)
+            print(f"请求失败，状态码: {response.status}: {await response.text()}")
+            return None
+
+    async def fetch_symbols(self, only_usdt: bool = True) -> List[str]:
+        tickers = await self.fetch_tickers()
+        if tickers:
+            symbols = [ticker['instrument']['instrumentId'] for ticker in tickers]
+            if only_usdt:
+                symbols = [symbol for symbol in symbols if symbol.endswith('USDT')]
+            return symbols
+        return []
+
+    async def stop(self):
+        await self.session.close()
+
+# ------------------ 主程序 ------------------
+async def main():
+    api = CoinglassApi()
+    df = await api.fetch_price_klines('Binance_BTCUSDT', datetime.now() - timedelta(days=1))
+    print(df)
+    await api.stop()
+
+if __name__ == '__main__':
+    asyncio.run(main())

hyperquant-0.1.3/test.py

@@ -0,0 +1,44 @@
+# import asyncio
+# from datetime import datetime, timedelta
+# from src.hyperquant.core import Exchange
+# from src.hyperquant.datavison.coinglass import CoinglassApi
+
+
+# async def main():
+#     api = CoinglassApi()
+#     await api.connect()
+#     df = await api.fetch_price_klines('Binance_BTCUSDT', datetime.now() - timedelta(days=1))
+#     print(df)
+#     await api.stop()
+
+# if __name__ == '__main__':
+#     asyncio.run(main())
+
+
+import asyncio
+from datetime import datetime, timedelta
+from hyperquant.datavison.binance import BinanceSwap
+
+
+if __name__ == "__main__":
+    import time
+
+    async def main():
+        symbol = "BTCUSDT"
+        interval = "1m"
+        # 取最近10分钟的K线
+        end_time = int(time.time() * 1000)
+        start_time = end_time - 10 * 60 * 1000
+        swap = BinanceSwap()
+
+
+        # 新增时间参数例子
+        start_date = datetime.now() - timedelta(minutes=5)
+        end_date = datetime.now()
+        klines = await swap.get_klines(symbol, interval, start_date, end_date)
+
+        print(klines)
+
+        await swap.session.close()
+
+    asyncio.run(main())