diff --git a/scripts/plot_dataframe.py b/scripts/plot_dataframe.py index 0d193726d..44961891c 100755 --- a/scripts/plot_dataframe.py +++ b/scripts/plot_dataframe.py @@ -1,17 +1,33 @@ #!/usr/bin/env python3 +import sys +import argparse import matplotlib # Install PYQT5 manually if you want to test this helper function matplotlib.use("Qt5Agg") import matplotlib.pyplot as plt from freqtrade import exchange, analyze +from freqtrade.misc import parse_args_common -def plot_analyzed_dataframe(pair: str) -> None: +def plot_parse_args(args ): + parser = parse_args_common(args, 'Graph utility') + parser.add_argument( + '-p', '--pair', + help = 'What currency pair', + dest = 'pair', + default = 'BTC_ETH', + type = str, + ) + return parser.parse_args(args) + + +def plot_analyzed_dataframe(args) -> None: """ Calls analyze() and plots the returned dataframe :param pair: pair as str :return: None """ + pair = args.pair # Init Bittrex to use public API exchange._API = exchange.Bittrex({'key': '', 'secret': ''}) @@ -50,4 +66,5 @@ def plot_analyzed_dataframe(pair: str) -> None: if __name__ == '__main__': - plot_analyzed_dataframe('BTC_ETH') + args = plot_parse_args(sys.argv[1:]) + plot_analyzed_dataframe(args)