freqtrade_origin/scripts/download_backtest_data.py

63 lines
1.8 KiB
Python
Raw Normal View History

2018-04-22 07:56:49 +00:00
#!/usr/bin/env python3
"""This script generate json data from bittrex"""
import json
import sys
import os
import arrow
from freqtrade import (exchange, arguments, misc)
2018-04-22 07:56:49 +00:00
DEFAULT_DL_PATH = 'user_data/data'
2018-04-22 07:56:49 +00:00
arguments = arguments.Arguments(sys.argv[1:], 'download utility')
arguments.testdata_dl_options()
args = arguments.parse_args()
timeframes = args.timeframes
2018-04-22 07:56:49 +00:00
2018-06-04 09:58:35 +00:00
dl_path = os.path.join(DEFAULT_DL_PATH, args.exchange)
if args.export:
dl_path = args.export
if not os.path.isdir(dl_path):
sys.exit(f'Directory {dl_path} does not exist.')
pairs_file = args.pairs_file if args.pairs_file else os.path.join(dl_path, 'pairs.json')
if not os.path.isfile(pairs_file):
sys.exit(f'No pairs file found with path {pairs_file}.')
2018-04-22 07:56:49 +00:00
with open(pairs_file) as file:
PAIRS = list(set(json.load(file)))
2018-04-22 07:56:49 +00:00
since_time = None
if args.days:
since_time = arrow.utcnow().shift(days=-args.days).timestamp * 1000
2018-04-22 07:56:49 +00:00
print(f'About to download pairs: {PAIRS} to {dl_path}')
# Init exchange
2018-04-22 07:56:49 +00:00
exchange._API = exchange.init_ccxt({'key': '',
'secret': '',
'name': args.exchange})
2018-04-22 07:56:49 +00:00
2018-04-22 07:56:49 +00:00
for pair in PAIRS:
for tick_interval in timeframes:
2018-04-22 07:56:49 +00:00
print(f'downloading pair {pair}, interval {tick_interval}')
data = exchange.get_ticker_history(pair, tick_interval, since_ms=since_time)
if not data:
print('\tNo data was downloaded')
break
2018-04-27 21:30:42 +00:00
print('\tData was downloaded for period %s - %s' % (
arrow.get(data[0][0] / 1000).format(),
arrow.get(data[-1][0] / 1000).format()))
# save data
2018-04-22 07:56:49 +00:00
pair_print = pair.replace('/', '_')
filename = f'{pair_print}-{tick_interval}.json'
misc.file_dump_json(os.path.join(dl_path, filename), data)