import threading import logging # import json from flask import Flask, request, jsonify # from flask_restful import Resource, Api from json import dumps from freqtrade.rpc.rpc import RPC, RPCException from ipaddress import IPv4Address logger = logging.getLogger(__name__) app = Flask(__name__) class ApiServerSuperWrap(RPC): """ This class is for REST calls across api server """ def __init__(self, freqtrade) -> None: """ Init the api server, and init the super class RPC :param freqtrade: Instance of a freqtrade bot :return: None """ super().__init__(freqtrade) self._config = freqtrade.config # Register application handling self.register_rest_other() self.register_rest_rpc_urls() thread = threading.Thread(target=self.run, daemon=True) thread.start() def register_rest_other(self): # Added as a placeholder for app URLs that are not implemented in rpc.rpc app.register_error_handler(404, self.page_not_found) app.add_url_rule('/', 'hello', view_func=self.hello, methods=['GET']) def register_rest_rpc_urls(self): # register the url rules available on the api server # This is where to register rest urls that make use of # rpc.rpc functions ''' First two arguments passed are /URL and 'Label' Label can be used as a shortcut when refactoring ''' app.add_url_rule('/stop', 'stop', view_func=self.stop, methods=['GET']) app.add_url_rule('/start', 'start', view_func=self.start, methods=['GET']) app.add_url_rule('/daily', 'daily', view_func=self.daily, methods=['GET']) def run(self): """ Method that runs forever """ """ Section to handle configuration and running of the Rest server also to check and warn if not bound to a loopback, warn on security risk. """ rest_ip = self._config['api_server']['listen_ip_address'] rest_port = self._config['api_server']['listen_port'] logger.info('Starting HTTP Server at {}:{}'.format(rest_ip, rest_port)) if not IPv4Address(rest_ip).is_loopback: logger.info("SECURITY WARNING - Local Rest Server listening to external connections") logger.info("SECURITY WARNING - This is insecure please set to your loopback," "e.g 127.0.0.1 in config.json") # Run the Server logger.info('Starting Local Rest Server') try: app.run(host=rest_ip, port=rest_port) except Exception: logger.exception("Api server failed to start, exception message is:") def cleanup(self) -> None: # TODO: implement me raise NotImplementedError @property def name(self) -> str: # TODO: implement me raise NotImplementedError def send_msg(self, msg: str) -> None: # TODO: implement me raise NotImplementedError """ Define the application methods here, called by app.add_url_rule each Telegram command should have a like local substitute """ def page_not_found(self, error): # return "404 not found", 404 return jsonify({'status': 'error', 'reason': '''There's no API call for %s''' % request.base_url, 'code': 404}), 404 def hello(self): # For simple rest server testing via browser # cmds = 'Try uri:/daily?timescale=7 /profit /balance /status # /status /table /performance /count, # /start /stop /help' rest_cmds = 'Commands implemented:
' \ '/daily?timescale=7' \ '
' \ '/stop' \ '
' \ '/start' return rest_cmds def daily(self): try: timescale = request.args.get('timescale') logger.info("LocalRPC - Daily Command Called") timescale = int(timescale) stats = self._rpc_daily_profit(timescale, self._config['stake_currency'], self._config['fiat_display_currency'] ) stats = dumps(stats, indent=4, sort_keys=True, default=str) return stats except RPCException as e: return e def start(self): """ Handler for /start. Starts TradeThread """ msg = self._rpc_start() return jsonify(msg) def stop(self): """ Handler for /stop. Stops TradeThread """ msg = self._rpc_stop() return jsonify(msg)