stable/freqtrade/data/dataprovider.py

86 lines
2.6 KiB
Python
Raw Normal View History

2018-11-30 19:42:16 +00:00
"""
Dataprovider
Responsible to provide data to the bot
including Klines, tickers, historic data
Common Interface for bot and strategy to access data.
"""
import logging
2018-12-17 05:52:13 +00:00
from pathlib import Path
from typing import List
2018-11-30 19:42:16 +00:00
2018-12-02 08:16:35 +00:00
from pandas import DataFrame
2018-12-17 05:52:13 +00:00
from freqtrade.data.history import load_pair_history
from freqtrade.exchange import Exchange
from freqtrade.state import RunMode
2018-11-30 19:42:16 +00:00
logger = logging.getLogger(__name__)
class DataProvider(object):
2018-12-02 08:16:35 +00:00
def __init__(self, config: dict, exchange: Exchange) -> None:
self._config = config
self._exchange = exchange
2018-11-30 19:42:16 +00:00
2018-12-02 14:57:49 +00:00
def refresh(self, pairlist: List[str]) -> None:
2018-11-30 19:42:16 +00:00
"""
Refresh data, called with each cycle
"""
2018-12-02 14:57:49 +00:00
self._exchange.refresh_tickers(pairlist, self._config['ticker_interval'])
2018-11-30 19:42:16 +00:00
2018-12-26 13:23:21 +00:00
@property
def available_pairs(self) -> List[str]:
"""
Return a list of pairs for which data is currently cached.
Should be whitelist + open trades.
"""
return list(self._exchange._klines.keys())
2018-12-29 12:00:50 +00:00
def ohlcv(self, pair: str, copy: bool = True) -> DataFrame:
2018-11-30 19:42:16 +00:00
"""
2018-12-02 08:16:35 +00:00
get ohlcv data for the given pair as DataFrame
2018-12-29 07:47:14 +00:00
Please check `available_pairs` to verify which pairs are currently cached.
2018-12-25 12:37:15 +00:00
:param pair: pair to get the data for
:param copy: copy dataframe before returning.
Use false only for RO operations (where the dataframe is not modified)
2018-11-30 19:42:16 +00:00
"""
2018-12-02 08:16:35 +00:00
# TODO: Should not be stored in exchange but in this class
if self.runmode in (RunMode.DRY_RUN, RunMode.LIVE):
return self._exchange.klines(pair, copy)
else:
2018-12-29 12:00:50 +00:00
return DataFrame()
2018-11-30 19:42:16 +00:00
2018-12-17 05:52:13 +00:00
def historic_ohlcv(self, pair: str, ticker_interval: str) -> DataFrame:
2018-11-30 19:42:16 +00:00
"""
get historic ohlcv data stored for backtesting
"""
2018-12-17 05:52:13 +00:00
return load_pair_history(pair=pair,
ticker_interval=ticker_interval,
refresh_pairs=False,
2018-12-25 12:20:25 +00:00
datadir=Path(self._config['datadir']) if self._config.get(
2018-12-17 05:52:13 +00:00
'datadir') else None
)
2018-11-30 19:42:16 +00:00
2018-12-02 08:16:35 +00:00
def ticker(self, pair: str):
"""
Return last ticker data
"""
2018-12-26 13:58:16 +00:00
# TODO: Implement me
2018-11-30 19:42:16 +00:00
pass
2018-12-02 08:16:35 +00:00
def orderbook(self, pair: str, max: int):
"""
return latest orderbook data
"""
2018-12-26 13:58:16 +00:00
# TODO: Implement me
2018-11-30 19:42:16 +00:00
pass
2018-12-02 20:57:30 +00:00
@property
def runmode(self) -> RunMode:
2018-12-02 20:57:30 +00:00
"""
Get runmode of the bot
can be "live", "dry-run", "backtest", "edgecli", "hyperopt" or "other".
2018-12-02 20:57:30 +00:00
"""
2018-12-25 13:35:48 +00:00
return RunMode(self._config.get('runmode', RunMode.OTHER))