import talib
import numpy as np
import bn
import tg
import datetime
import settings


def check_ma_arrange(data):
     # 提取收盘价
    close_prices = np.array([float(entry[4]) for entry in data])
    
    # 计算移动平均线
    ema7 = talib.EMA(close_prices, timeperiod=7)
    ema30 = talib.EMA(close_prices, timeperiod=30)
    ema100 = talib.EMA(close_prices, timeperiod=100)
    ema200 = talib.EMA(close_prices, timeperiod=200)

    bullish = ema7[-1] > ema30[-1] > ema100[-1] > ema200[-1]
    bearish = ema7[-1] < ema30[-1] < ema100[-1] < ema200[-1]

    return bullish, bearish
    

def strategy_run(symbol, interval):
    # 获取kline数据
    data = bn.klines(symbol, interval)

    bullish, bearish = check_ma_arrange(data)

    text=""
    if bullish:
        text = f'📶信号预警📶\r\n\r\n品种：【${symbol}】\r\n周期：{interval}\r\n信号：【多头】排列\r\n\r\n{datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")}'
    if bearish:
        text = f'📶信号预警📶\r\n\r\n品种：【${symbol}】\r\n周期：{interval}\r\n信号：【空头】排列\r\n\r\n{datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")}'
    
    if text != "":
        print(text)
        tg.send_message(settings.chat_id, text)