stable/freqtrade/utils.py

288 lines
11 KiB
Python
Raw Normal View History

2019-11-01 18:50:42 +00:00
import csv
2019-06-12 09:33:20 +00:00
import logging
import sys
from collections import OrderedDict
from pathlib import Path
from typing import Any, Dict, List
2019-06-12 09:33:20 +00:00
import arrow
import rapidjson
from tabulate import tabulate
from freqtrade import OperationalException
2019-11-01 18:50:42 +00:00
from freqtrade.configuration import (Configuration, TimeRange,
remove_credentials)
from freqtrade.configuration.directory_operations import (copy_sample_files,
create_userdata_dir)
from freqtrade.data.history import (convert_trades_to_ohlcv,
refresh_backtest_ohlcv_data,
2019-08-27 05:14:37 +00:00
refresh_backtest_trades_data)
2019-11-01 18:50:42 +00:00
from freqtrade.exchange import (available_exchanges, ccxt_exchanges,
market_is_active, symbol_is_pair)
from freqtrade.misc import plural, render_template
from freqtrade.resolvers import ExchangeResolver
2019-06-12 09:33:20 +00:00
from freqtrade.state import RunMode
logger = logging.getLogger(__name__)
def setup_utils_configuration(args: Dict[str, Any], method: RunMode) -> Dict[str, Any]:
2019-06-12 09:33:20 +00:00
"""
2019-06-16 18:37:43 +00:00
Prepare the configuration for utils subcommands
2019-06-12 09:33:20 +00:00
:param args: Cli args from Arguments()
:return: Configuration
"""
configuration = Configuration(args, method)
config = configuration.get_config()
2019-06-12 09:33:20 +00:00
# Ensure we do not use Exchange credentials
remove_credentials(config)
2019-06-12 09:33:20 +00:00
return config
def start_trading(args: Dict[str, Any]) -> int:
"""
Main entry point for trading mode
"""
from freqtrade.worker import Worker
# Load and run worker
2019-11-16 08:47:35 +00:00
worker = None
2019-11-15 19:10:17 +00:00
try:
worker = Worker(args)
worker.run()
except KeyboardInterrupt:
logger.info('SIGINT received, aborting ...')
finally:
if worker:
logger.info("worker found ... calling exit")
worker.exit()
return 0
def start_list_exchanges(args: Dict[str, Any]) -> None:
2019-06-12 09:33:20 +00:00
"""
Print available exchanges
2019-06-12 09:33:20 +00:00
:param args: Cli args from Arguments()
:return: None
"""
2019-09-30 21:33:33 +00:00
exchanges = ccxt_exchanges() if args['list_exchanges_all'] else available_exchanges()
if args['print_one_column']:
2019-09-30 21:33:33 +00:00
print('\n'.join(exchanges))
2019-06-12 09:33:20 +00:00
else:
2019-09-30 21:33:33 +00:00
if args['list_exchanges_all']:
print(f"All exchanges supported by the ccxt library: {', '.join(exchanges)}")
else:
print(f"Exchanges available for Freqtrade: {', '.join(exchanges)}")
def start_create_userdir(args: Dict[str, Any]) -> None:
"""
2019-11-13 08:38:06 +00:00
Create "user_data" directory to contain user data strategies, hyperopt, ...)
:param args: Cli args from Arguments()
:return: None
"""
if "user_data_dir" in args and args["user_data_dir"]:
2019-11-01 12:28:35 +00:00
userdir = create_userdata_dir(args["user_data_dir"], create_dir=True)
2019-11-01 13:08:55 +00:00
copy_sample_files(userdir, overwrite=args["reset"])
else:
logger.warning("`create-userdir` requires --userdir to be set.")
sys.exit(1)
def start_new_strategy(args: Dict[str, Any]) -> None:
config = setup_utils_configuration(args, RunMode.UTIL_NO_EXCHANGE)
if "strategy" in args and args["strategy"]:
2019-11-15 05:49:58 +00:00
if args["strategy"] == "DefaultStrategy":
2019-11-01 18:50:42 +00:00
raise OperationalException("DefaultStrategy is not allowed as name.")
new_path = config['user_data_dir'] / "strategies" / (args["strategy"] + ".py")
if new_path.exists():
raise OperationalException(f"`{new_path}` already exists. "
"Please choose another Strategy Name.")
strategy_text = render_template(template='base_strategy.py.j2',
arguments={"strategy": args["strategy"]})
logger.info(f"Writing strategy to `{new_path}`.")
new_path.write_text(strategy_text)
else:
2019-11-12 12:33:37 +00:00
raise OperationalException("`new-strategy` requires --strategy to be set.")
2019-11-02 09:42:17 +00:00
def start_new_hyperopt(args: Dict[str, Any]) -> None:
config = setup_utils_configuration(args, RunMode.UTIL_NO_EXCHANGE)
if "hyperopt" in args and args["hyperopt"]:
2019-11-15 05:49:58 +00:00
if args["hyperopt"] == "DefaultHyperopt":
2019-11-16 14:52:32 +00:00
raise OperationalException("DefaultHyperopt is not allowed as name.")
2019-11-02 09:42:17 +00:00
new_path = config['user_data_dir'] / "hyperopts" / (args["hyperopt"] + ".py")
if new_path.exists():
raise OperationalException(f"`{new_path}` already exists. "
"Please choose another Strategy Name.")
strategy_text = render_template(template='base_hyperopt.py.j2',
arguments={"hyperopt": args["hyperopt"]})
logger.info(f"Writing hyperopt to `{new_path}`.")
new_path.write_text(strategy_text)
else:
2019-11-12 12:33:37 +00:00
raise OperationalException("`new-hyperopt` requires --hyperopt to be set.")
2019-11-02 09:42:17 +00:00
def start_download_data(args: Dict[str, Any]) -> None:
"""
Download data (former download_backtest_data.py script)
"""
2019-11-01 14:39:25 +00:00
config = setup_utils_configuration(args, RunMode.UTIL_EXCHANGE)
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}-')
if 'pairs' not in config:
raise OperationalException(
2019-09-24 04:35:41 +00:00
"Downloading data requires a list of pairs. "
"Please check the documentation on how to configure this.")
dl_path = Path(config['datadir'])
logger.info(f'About to download pairs: {config["pairs"]}, '
f'intervals: {config["timeframes"]} to {dl_path}')
pairs_not_available: List[str] = []
# Init exchange
exchange = ExchangeResolver(config['exchange']['name'], config).exchange
try:
2019-10-08 18:31:01 +00:00
if config.get('download_trades'):
2019-08-27 05:14:37 +00:00
pairs_not_available = refresh_backtest_trades_data(
exchange, pairs=config["pairs"], datadir=Path(config['datadir']),
timerange=timerange, erase=config.get("erase"))
# Convert downloaded trade data to different timeframes
convert_trades_to_ohlcv(
2019-10-14 04:19:59 +00:00
pairs=config["pairs"], timeframes=config["timeframes"],
datadir=Path(config['datadir']), timerange=timerange, erase=config.get("erase"))
2019-10-08 18:31:01 +00:00
else:
pairs_not_available = refresh_backtest_ohlcv_data(
exchange, pairs=config["pairs"], timeframes=config["timeframes"],
dl_path=Path(config['datadir']), timerange=timerange, erase=config.get("erase"))
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 {exchange.name}.")
2019-09-29 08:54:20 +00:00
def start_list_timeframes(args: Dict[str, Any]) -> None:
"""
Print ticker intervals (timeframes) available on Exchange
"""
2019-11-01 14:39:25 +00:00
config = setup_utils_configuration(args, RunMode.UTIL_EXCHANGE)
# Do not use ticker_interval set in the config
config['ticker_interval'] = None
2019-09-29 08:54:20 +00:00
# Init exchange
2019-10-13 08:33:22 +00:00
exchange = ExchangeResolver(config['exchange']['name'], config, validate=False).exchange
2019-09-29 08:54:20 +00:00
if args['print_one_column']:
print('\n'.join(exchange.timeframes))
2019-09-29 08:54:20 +00:00
else:
print(f"Timeframes available for the exchange `{exchange.name}`: "
f"{', '.join(exchange.timeframes)}")
def start_list_markets(args: Dict[str, Any], pairs_only: bool = False) -> None:
"""
Print pairs/markets on the exchange
:param args: Cli args from Arguments()
:param pairs_only: if True print only pairs, otherwise print all instruments (markets)
:return: None
"""
2019-11-01 14:39:25 +00:00
config = setup_utils_configuration(args, RunMode.UTIL_EXCHANGE)
# Init exchange
exchange = ExchangeResolver(config['exchange']['name'], config, validate=False).exchange
# By default only active pairs/markets are to be shown
active_only = not args.get('list_pairs_all', False)
base_currencies = args.get('base_currencies', [])
quote_currencies = args.get('quote_currencies', [])
try:
pairs = exchange.get_markets(base_currencies=base_currencies,
quote_currencies=quote_currencies,
pairs_only=pairs_only,
active_only=active_only)
# Sort the pairs/markets by symbol
pairs = OrderedDict(sorted(pairs.items()))
except Exception as e:
raise OperationalException(f"Cannot get markets. Reason: {e}") from e
else:
2019-10-15 23:22:27 +00:00
summary_str = ((f"Exchange {exchange.name} has {len(pairs)} ") +
("active " if active_only else "") +
(plural(len(pairs), "pair" if pairs_only else "market")) +
(f" with {', '.join(base_currencies)} as base "
f"{plural(len(base_currencies), 'currency', 'currencies')}"
if base_currencies else "") +
(" and" if base_currencies and quote_currencies else "") +
(f" with {', '.join(quote_currencies)} as quote "
f"{plural(len(quote_currencies), 'currency', 'currencies')}"
if quote_currencies else ""))
2019-10-15 23:22:27 +00:00
headers = ["Id", "Symbol", "Base", "Quote", "Active",
*(['Is pair'] if not pairs_only else [])]
tabular_data = []
for _, v in pairs.items():
tabular_data.append({'Id': v['id'], 'Symbol': v['symbol'],
'Base': v['base'], 'Quote': v['quote'],
'Active': market_is_active(v),
**({'Is pair': symbol_is_pair(v['symbol'])}
if not pairs_only else {})})
2019-10-15 23:22:27 +00:00
if (args.get('print_one_column', False) or
args.get('list_pairs_print_json', False) or
args.get('print_csv', False)):
# Print summary string in the log in case of machine-readable
# regular formats.
logger.info(f"{summary_str}.")
else:
# Print empty string separating leading logs and output in case of
# human-readable formats.
print()
2019-10-17 01:34:05 +00:00
if len(pairs):
if args.get('print_list', False):
# print data as a list, with human-readable summary
print(f"{summary_str}: {', '.join(pairs.keys())}.")
elif args.get('print_one_column', False):
print('\n'.join(pairs.keys()))
2019-10-17 01:34:05 +00:00
elif args.get('list_pairs_print_json', False):
print(rapidjson.dumps(list(pairs.keys()), default=str))
elif args.get('print_csv', False):
writer = csv.DictWriter(sys.stdout, fieldnames=headers)
2019-10-15 23:22:27 +00:00
writer.writeheader()
writer.writerows(tabular_data)
2019-10-17 01:34:05 +00:00
else:
# print data as a table, with the human-readable summary
print(f"{summary_str}:")
print(tabulate(tabular_data, headers='keys', tablefmt='pipe'))
2019-10-17 01:34:05 +00:00
elif not (args.get('print_one_column', False) or
args.get('list_pairs_print_json', False) or
args.get('print_csv', False)):
2019-10-23 20:57:17 +00:00
print(f"{summary_str}.")