adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add /strategies endpoint

Browse Source
This commit is contained in:
Matthias 2020-07-31 07:31:20 +02:00
parent f227f6a755
commit 18bbfdd341
2 changed files with 26 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
freqtrade/rpc/api_server.py
View File

@ -1,8 +1,9 @@
from copy import deepcopy
import logging import logging
import threading import threading
from copy import deepcopy
from datetime import date, datetime from datetime import date, datetime
from ipaddress import IPv4Address from ipaddress import IPv4Address
from pathlib import Path
from typing import Any, Callable, Dict from typing import Any, Callable, Dict
from arrow import Arrow from arrow import Arrow
@ -16,7 +17,7 @@ from werkzeug.security import safe_str_cmp
from werkzeug.serving import make_server from werkzeug.serving import make_server
from freqtrade.__init__ import __version__ from freqtrade.__init__ import __version__
from freqtrade.constants import DATETIME_PRINT_FORMAT from freqtrade.constants import DATETIME_PRINT_FORMAT, USERPATH_STRATEGIES
from freqtrade.persistence import Trade from freqtrade.persistence import Trade
from freqtrade.rpc.fiat_convert import CryptoToFiatConverter from freqtrade.rpc.fiat_convert import CryptoToFiatConverter
from freqtrade.rpc.rpc import RPC, RPCException from freqtrade.rpc.rpc import RPC, RPCException
@ -219,6 +220,9 @@ class ApiServer(RPC):
view_func=self._analysed_history, methods=['GET']) view_func=self._analysed_history, methods=['GET'])
self.app.add_url_rule(f'{BASE_URI}/plot_config', 'plot_config', self.app.add_url_rule(f'{BASE_URI}/plot_config', 'plot_config',
view_func=self._plot_config, methods=['GET']) view_func=self._plot_config, methods=['GET'])
self.app.add_url_rule(f'{BASE_URI}/strategies', 'strategies',
view_func=self._list_strategies, methods=['GET'])
# Combined actions and infos # Combined actions and infos
self.app.add_url_rule(f'{BASE_URI}/blacklist', 'blacklist', view_func=self._blacklist, self.app.add_url_rule(f'{BASE_URI}/blacklist', 'blacklist', view_func=self._blacklist,
methods=['GET', 'POST']) methods=['GET', 'POST'])
@ -561,3 +565,14 @@ class ApiServer(RPC):
Handler for /plot_config. Handler for /plot_config.
""" """
return self.rest_dump(self._rpc_plot_config()) return self.rest_dump(self._rpc_plot_config())
@require_login
@rpc_catch_errors
def _list_strategies(self):
directory = Path(self._config.get(
'strategy_path', self._config['user_data_dir'] / USERPATH_STRATEGIES))
from freqtrade.resolvers.strategy_resolver import StrategyResolver
strategy_objs = StrategyResolver.search_all_objects(directory, False)
strategy_objs = sorted(strategy_objs, key=lambda x: x['name'])
return self.rest_dump([x['name'] for x in strategy_objs])

9
tests/rpc/test_rpc_apiserver.py
View File

@ -867,3 +867,12 @@ def test_api_plot_config(botclient):
assert_response(rc) assert_response(rc)
assert rc.json == ftbot.strategy.plot_config assert rc.json == ftbot.strategy.plot_config
assert isinstance(rc.json['main_plot'], dict) assert isinstance(rc.json['main_plot'], dict)
def test_api_strategies(botclient):
ftbot, client = botclient
rc = client_get(client, f"{BASE_URI}/strategies")
assert_response(rc)
assert rc.json == ['DefaultStrategy', 'TestStrategyLegacy']