# source: https://raw.githubusercontent.com/meesvw/freqtrade-strategies/b2ec4c9d3076daabd1e24408a5aab55aa776b34a/strategies/rsimacdadx.py
# --- Do not remove these libs ---
from freqtrade.strategy.interface import IStrategy
from pandas import DataFrame
import talib.abstract as ta
import freqtrade.vendor.qtpylib.indicators as qtpylib


# --------------------------------


class github_meesvw_freqtrade_strategies__rsimacdadx__20220225_165527(IStrategy):
    minimal_roi = {
        "60":  0.01,
        "30":  0.03,
        "20":  0.04,
        "0":  0.05
    }

    stoploss = -0.10
    timeframe = '5m'

    def populate_indicators(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        macd = ta.MACD(dataframe)
        dataframe['adx'] = ta.ADX(dataframe, timeperiod=14)
        dataframe['rsi'] = ta.RSI(dataframe, timeperiod=16)
        dataframe['macd'] = macd['macd']
        dataframe['macdsignal'] = macd['macdsignal']
        return dataframe

    def populate_buy_trend(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        dataframe.loc[
            (
                    (dataframe['rsi'] < 28) &
                    (dataframe['macd'] < dataframe['macdsignal']) &
                    (dataframe['adx'] > 40)
            ),
            'buy'] = 1
        return dataframe

    def populate_sell_trend(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        dataframe.loc[
            (
                    (dataframe['rsi'] > 70) &
                    (dataframe['macd'] > dataframe['macdsignal']) &
                    (dataframe['adx'] > 40)

            ),
            'sell'] = 1
        return dataframe