stable/scripts/plot_profit.py

66 lines
2.0 KiB
Python
Raw Normal View History

2018-01-12 09:55:49 +00:00
#!/usr/bin/env python3
"""
Script to display profits
2019-06-29 18:07:25 +00:00
Use `python plot_profit.py --help` to display the command line arguments
"""
2018-03-25 19:37:14 +00:00
import logging
2018-01-12 09:55:49 +00:00
import sys
2019-06-30 07:28:49 +00:00
from typing import Any, Dict, List
2018-01-12 09:55:49 +00:00
from freqtrade.configuration import Arguments
from freqtrade.configuration.arguments import ARGS_PLOT_PROFIT
2019-06-30 07:28:49 +00:00
from freqtrade.optimize import setup_configuration
from freqtrade.plot.plotting import init_plotscript, generate_profit_graph, store_plot_file
from freqtrade.state import RunMode
2018-03-25 19:37:14 +00:00
logger = logging.getLogger(__name__)
2018-01-12 09:55:49 +00:00
2019-06-30 07:28:49 +00:00
def plot_profit(config: Dict[str, Any]) -> None:
2018-01-12 09:55:49 +00:00
"""
Plots the total profit for all pairs.
Note, the profit calculation isn't realistic.
But should be somewhat proportional, and therefor useful
in helping out to find a good algorithm.
"""
plot_elements = init_plotscript(config)
trades = plot_elements['trades']
# Filter trades to relevant pairs
trades = trades[trades['pair'].isin(plot_elements["pairs"])]
2019-06-30 07:28:49 +00:00
# Create an average close price of all the pairs that were involved.
# this could be useful to gauge the overall market trend
fig = generate_profit_graph(plot_elements["pairs"], plot_elements["tickers"], trades)
2019-06-30 08:47:55 +00:00
store_plot_file(fig, filename='freqtrade-profit-plot.html', auto_open=True)
2018-03-05 04:24:01 +00:00
2019-06-30 07:28:49 +00:00
def plot_parse_args(args: List[str]) -> Dict[str, Any]:
"""
Parse args passed to the script
:param args: Cli arguments
:return: args: Array with all arguments
"""
arguments = Arguments(args, 'Graph profits')
arguments._build_args(optionlist=ARGS_PLOT_PROFIT)
parsed_args = arguments._parse_args()
2019-06-30 07:28:49 +00:00
# Load the configuration
config = setup_configuration(parsed_args, RunMode.OTHER)
return config
def main(sysargv: List[str]) -> None:
"""
This function will initiate the bot and start the trading loop.
:return: None
"""
logger.info('Starting Plot Dataframe')
plot_profit(
plot_parse_args(sysargv)
)
2018-01-12 09:55:49 +00:00
if __name__ == '__main__':
main(sys.argv[1:])