Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
executable file 184 lines (151 sloc) 9.41 KB
#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# File: example_stream_management_extended.py
#
# Part of ‘UNICORN Binance WebSocket API’
# Project website: https://github.com/unicorn-data-analysis/unicorn-binance-websocket-api
# Documentation: https://www.unicorn-data.com/unicorn-binance-websocket-api.html
# PyPI: https://pypi.org/project/unicorn-binance-websocket-api/
#
# Author: UNICORN Data Analysis
# https://www.unicorn-data.com/
#
# Copyright (c) 2019, UNICORN Data Analysis
# All rights reserved.
#
# Permission is hereby granted, free of charge, to any person obtaining a
# copy of this software and associated documentation files (the
# "Software"), to deal in the Software without restriction, including
# without limitation the rights to use, copy, modify, merge, publish, dis-
# tribute, sublicense, and/or sell copies of the Software, and to permit
# persons to whom the Software is furnished to do so, subject to the fol-
# lowing conditions:
#
# The above copyright notice and this permission notice shall be included
# in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL-
# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
# IN THE SOFTWARE.
from unicorn_binance_websocket_api.unicorn_binance_websocket_api_manager import BinanceWebSocketApiManager
import logging
import time
# import class to process stream data
from unicorn_binance_websocket_api_process_streams import BinanceWebSocketApiProcessStreams
# https://docs.python.org/3/library/logging.html#logging-levels
logging.getLogger('unicorn-log').setLevel(logging.INFO)
logging.getLogger('unicorn-log').addHandler(logging.StreamHandler())
# create instance of BinanceWebSocketApiManager and provide the function for stream processing
binance_websocket_api_manager = BinanceWebSocketApiManager(BinanceWebSocketApiProcessStreams.process_stream_data)
# define markets
markets = {'bnbbtc', 'ethbtc', 'btcusdt', 'bchabcusdt', 'eosusdt'}
markets_mega_list = ['bnbbtc', 'ethbtc', 'btcusdt', 'bchabcusdt', 'xrpusdt', 'rvnbtc', 'ltcusdt', 'adausdt', 'eosusdt',
'neousdt', 'bnbusdt', 'adabtc', 'ethusdt', 'trxbtc', 'trxbtc', 'bchabcbtc', 'ltcbtc', 'xrpbtc',
'ontbtc', 'bttusdt', 'eosbtc', 'xlmbtc', 'bttbtc', 'tusdusdt', 'xlmusdt', 'qkcbtc', 'zrxbtc',
'neobtc', 'adaeth', 'icxusdt', 'btctusd', 'icxbtc', 'btcusdc', 'wanbtc', 'zecbtc', 'wtcbtc',
'batbtc', 'adabnb', 'etcusdt', 'qtumusdt', 'xmrbtc', 'trxeth', 'adatusd', 'trxxrp', 'trxbnb',
'dashbtc', 'rvnbnb', 'bchabctusd', 'etcbtc', 'bnbeth', 'ethpax', 'nanobtc', 'xembtc', 'xrpbnb',
'bchabcpax', 'xrpeth', 'bttbnb', 'ltcbnb', 'agibtc', 'zrxusdt', 'xlmbnb', 'ltceth', 'eoseth',
'ltctusd', 'polybnb', 'scbtc', 'steembtc', 'trxtusd', 'npxseth', 'kmdbtc', 'polybtc', 'gasbtc',
'engbtc', 'zileth', 'xlmeth', 'eosbnb', 'xrppax', 'lskbtc', 'npxsbtc', 'xmrusdt', 'ltcpax', 'xmrusdt',
'ethtusd', 'batusdt', 'mcobtc', 'neoeth', 'bntbtc', 'eostusd', 'lrcbtc', 'funbtc', 'zecusdt',
'bnbpax', 'linkusdt', 'hceth', 'zrxeth', 'icxeth', 'xmreth', 'neobnb', 'etceth', 'zeceth', 'xmrbnb',
'wanbnb', 'zrxbnb', 'agibnb', 'funeth', 'arketh', 'engeth']
# define channels
channels = {'trade', 'kline_1m', 'kline_5m', 'kline_15m', 'kline_30m', 'kline_1h', 'kline_12h', 'kline_1w', 'miniTicker'}
# to establish a connection to a websocket server, the client has to call an URI. this URI must not be longer then 8004
# signs, if you want to get an active connection to a Binance webstream.
# https://github.com/unicorn-data-analysis/unicorn-binance-websocket-api/blob/master/tools/test_max_websocket_uri_length.py
if binance_websocket_api_manager.is_websocket_uri_length_valid(channels, markets):
print("URI with `markets` length is valid!")
else:
print("URI with `markets` length is too long!")
if binance_websocket_api_manager.is_websocket_uri_length_valid(channels, markets_mega_list):
print("URI with `markets_mega_list` length is valid!")
else:
print("URI with `markets_mega_list` length is too long!")
print("############################################################################################################\r\n")
# create and start some streams
first_multi_stream_id = binance_websocket_api_manager.create_stream(channels, markets)
ticker_stream_id = binance_websocket_api_manager.create_stream(["arr"], ["!miniTicker"])
miniticker_stream_id = binance_websocket_api_manager.create_stream(["arr"], ["!ticker"])
time.sleep(4)
first_multi_stream_info = binance_websocket_api_manager.get_stream_info(first_multi_stream_id)
print(first_multi_stream_id, "received", first_multi_stream_info['processed_receives_total'], "records till now! ##\r\n")
# stop the streams
binance_websocket_api_manager.stop_stream(first_multi_stream_id)
binance_websocket_api_manager.stop_stream(miniticker_stream_id)
print("############################################################################################################\r\n"
"# waiting, till stream", first_multi_stream_id, "has stopped!\r\n"
"############################################################################################################")
binance_websocket_api_manager.wait_till_stream_has_stopped(first_multi_stream_id)
print("############################################################################################################\r\n"
"#", first_multi_stream_id, "stopped\r\n"
"############################################################################################################")
first_multi_stream_runtime = time.time() - first_multi_stream_info['start_time']
print("############################################################################################################\r\n"
"# stopping ticker stream with ID", ticker_stream_id, "(received",
first_multi_stream_info['processed_receives_total'], "records in ", first_multi_stream_runtime, " seconds!)\r\n"
"############################################################################################################")
binance_websocket_api_manager.stop_stream(ticker_stream_id)
if binance_websocket_api_manager.wait_till_stream_has_stopped(ticker_stream_id):
print("\r\n#### ticker stream with id (", ticker_stream_id, "has stopped! ####")
# get stream infos
print("\r\n\r\ntrade_stream_info:")
stream_list = binance_websocket_api_manager.get_stream_list()
print(stream_list, "\r\n")
time.sleep(3)
# replace stream with an other one:
# first add or remove markets from the markets list
markets.add('neobtc')
markets.add('rvnbtc')
markets.remove('eosusdt')
# edit channels list
channels.remove('miniTicker')
channels.add('kline_2w')
channels.add('depth5')
# start the new multi stream
second_multi_stream_id = binance_websocket_api_manager.create_stream(channels, markets)
# get info about the new stream
second_multi_stream_info = binance_websocket_api_manager.get_stream_info(second_multi_stream_id)
print(second_multi_stream_info)
# wait till second multi stream socket received its first data row
if binance_websocket_api_manager.wait_till_stream_has_started(second_multi_stream_id):
# now close the first multi socket stream
binance_websocket_api_manager.stop_stream(first_multi_stream_id)
time.sleep(3)
binance_websocket_api_manager.stop_stream(second_multi_stream_id)
# print stream list
print("\r\n\r\ntrade_stream_list:")
print(binance_websocket_api_manager.get_stream_list())
print("\r\ninfo first multi stream")
print(binance_websocket_api_manager.get_stream_info(first_multi_stream_id))
print("\r\ninfo ticker stream:")
print(binance_websocket_api_manager.get_stream_info(ticker_stream_id))
print("\r\ninfo miniTicker stream")
print(binance_websocket_api_manager.get_stream_info(miniticker_stream_id))
print("\r\ninfo second multi stream")
print(binance_websocket_api_manager.get_stream_info(second_multi_stream_id))
second_multi_stream_info = binance_websocket_api_manager.get_stream_info(second_multi_stream_id)
print("\r\n####", second_multi_stream_id, "status", second_multi_stream_info['status'], "####\r\n")
while binance_websocket_api_manager.get_active_stream_list():
print(binance_websocket_api_manager.get_active_stream_list())
time.sleep(2)
second_multi_stream_info = binance_websocket_api_manager.get_stream_info(second_multi_stream_id)
print("\r\n####", second_multi_stream_id, "status:", second_multi_stream_info['status'], "and received",
second_multi_stream_info['processed_receives_total'], "records! ####\r\n")
print("\r\ntrade_stream_list:")
print(binance_websocket_api_manager.get_stream_list())
binance_websocket_api_manager.print_stream_info(first_multi_stream_id)
binance_websocket_api_manager.print_stream_info(ticker_stream_id)
binance_websocket_api_manager.print_stream_info(miniticker_stream_id)
binance_websocket_api_manager.print_stream_info(second_multi_stream_id)
binance_websocket_api_manager.print_summary()
print("\r\n=============================== Stopping BinanceWebSocketManager ======================================\r\n")
binance_websocket_api_manager.stop_manager_with_all_streams()
print("finished!")
You can’t perform that action at this time.