stable/scripts/plot_dataframe.py

113 lines
3.3 KiB
Python
Raw Normal View History

#!/usr/bin/env python3
"""
2019-01-25 05:42:29 +00:00
Script to display when the bot will buy on specific pair(s)
2019-06-29 18:07:25 +00:00
Use `python plot_dataframe.py --help` to display the command line arguments
Indicators recommended
Row 1: sma, ema3, ema5, ema10, ema50
Row 3: macd, rsi, fisher_rsi, mfi, slowd, slowk, fastd, fastk
Example of usage:
2019-01-26 09:56:29 +00:00
> python3 scripts/plot_dataframe.py --pairs BTC/EUR,XRP/BTC -d user_data/data/
--indicators1 sma,ema3 --indicators2 fastk,fastd
"""
import logging
import sys
2019-03-23 18:18:10 +00:00
from typing import Any, Dict, List
import pandas as pd
from freqtrade.arguments import ARGS_PLOT_DATAFRAME, Arguments
2019-06-30 07:42:10 +00:00
from freqtrade.data.btanalysis import extract_trades_of_period
2019-05-25 18:14:31 +00:00
from freqtrade.optimize import setup_configuration
2019-06-30 07:42:10 +00:00
from freqtrade.plot.plotting import (FTPlots, generate_candlestick_graph,
store_plot_file,
generate_plot_filename)
2019-05-25 18:14:31 +00:00
from freqtrade.state import RunMode
logger = logging.getLogger(__name__)
2019-01-25 05:42:29 +00:00
def generate_dataframe(strategy, tickers, pair) -> pd.DataFrame:
"""
Get tickers then Populate strategy indicators and signals, then return the full dataframe
:return: the DataFrame of a pair
"""
dataframes = strategy.tickerdata_to_dataframe(tickers)
dataframe = dataframes[pair]
dataframe = strategy.advise_buy(dataframe, {'pair': pair})
dataframe = strategy.advise_sell(dataframe, {'pair': pair})
2019-01-25 05:42:29 +00:00
return dataframe
2019-06-16 11:41:36 +00:00
def analyse_and_plot_pairs(config: Dict[str, Any]):
2019-01-25 05:42:29 +00:00
"""
From arguments provided in cli:
-Initialise backtest env
-Get tickers data
-Generate Dafaframes populated with indicators and signals
-Load trades excecuted on same periods
-Generate Plotly plot objects
-Generate plot files
:return: None
"""
2019-06-30 07:42:10 +00:00
plot = FTPlots(config)
2019-06-29 18:41:22 +00:00
2019-01-25 05:42:29 +00:00
pair_counter = 0
2019-06-30 07:42:10 +00:00
for pair, data in plot.tickers.items():
2019-01-25 05:42:29 +00:00
pair_counter += 1
logger.info("analyse pair %s", pair)
tickers = {}
tickers[pair] = data
2019-06-30 07:42:10 +00:00
dataframe = generate_dataframe(plot.strategy, tickers, pair)
2019-01-25 05:42:29 +00:00
2019-06-30 07:42:10 +00:00
trades_pair = plot.trades.loc[plot.trades['pair'] == pair]
2019-06-29 18:41:22 +00:00
trades_pair = extract_trades_of_period(dataframe, trades_pair)
2019-01-25 05:42:29 +00:00
fig = generate_candlestick_graph(
2019-01-25 05:42:29 +00:00
pair=pair,
data=dataframe,
2019-06-29 18:41:22 +00:00
trades=trades_pair,
2019-06-16 11:41:36 +00:00
indicators1=config["indicators1"].split(","),
indicators2=config["indicators2"].split(",")
2019-01-25 05:42:29 +00:00
)
store_plot_file(fig, generate_plot_filename(pair, config['ticker_interval']))
2019-01-25 05:42:29 +00:00
logger.info('End of ploting process %s plots generated', pair_counter)
2019-06-16 17:53:48 +00:00
def plot_parse_args(args: List[str]) -> Dict[str, Any]:
2019-05-28 18:02:17 +00:00
"""
Parse args passed to the script
:param args: Cli arguments
:return: args: Array with all arguments
"""
arguments = Arguments(args, 'Graph dataframe')
arguments.build_args(optionlist=ARGS_PLOT_DATAFRAME)
2019-06-16 11:41:36 +00:00
parsed_args = arguments.parse_args()
# Load the configuration
2019-06-28 04:47:40 +00:00
config = setup_configuration(parsed_args, RunMode.OTHER)
2019-06-16 11:41:36 +00:00
return config
2019-05-28 18:02:17 +00:00
2019-05-30 18:26:46 +00:00
def main(sysargv: List[str]) -> None:
"""
This function will initiate the bot and start the trading loop.
:return: None
"""
logger.info('Starting Plot Dataframe')
2019-01-25 05:42:29 +00:00
analyse_and_plot_pairs(
plot_parse_args(sysargv)
)
2019-01-25 05:42:29 +00:00
exit()
if __name__ == '__main__':
main(sys.argv[1:])