# source: https://raw.githubusercontent.com/kirillinda/freqtrade_strat/d387a6acd76377606a2a86b5ad907234a79282b4/new_strat.py
import numpy as np  # noqa
import pandas as pd  # noqa
from pandas import DataFrame
from technical.indicators import PMAX, zema
from typing import Dict, List
from functools import reduce
from freqtrade.strategy.interface import IStrategy

# --------------------------------
# Add your lib to import here
import talib.abstract as ta
import freqtrade.vendor.qtpylib.indicators as qtpylib


class github_kirillinda_freqtrade_strat__new_strat__20211012_212250(IStrategy):
    stoploss = -1
    timeframe = '5m'

    def populate_indicators(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        length = 10
        MAtype = 9
        src_val = 2
        multiplier = 3

        dataframe['ZLEMA'] = zema(dataframe, period=length)
        dataframe = PMAX(dataframe, period=length, multiplier=multiplier, length=length, MAtype=MAtype, src=src_val)
        #print(dataframe.keys)
        return dataframe
    
    
    def populate_buy_trend(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        dataframe.loc[
            (
                (qtpylib.crossed_above(dataframe['ZLEMA'], dataframe['pm_40_3_40_9']))
            ),
            'buy'] = 1
    
        return dataframe
    
    
    def populate_sell_trend(self, dataframe: DataFrame, metadata: dict) -> DataFrame:
        dataframe.loc[
            (
                (qtpylib.crossed_below(dataframe['ZLEMA'], dataframe['pm_40_3_40_9']))
            ),
            'sell'] = 1
    
        return dataframe