import freqtrade.vendor.qtpylib.indicators as qtpylib
import numpy as np
import talib.abstract as ta
from freqtrade.strategy.interface import IStrategy
from pandas import DataFrame, DatetimeIndex, merge, Series


class YOLO(IStrategy):

    buy_params = {
		'adx': 34, 
		'aroon-down': 33, 
		'aroon-up': 98
    }

    sell_params = {

    }

    minimal_roi = {
        "0": 0.03,
        "7": 0.02,
        "33": 0.01,
        "71": 0.005
    }

    stoploss = -0.01

    trailing_stop = True
    trailing_stop_positive = 0.03289
    trailing_stop_positive_offset = 0.05723
    trailing_only_offset_is_reached = False

    """
    END HYPEROPT
    """
    
    timeframe = '1m'

    use_sell_signal = False
    sell_profit_only = False
    ignore_roi_if_buy_signal = True


    def populate_indicators(self, dataframe: DataFrame, metadata: dict) -> DataFrame:

        dataframe['adx'] = ta.ADX(dataframe, timeperiod=90) #90
        aroon = ta.AROON(dataframe, timeperiod=60) #60

        dataframe['aroon-down'] = aroon['aroondown'] 
        dataframe['aroon-up'] = aroon['aroonup']

        return dataframe

    def populate_buy_trend(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        params = self.buy_params

        dataframe.loc[
            ( 
                (dataframe['adx'] > params['adx']) &
                (dataframe['aroon-up'] > params['aroon-up']) &
                (dataframe['aroon-down'] < params['aroon-down']) &
                (dataframe['volume'] > 0)
            ),
            'buy'
        ] = 1

        return dataframe

    def populate_sell_trend(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        """
        no sell signal
        """
        dataframe['sell'] = 0
        return dataframe