def init_engine_and_indicators(self):
self.initializing = True
try:
self.cbpro_websocket.close()
except:
pass
# Periods to update indicators for
self.indicator_period_list = []
# Periods to actively trade on (typically 1 per product)
self.trade_period_list = {}
# List of products that we are actually monitoring
self.product_list = set()
fiat_currency = self.config['fiat']
if self.config['sandbox']:
api_url = "https://api-public.sandbox.pro.coinbase.com"
else:
api_url = "https://api.pro.coinbase.com"
auth_client = cbpro.AuthenticatedClient(self.config['key'],
self.config['secret'],
self.config['passphrase'],
api_url=api_url)
for cur_period in self.config['periods']:
self.logger.debug("INITIALIZING %s", cur_period['name'])
if cur_period.get('meta'):
new_period = period.MetaPeriod(
period_size=(60 * cur_period['length']),
fiat=fiat_currency,
product=cur_period['product'],
name=cur_period['name'],
cbpro_client=auth_client)
else:
new_period = period.Period(period_size=(60 *
cur_period['length']),
product=cur_period['product'],
name=cur_period['name'],
cbpro_client=auth_client)
self.indicator_period_list.append(new_period)
self.product_list.add(cur_period['product'])
if cur_period['trade']:
if self.trade_period_list.get(cur_period['product']) is None:
self.trade_period_list[cur_period['product']] = []
self.trade_period_list[cur_period['product']].append(
new_period)
max_slippage = Decimal(str(self.config['max_slippage']))
self.trade_engine = engine.TradeEngine(auth_client,
product_list=self.product_list,
fiat=fiat_currency,
is_live=self.config['live'],
max_slippage=max_slippage)
self.cbpro_websocket = engine.TradeAndHeartbeatWebsocket(
fiat=fiat_currency, sandbox=self.config['sandbox'])
self.cbpro_websocket.start()
self.indicator_period_list[0].verbose_heartbeat = True
self.indicator_subsys = indicators.IndicatorSubsystem(
self.indicator_period_list)
self.last_indicator_update = time.time()
self.init_interface()
self.initializing = False
def __init__(self):
cmd.Cmd.__init__(self)
self.config_manager = config_handler.ConfigHandler()
self.config = self.config_manager.get_config()
self.per = period.Period()
self.connection = utm_connect.ServerConnect(self.config)
self.prompt = 'Period (%s %s) Connect (%s) >>> ' % (
self.per.get_period()[0], self.per.get_period()[1],
self.connection.get_status())
self.intro = 'Вас приветствует DGenerator. Для получения справки введите ? или воспользуйтесь командой help'
self.doc_header = 'Доступные команды (для справки по конкретной команде наберите "help <Команда>")'
locale.setlocale(locale.LC_ALL, 'Russian_Russia.1251')
def test_init__initalize_false(self):
test_period = period.Period(period_size=15,
name="BTC15",
product="BTC-USD",
initialize=False)
assert isinstance(test_period, period.Period)
assert test_period.period_size == 15
assert test_period.name == "BTC15"
assert test_period.product == "BTC-USD"
assert test_period.verbose_heartbeat is False
assert isinstance(test_period.candlesticks, type(np.array([])))
np.testing.assert_array_equal(test_period.candlesticks, np.array([]))
def test_init__initalize_true(self, mocker):
mocker.patch("cbpro.PublicClient.get_product_historic_rates",
return_value=self.fake_hist_data)
test_period = period.Period(period_size=5,
name="ETH5",
product="ETH-USD",
initialize=True)
assert isinstance(test_period, period.Period)
assert test_period.period_size == 5
assert test_period.name == "ETH5"
assert test_period.product == "ETH-USD"
assert test_period.verbose_heartbeat is False
assert isinstance(test_period.candlesticks, type(np.array([])))
assert len(test_period.candlesticks) == len(self.fake_hist_data) - 1
assert isinstance(test_period.curr_candlestick, period.Candlestick)
assert test_period.curr_candlestick_start == self.start_time
assert isinstance(test_period.candlesticks[0], type(np.array([])))
error_logger.addHandler(logging.FileHandler("error.log"))
# Periods to update indicators for
indicator_period_list = []
# Periods to actively trade on (typically 1 per product)
trade_period_list = {}
# List of products that we are actually monitoring
product_list = set()
fiat_currency = config['fiat']
for cur_period in config['periods']:
if cur_period.get('meta'):
new_period = period.MetaPeriod(period_size=(60 * cur_period['length']), fiat=fiat_currency,
product=cur_period['product'], name=cur_period['name'])
else:
new_period = period.Period(period_size=(60 * cur_period['length']),
product=cur_period['product'], name=cur_period['name'])
indicator_period_list.append(new_period)
product_list.add(cur_period['product'])
if cur_period['trade']:
if trade_period_list.get(cur_period['product']) is None:
trade_period_list[cur_period['product']] = []
trade_period_list[cur_period['product']].append(new_period)
auth_client = gdax.AuthenticatedClient(config['key'], config['secret'], config['passphrase'])
max_slippage = Decimal(str(config['max_slippage']))
trade_engine = engine.TradeEngine(auth_client, product_list=product_list, fiat=fiat_currency, is_live=config['live'], max_slippage=max_slippage)
gdax_websocket = TradeAndHeartbeatWebsocket(fiat=fiat_currency)
gdax_websocket.start()
indicator_period_list[0].verbose_heartbeat = True
indicator_subsys = indicators.IndicatorSubsystem(indicator_period_list)
last_indicator_update = time.time()
self.websocket_queue.put(msg)
logger = logging.getLogger('trader-logger')
logger.setLevel(logging.DEBUG)
logger.addHandler(logging.FileHandler("debug.log"))
if config.FRONTEND == 'debug':
logger.addHandler(logging.StreamHandler())
error_logger = logging.getLogger('error-logger')
error_logger.addHandler(logging.FileHandler("error.log"))
gdax_websocket = TradeAndHeartbeatWebsocket()
auth_client = gdax.AuthenticatedClient(config.KEY, config.SECRET,
config.PASSPHRASE)
trade_engine = engine.TradeEngine(auth_client, is_live=config.LIVE)
btc_30 = period.Period(period_size=(60 * 30), product='BTC-USD', name='BTC30')
eth_30 = period.Period(period_size=(60 * 30), product='ETH-USD', name='ETH30')
ltc_30 = period.Period(period_size=(60 * 30), product='LTC-USD', name='LTC30')
period_list = [btc_30, eth_30, ltc_30]
gdax_websocket.start()
period_list[0].verbose_heartbeat = True
indicator_subsys = indicators.IndicatorSubsystem(period_list)
last_indicator_update = time.time()
if config.FRONTEND == 'curses':
curses_enable = True
else:
curses_enable = False
interface = curses_interface.cursesDisplay(enable=curses_enable)
while (True):
if len(arguments) == 2 and arguments[0] in range(
1, 13) and arguments[1] in range(2010, 2100):
per.set_period(arguments)
cli.change_promt('DGenerator (%s %s)>>> ' %
tuple(per.get_period()))
else:
print(
'Некорректные параметры. Воспользутесь командой help set_period для получения помощи.'
)
def do_test(self, args):
print(per.get_period())
# Common command methods section
def do_exit(self, arg):
"""Обеспечивает выход из приложения."""
print('Завершение сеанса. До скорой встречи.')
return True
# Dgenerator's CLI loop
if __name__ == "__main__":
per = period.Period()
cli = Cli()
try:
cli.cmdloop()
except KeyboardInterrupt:
print('Завершение сеанса. До скорой встречи.')
def __init__(self):
config_path = os.getenv("CBPRO_CONFIG", "/opt/project/config.yml")
with open(config_path, 'r') as ymlfile:
config = yaml.safe_load(ymlfile)
self.logger = logging.getLogger('trader-logger')
self.logger.setLevel(logging.DEBUG)
if config['logging']:
self.logger.addHandler(logging.StreamHandler())
if config['frontend'] == 'debug':
self.logger.addHandler(logging.StreamHandler())
self.error_logger = logging.getLogger('error-logger')
self.error_logger.addHandler(logging.StreamHandler())
# Periods to update indicators for
self.indicator_period_list = []
# Periods to actively trade on (typically 1 per product)
self.trade_period_list = {}
# List of products that we are actually monitoring
self.product_list = set()
fiat_currency = config['fiat']
if config['sandbox']:
api_url = "https://api-public.sandbox.pro.coinbase.com"
else:
api_url = "https://api.pro.coinbase.com"
auth_client = cbpro.AuthenticatedClient(config['key'], config['secret'], config['passphrase'], api_url=api_url)
for curr_period in config['periods']:
if curr_period.get('meta'):
new_period = period.MetaPeriod(period_size=(60 * curr_period['length']), fiat=fiat_currency,
product=curr_period['product'], name=curr_period['name'], cbpro_client=auth_client)
else:
new_period = period.Period(period_size=(60 * curr_period['length']),
product=curr_period['product'], name=curr_period['name'], cbpro_client=auth_client)
self.indicator_period_list.append(new_period)
self.product_list.add(curr_period['product'])
if curr_period['trade']:
if self.trade_period_list.get(curr_period['product']) is None:
self.trade_period_list[curr_period['product']] = []
self.trade_period_list[curr_period['product']].append(new_period)
max_slippage = Decimal(str(config['max_slippage']))
self.trade_engine = engine.TradeEngine(auth_client, product_list=self.product_list, fiat=fiat_currency, is_live=config['live'], max_slippage=max_slippage)
self.cbpro_websocket = engine.TradeAndHeartbeatWebsocket(fiat=fiat_currency, sandbox=config['sandbox'])
self.cbpro_websocket.start()
self.indicator_period_list[0].verbose_heartbeat = True
self.indicator_subsys = indicators.IndicatorSubsystem(self.indicator_period_list)
self.last_indicator_update = time.time()
if config['frontend'] == 'curses':
curses_enable = True
else:
curses_enable = False
self.interface = interface.cursesDisplay(enable=curses_enable)
if config['frontend'] == 'web':
web_interface = interface.web(self.indicator_subsys, self.trade_engine)
server_thread = threading.Thread(target=web_interface.start, daemon=True)
server_thread.start()
def on_message(self, msg):
if msg.get('type') == "heartbeat" or msg.get('type') == "match":
self.websocket_queue.put(msg)
logger = logging.getLogger('trader-logger')
logger.setLevel(logging.DEBUG)
logger.addHandler(logging.FileHandler("debug.log"))
if config.FRONTEND == 'debug':
logger.addHandler(logging.StreamHandler())
gdax_websocket = TradeAndHeartbeatWebsocket()
auth_client = gdax.AuthenticatedClient(config.KEY, config.SECRET,
config.PASSPHRASE)
trade_engine = engine.TradeEngine(auth_client, is_live=config.LIVE)
one_min = period.Period(period_size=(60 * 1), name='1')
period_list = [one_min]
gdax_websocket.start()
period_list[0].verbose_heartbeat = True
indicator_subsys = indicators.IndicatorSubsystem(period_list)
last_indicator_update = time.time()
if config.FRONTEND == 'curses':
curses_enable = True
else:
curses_enable = False
interface = curses_interface.cursesDisplay(enable=curses_enable)
while (True):
try:
msg = gdax_websocket.websocket_queue.get(timeout=15)
def main(config):
logger = logging.getLogger('trader-logger')
logger.setLevel(logging.DEBUG)
if config['logging']:
logger.addHandler(logging.FileHandler("debug.log"))
if config['frontend'] == 'debug':
logger.addHandler(logging.StreamHandler())
error_logger = logging.getLogger('error-logger')
error_logger.addHandler(logging.FileHandler("error.log"))
# Periods to update indicators for
indicator_period_list = []
# Periods to actively trade on (typically 1 per product)
trade_period_list = {}
# List of products that we are actually monitoring
product_list = set()
fiat_currency = config['fiat']
for cur_period in config['periods']:
if cur_period.get('meta'):
new_period = period.MetaPeriod(period_size=(60 *
cur_period['length']),
fiat=fiat_currency,
product=cur_period['product'],
name=cur_period['name'])
else:
new_period = period.Period(period_size=(60 * cur_period['length']),
product=cur_period['product'],
name=cur_period['name'])
indicator_period_list.append(new_period)
product_list.add(cur_period['product'])
if cur_period['trade']:
if trade_period_list.get(cur_period['product']) is None:
trade_period_list[cur_period['product']] = []
trade_period_list[cur_period['product']].append(new_period)
auth_client = gdax.AuthenticatedClient(config['key'], config['secret'],
config['passphrase'])
max_slippage = Decimal(str(config['max_slippage']))
trade_engine = engine.TradeEngine(auth_client,
product_list=product_list,
fiat=fiat_currency,
is_live=config['live'],
max_slippage=max_slippage)
gdax_websocket = TradeAndHeartbeatWebsocket(fiat=fiat_currency)
gdax_websocket.start()
indicator_period_list[0].verbose_heartbeat = True
indicator_subsys = indicators.IndicatorSubsystem(indicator_period_list)
last_indicator_update = time.time()
if config['frontend'] == 'curses':
interface = interfaces.cursesDisplay(enable=True)
elif config['frontend'] == 'flask':
interface = interfaces.flaskInterface(enable=True)
else:
interface = interfaces.cursesDisplay(enable=False)
while (True):
try:
msg = gdax_websocket.websocket_queue.get(timeout=15)
for product in trade_engine.products:
product.order_book.process_message(msg)
if msg.get('type') == "match":
for cur_period in indicator_period_list:
cur_period.process_trade(msg)
if time.time() - last_indicator_update >= 1.0:
for cur_period in indicator_period_list:
indicator_subsys.recalculate_indicators(cur_period)
for product_id, period_list in trade_period_list.items():
trade_engine.determine_trades(
product_id, period_list,
indicator_subsys.current_indicators)
last_indicator_update = time.time()
elif msg.get('type') == "heartbeat":
for cur_period in indicator_period_list:
cur_period.process_heartbeat(msg)
for product_id, period_list in trade_period_list.items():
if len(indicator_subsys.current_indicators[
cur_period.name]) > 0:
trade_engine.determine_trades(
product_id, period_list,
indicator_subsys.current_indicators)
trade_engine.print_amounts()
interface.update(trade_engine, indicator_subsys.current_indicators,
indicator_period_list, msg)
except KeyboardInterrupt:
trade_engine.close(exit=True)
gdax_websocket.close()
interface.close()
break
except Exception as e:
error_logger.exception(datetime.datetime.now())
trade_engine.close()
gdax_websocket.close()
# Period data cannot be trusted. Re-initialize
for cur_period in indicator_period_list:
cur_period.initialize()
time.sleep(10)
gdax_websocket.start()
product_list = set()
fiat_currency = config['fiat']
if config['sandbox']:
api_url = "https://api-public.sandbox.pro.coinbase.com"
else:
api_url = "https://api.pro.coinbase.com"
auth_client = cbpro.AuthenticatedClient(config['key'], config['secret'], config['passphrase'], api_url=api_url)
for cur_period in config['periods']:
if cur_period.get('meta'):
new_period = period.MetaPeriod(period_size=(60 * cur_period['length']), fiat=fiat_currency,
product=cur_period['product'], name=cur_period['name'], cbpro_client=auth_client)
else:
new_period = period.Period(period_size=(60 * cur_period['length']),
product=cur_period['product'], name=cur_period['name'], cbpro_client=auth_client)
indicator_period_list.append(new_period)
product_list.add(cur_period['product'])
if cur_period['trade']:
if trade_period_list.get(cur_period['product']) is None:
trade_period_list[cur_period['product']] = []
trade_period_list[cur_period['product']].append(new_period)
max_slippage = Decimal(str(config['max_slippage']))
trade_engine = engine.TradeEngine(auth_client, product_list=product_list, fiat=fiat_currency, is_live=config['live'], max_slippage=max_slippage)
cbpro_websocket = engine.TradeAndHeartbeatWebsocket(fiat=fiat_currency, sandbox=config['sandbox'])
cbpro_websocket.start()
indicator_period_list[0].verbose_heartbeat = True
indicator_subsys = indicators.IndicatorSubsystem(indicator_period_list)
last_indicator_update = time.time()
