Be explicit in what is used, db or trades

This commit is contained in:
Matthias
2019-06-22 16:18:49 +02:00
parent 8758218b09
commit 3e61ada34a
3 changed files with 17 additions and 7 deletions

View File

@@ -33,10 +33,10 @@ import pandas as pd
from freqtrade.arguments import Arguments, TimeRange
from freqtrade.data import history
from freqtrade.data.btanalysis import load_trades, extract_trades_of_period
from freqtrade.data.btanalysis import (extract_trades_of_period,
load_backtest_data, load_trades_from_db)
from freqtrade.optimize import setup_configuration
from freqtrade.plot.plotting import (generate_graph,
generate_plot_file)
from freqtrade.plot.plotting import generate_graph, generate_plot_file
from freqtrade.resolvers import ExchangeResolver, StrategyResolver
from freqtrade.state import RunMode
@@ -124,10 +124,12 @@ def analyse_and_plot_pairs(config: Dict[str, Any]):
tickers = {}
tickers[pair] = data
dataframe = generate_dataframe(strategy, tickers, pair)
trades = None
if config["trade_source"] == "DB":
trades = load_trades_from_db(config["db_url"])
elif config["trade_source"] == "file":
trades = load_backtest_data(Path(config["exportfilename"]))
trades = load_trades(exportfilename=config["exportfilename"],
db_url=config["db_url"],
)
trades = trades.loc[trades['pair'] == pair]
trades = extract_trades_of_period(dataframe, trades)