hyperquant 0.42__tar.gz → 0.43__tar.gz

{hyperquant-0.42 → hyperquant-0.43}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperquant
-Version: 0.42
+Version: 0.43
 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

{hyperquant-0.42 → hyperquant-0.43}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "hyperquant"
-version = "0.42"
+version = "0.43"
 description = "A minimal yet hyper-efficient backtesting framework for quantitative trading"
 authors = [
     { name = "MissinA", email = "1421329142@qq.com" }

{hyperquant-0.42 → hyperquant-0.43}/src/hyperquant/broker/models/ourbit.py

@@ -592,7 +592,11 @@ class SpotBalance(DataStore):
     def _on_message(self, msg: dict[str, Any]) -> None:
         data = msg.get("d", {})
         item = self._fmt_ws(data)
-        self._update([item])
+        av = float(item.get("available", 0))
+        if av == 0:
+            self._find_and_delete({'currency': item.get("currency")})
+        else:
+            self._update([item])
 
 
 # SpotOrders: 现货订单数据存储
@@ -929,8 +933,7 @@ class OurbitSpotDataStore(DataStoreCollection):
         """
         现货账户余额数据流
 
-        Data structure:
-
+        _KEYS = ["currency"]
         .. code:: python
 
             [

{hyperquant-0.42 → hyperquant-0.43}/src/hyperquant/core.py

@@ -269,10 +269,10 @@ class ExchangeBase:
         )
         
 class Exchange(ExchangeBase):
-    def __init__(self, trade_symbols, fee=0.0002, initial_balance=10000, recorded=False):
+    def __init__(self, trade_symbols:list=[], fee=0.0002, initial_balance=10000, recorded=False):
         super().__init__(initial_balance=initial_balance, recorded=recorded)
         self.fee = fee
-        self.trade_symbols = trade_symbols
+        self.trade_symbols:list = trade_symbols
         self.id_gen = 0
         self.account['USDT'].update({
             'hold': 0,
@@ -280,8 +280,12 @@ class Exchange(ExchangeBase):
             'short': 0
         })
         for symbol in trade_symbols:
-            self.account[symbol] = {'amount': 0, 'hold_price': 0, 'value': 0, 'price': 0,
-                                     'realised_profit': 0, 'unrealised_profit': 0, 'fee': 0}
+            self.account[symbol] = self._act_template
+    
+    @property
+    def _act_template(self):
+        return {'amount': 0, 'hold_price': 0, 'value': 0, 'price': 0,
+                                     'realised_profit': 0, 'unrealised_profit': 0, 'fee': 0}.copy()
 
     def Trade(self, symbol, direction, price, amount, **kwargs):
         if self.recorded and 'time' not in kwargs:
@@ -307,8 +311,7 @@ class Exchange(ExchangeBase):
 
         if symbol not in self.trade_symbols:
             self.trade_symbols.append(symbol)
-            self.account[symbol] = {'amount': 0, 'hold_price': 0, 'value': 0, 'price': 0,
-                                     'realised_profit': 0, 'unrealised_profit': 0, 'fee': 0}
+            self.account[symbol] = self._act_template
 
         cover_amount = 0 if direction * self.account[symbol]['amount'] >= 0 else min(abs(self.account[symbol]['amount']), amount)
         open_amount = amount - cover_amount

hyperquant-0.43/test.py

@@ -0,0 +1,55 @@
+import asyncio
+import random
+import time
+import pybotters
+import hyperquant
+from hyperquant.broker.models.ourbit import OurbitSpotDataStore
+from hyperquant.broker.ourbit import OurbitSpot
+from hyperquant.core import Exchange
+
+
+async def main():
+    store = OurbitSpotDataStore()
+
+    async with pybotters.Client(
+        apis={
+            "ourbit": [
+                "WEB3bf088f8b2f2fae07592fe1a6240e2d798100a9cb2a91f8fda1056b6865ab3ee"
+            ]
+        }
+    ) as client:
+        res = await client.fetch("POST", "https://www.ourbit.com/ucenter/api/user_info")
+
+        ob_spot = OurbitSpot(client)
+        await ob_spot.__aenter__()
+        await ob_spot.update('balance')
+
+        e = Exchange([], fee=0, initial_balance=0)
+
+        print(ob_spot.store.balance.find())
+
+        for info in ob_spot.store.balance.find():
+            coin = info['currency']
+            if info['currency'] == 'USDT':
+                e.account['USDT']['total'] = float(info['available'])
+            else:
+                if coin not in e.account:
+                    e.account[coin] = e._act_template
+                e.account[coin]['amount'] = float(info['available'])
+                e.account[coin]['hold_price'] = float(info['avg_price'])
+                e.account[coin]['price'] = float(info['avg_price'])
+                e.account[coin]['value'] = float(info['avg_price']) * float(info['available'])
+
+        # "amount": 1,
+        # "hold_price": 1000.0,
+        # "value": 1000.0,
+        # "price": 1000.0,
+        print(e.account)
+
+
+asyncio.run(main())
+# e = Exchange([], fee=0)
+# e.Buy('btc', 1000, 1)
+# e.Update({'btc':1500})
+# print(e.account)
+# print(e.available_margin)

{hyperquant-0.42 → hyperquant-0.43}/tmp.py

@@ -21,29 +21,31 @@ async def main():
 
         ob_spot = OurbitSpot(client)
         await ob_spot.__aenter__()
-        # await ob_spot.sub_personal()
+        await ob_spot.sub_personal()
         # await ob_spot.update('ticker')
         # print(ob_spot.store.ticker.find())
 
         # await ob_spot.sub_orderbook('DOLO_USDT')
         # print(ob_spot.store.book.find())
-        await ob_spot.update('ticker')
-        symbols = [d['symbol'] for d in ob_spot.store.ticker.find()][:5]
+        await ob_spot.update('balance')
+        # symbols = [d['symbol'] for d in ob_spot.store.ticker.find()][:5]
 
-        await ob_spot.sub_orderbook(symbols)
+        # await ob_spot.sub_orderbook(symbols)
 
-        # print(len(ob_spot.store.book.find()))
-        import pandas as pd 
-        print(pd.DataFrame(ob_spot.store.book.find({'S': 'a'})))
+        # # print(len(ob_spot.store.book.find()))
+        # import pandas as pd 
+        # print(pd.DataFrame(ob_spot.store.book.find({'S': 'a'})))
 
 
-        while True:
-            await ob_spot.store.book.wait()
-            print(len(ob_spot.store.book.find()))
-
         # while True:
-        #     await ob_spot.store.balance.wait()
-        #     print(ob_spot.store.balance.find())
+        #     await ob_spot.store.book.wait()
+        #     print(len(ob_spot.store.book.find()))
+
+
+        while True:
+            await ob_spot.store.balance.wait()
+            print(ob_spot.store.balance.find())
+        
         return
 
         await store.initialize(
@@ -136,4 +138,4 @@ async def test_cancel_order():
         await ob_spot.cancel_orders([oid])
 
 if __name__ == "__main__":
-    asyncio.run(test_watch_order())
+    asyncio.run(main())

{hyperquant-0.42 → hyperquant-0.43}/uv.lock

@@ -530,7 +530,7 @@ wheels = [
 
 [[package]]
 name = "hyperquant"
-version = "0.41"
+version = "0.42"
 source = { editable = "." }
 dependencies = [
     { name = "aiohttp" },

hyperquant-0.42/test.py

@@ -1,8 +0,0 @@
-from hyperquant.core import Exchange
-
-
-e = Exchange([], fee=0)
-
-e.Buy('btc', 100, 1, ts=1234)
-
-print(e['btc'])