def test_level2_of(self): # Sample of level2. {'msgid': 21014, 'quotes': {'QJSIM¦SBER': {'lines': {'22806': # {'b': 234, 's': 0, 'by': 0, 'sy': 0}, '22841': {'b': 437, 's': 0, 'by': 0, 'sy': 0}, # '22853': {'b': 60, 's': 0, 'by': 0, 'sy': 0}, '22878': {'b': 82, 's': 0, 'by': 0, 'sy': 0}, # '22886': {'b': 138, 's': 0, 'by': 0, 'sy': 0}, '22895': {'b': 1, 's': 0, 'by': 0, 'sy': 0},... data = {1: {'b': 4, 's': 7, 'by': 0, 'sy': 0}, 2: {'b': 5, 's': 8, 'by': 0, 'sy': 0}, 3: {'b': 6, 's': 9, 'by': 0, 'sy': 0}} level2 = WebQuikFeed._level2_of(datetime(2021, 7, 23, 12, 51), Asset("code1", "sec1"), data) self.assertEqual(level2.dt, datetime(2021, 7, 23, 12, 51)) self.assertEqual(level2.asset, Asset("code1", "sec1")) self.assertEqual([1, 2, 3], [item.price for item in level2.items]) self.assertEqual([4, 5, 6], [item.bid_vol for item in level2.items]) self.assertEqual([7, 8, 9], [item.ask_vol for item in level2.items])
def on_trade_session_open(self, msg): """ On start, trade session is opened. Now we can request data and set orders """ self._logger.debug(f"Trade session opened. Requesting feeds for subscribers: {self._feed_subscribers}") for asset in filter(lambda a: a != Asset.any_asset(), self._feed_subscribers): self._request_feed(asset)
def test_quote_of(self): data = {"bid": 1, "offer": 2, "last": 3, "lastchange": 4} quote = WebQuikFeed._quote_of("stock1¦asset1", data) self.assertIsNotNone(quote.dt) self.assertEqual(Asset("stock1", "asset1"), quote.asset) self.assertEqual(1, quote.bid) self.assertEqual(2, quote.ask) self.assertEqual(3, quote.last) self.assertEqual(4, quote.last_change)
def _asset_of(quik_str: str) -> Optional[Asset]: # Example: "QJSIM¦SBER¦0", we need to streap trailing \0 if not quik_str: return None # groups = re.search('([\\w\\d]+(?=\\|))*\\|*([\\w\\d]+)?', quik_str) # groups = re.match('(?P<class_code>[\\w\\d]+(?=\\|))?\\|?(?P<sec_code>[\\w\\d]+)', quik_str) groups = re.match('(?P<class_code>[\\w\\d]+(?=¦))?¦?(?P<sec_code>[\\w\\d]+)', quik_str) asset = Asset(groups["class_code"], groups["sec_code"]) return asset
def test_ohlcv_of(self): data = {"d": "2019-10-01 10:02:00", "o": 1, "c": 2, "h": 3, "l": 4, "v": 5} ohlcv = WebQuikFeed._ohlcv_of("stock1¦asset1", data) self.assertEqual(ohlcv.asset, Asset("stock1","asset1")) self.assertEqual(ohlcv.dt, datetime(2019, 10, 1, 10, 2, 0)) self.assertEqual(ohlcv.o, 1) self.assertEqual(ohlcv.h, 3) self.assertEqual(ohlcv.l, 4) self.assertEqual(ohlcv.c, 2) self.assertEqual(ohlcv.v, 5)
def _on_level2(self, data: dict): """ Level 2 data handler. Quik sends us full level2 snapshot. """ # Sample of level2. {'msgid': 21014, 'quotes': {'QJSIM¦SBER': {'lines': {'22806': # {'b': 234, 's': 0, 'by': 0, 'sy': 0}, '22841': {'b': 437, 's': 0, 'by': 0, 'sy': 0}, # '22853': {'b': 60, 's': 0, 'by': 0, 'sy': 0}, '22878': {'b': 82, 's': 0, 'by': 0, 'sy': 0}, # '22886': {'b': 138, 's': 0, 'by': 0, 'sy': 0}, '22895': {'b': 1, 's': 0, 'by': 0, 'sy': 0},... # Go through all assets in level2 message # Todo: get rid of nested check for asset_str in data['quotes']: # asset_class, asset_code = self._connector.asset2tuple(asset_str) asset = WebQuikFeed._asset_of(asset_str) if asset not in self._feed_subscribers: continue # {'22806': {'b': 234, 's': 0, 'by': 0, 'sy': 0}, ..} level2_quik: dict = data['quotes'][asset_str]['lines'] # level2 = {} # Level2(datetime.now()) # Fill in level2 items items = [] for key in level2_quik.keys(): price = int(key) bid = level2_quik[key]['b'] if bid == 0: bid = None ask = level2_quik[key]['s'] if ask == 0: ask = None items.append(Level2Item(price, bid, ask)) level2 = Level2.of(datetime.now(),asset, items) # If somebody subscribed to level2 of this asset, send her this data. subscribers = self._feed_subscribers[asset] + self._feed_subscribers[Asset.any_asset()] for subscriber in filter(lambda s: s.on_level2, subscribers): subscriber.on_level2(level2)
def test_ticker_of_empty(self): ticker = WebQuikFeed._ticker_of(Asset("", "")) self.assertEqual("|", ticker)
def test_ticker_of_None(self): ticker = WebQuikFeed._ticker_of(Asset(None, None)) self.assertEqual("None|None", ticker)
def test_ticker_of_full_asset(self): ticker = WebQuikFeed._ticker_of(Asset("class1", "sec1")) self.assertEqual("class1|sec1", ticker)
def _on_quotes(self, data: dict): """ Bid/ask spreads callback Msg sample: {"msgid":21011,"dataResult":{"CETS\u00A6BYNRUBTODTOM":{"bid":0, "ask":10, last":0,"lastchange":... """ self._logger.debug('Got bid/ask quotes: %s', data) for quik_asset in data['dataResult'].keys(): asset = WebQuikFeed._asset_of(quik_asset) if asset in self._feed_subscribers.keys(): quote = WebQuikFeed._quote_of(quik_asset, data['dataResult'][quik_asset]) # Send to subscriber for subscriber in self._feed_subscribers[asset] + self._feed_subscribers[Asset.any_asset()]: subscriber.on_quote(quote)