# source: https://raw.githubusercontent.com/jafarlihi/backstrat/b2c27ed3dae2af2c616bc5721d0e26a0f7713aa1/freqtrade-strats/JustROCR5.py
from freqtrade.strategy.interface import IStrategy
from pandas import DataFrame
import talib.abstract as ta


class github_jafarlihi_backstrat__JustROCR5__20210110_165815(IStrategy):
    minimal_roi = {
        "0": 0.05
    }

    stoploss = -0.01
    trailing_stop = True
    timeframe = '1m'

    def populate_indicators(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        dataframe['rocr'] = ta.ROCR(dataframe, timeperiod=5)
        dataframe['rocr_2'] = ta.ROCR(dataframe, timeperiod=2)

        return dataframe

    def populate_buy_trend(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        dataframe.loc[
            (
                (dataframe['rocr'] > 1.10) &
                (dataframe['rocr_2'] > 1.01)
            ),
            'buy'] = 1
        return dataframe

    def populate_sell_trend(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        dataframe.loc[
            (
            ),
            'sell'] = 1
        return dataframe