111 lines
3.7 KiB
Python
111 lines
3.7 KiB
Python
import logging
|
|
import sys
|
|
from argparse import Namespace
|
|
from pathlib import Path
|
|
from typing import Any, Dict
|
|
|
|
import arrow
|
|
|
|
from freqtrade.configuration import Configuration, TimeRange
|
|
from freqtrade.configuration.directory_operations import create_userdata_dir
|
|
from freqtrade.data.history import download_pair_history
|
|
from freqtrade.exchange import available_exchanges
|
|
from freqtrade.resolvers import ExchangeResolver
|
|
from freqtrade.state import RunMode
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def setup_utils_configuration(args: Namespace, method: RunMode) -> Dict[str, Any]:
|
|
"""
|
|
Prepare the configuration for utils subcommands
|
|
:param args: Cli args from Arguments()
|
|
:return: Configuration
|
|
"""
|
|
configuration = Configuration(args, method)
|
|
config = configuration.get_config()
|
|
|
|
config['exchange']['dry_run'] = True
|
|
# Ensure we do not use Exchange credentials
|
|
config['exchange']['key'] = ''
|
|
config['exchange']['secret'] = ''
|
|
|
|
return config
|
|
|
|
|
|
def start_list_exchanges(args: Namespace) -> None:
|
|
"""
|
|
Print available exchanges
|
|
:param args: Cli args from Arguments()
|
|
:return: None
|
|
"""
|
|
|
|
if args.print_one_column:
|
|
print('\n'.join(available_exchanges()))
|
|
else:
|
|
print(f"Exchanges supported by ccxt and available for Freqtrade: "
|
|
f"{', '.join(available_exchanges())}")
|
|
|
|
|
|
def start_create_userdir(args: Namespace) -> None:
|
|
"""
|
|
Create "user_data" directory to contain user data strategies, hyperopts, ...)
|
|
:param args: Cli args from Arguments()
|
|
:return: None
|
|
"""
|
|
if "user_data_dir" in args and args.user_data_dir:
|
|
create_userdata_dir(args.user_data_dir, create_dir=True)
|
|
else:
|
|
logger.warning("`create-userdir` requires --userdir to be set.")
|
|
sys.exit(1)
|
|
|
|
|
|
def start_download_data(args: Namespace) -> None:
|
|
"""
|
|
Download data (former download_backtest_data.py script)
|
|
"""
|
|
config = setup_utils_configuration(args, RunMode.OTHER)
|
|
|
|
timerange = TimeRange()
|
|
if 'days' in config:
|
|
time_since = arrow.utcnow().shift(days=-config['days']).strftime("%Y%m%d")
|
|
timerange = TimeRange.parse_timerange(f'{time_since}-')
|
|
|
|
dl_path = Path(config['datadir'])
|
|
logger.info(f'About to download pairs: {config["pairs"]}, '
|
|
f'intervals: {config["timeframes"]} to {dl_path}')
|
|
|
|
pairs_not_available = []
|
|
|
|
try:
|
|
# Init exchange
|
|
exchange = ExchangeResolver(config['exchange']['name'], config).exchange
|
|
|
|
for pair in config["pairs"]:
|
|
if pair not in exchange.markets:
|
|
pairs_not_available.append(pair)
|
|
logger.info(f"Skipping pair {pair}...")
|
|
continue
|
|
for ticker_interval in config["timeframes"]:
|
|
pair_print = pair.replace('/', '_')
|
|
filename = f'{pair_print}-{ticker_interval}.json'
|
|
dl_file = dl_path.joinpath(filename)
|
|
if config.get("erase") and dl_file.exists():
|
|
logger.info(
|
|
f'Deleting existing data for pair {pair}, interval {ticker_interval}.')
|
|
dl_file.unlink()
|
|
|
|
logger.info(f'Downloading pair {pair}, interval {ticker_interval}.')
|
|
download_pair_history(datadir=dl_path, exchange=exchange,
|
|
pair=pair, ticker_interval=str(ticker_interval),
|
|
timerange=timerange)
|
|
|
|
except KeyboardInterrupt:
|
|
sys.exit("SIGINT received, aborting ...")
|
|
|
|
finally:
|
|
if pairs_not_available:
|
|
logger.info(
|
|
f"Pairs [{','.join(pairs_not_available)}] not available "
|
|
f"on exchange {config['exchange']['name']}.")
|