plot script can take arguments

This commit is contained in:
kryofly 2018-01-06 08:38:47 +01:00
parent e4500af736
commit e6e57e47cf

View File

@ -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)