import time
import hmac
import hashlib
import base64
import urllib.parse
import json
import requests
import pandas as pd


def get_klines(symbol, interval, limit=500):
    """獲取K線數(shù)據(jù)"""
    url = "https://fapi.binance.com/fapi/v1/klines"
    params = {
        'symbol': symbol,
        'interval': interval,
        'limit': limit,
    }
    response = requests.get(url, params=params)
    return response.json()


def calculate_macd(data, short_period=6, long_period=7, signal_period=4):
    """計(jì)算MACD指標(biāo)"""
    # 將獲取的數(shù)據(jù)轉(zhuǎn)換為DataFrame
    df = pd.DataFrame(data, columns=['OpenTime', 'Open', 'High', 'Low', 'Close', 'Volume', 'CloseTime',
                                     'QuoteAssetVolume', 'NumberOfTrades', 'TakerBuyBaseAssetVolume',
                                     'TakerBuyQuoteAssetVolume', 'Ignore'])

    # 將收盤(pán)價(jià)轉(zhuǎn)換為數(shù)值類(lèi)型
    df['Close'] = pd.to_numeric(df['Close'])

    # 計(jì)算快速EMA和慢速EMA
    df['EMA_fast'] = df['Close'].ewm(span=short_period, adjust=False).mean()
    df['EMA_slow'] = df['Close'].ewm(span=long_period, adjust=False).mean()

    # 計(jì)算DIF
    df['DIF'] = df['EMA_fast'] - df['EMA_slow']

    # 計(jì)算DEA
    df['DEA'] = df['DIF'].ewm(span=signal_period, adjust=False).mean()

    # 計(jì)算MACD
    df['MACD'] = (df['DIF'] - df['DEA']) * 2
    return df


def find_macd_crossovers(df):
    """找到MACD的金叉和死叉"""
    crossover_points = {'上穿': [], '下穿': []}
    for i in range(1, len(df)):
        if df['DIF'].iloc[i] > 0 and df['DIF'].iloc[i - 1] <= 0:
            crossover_points['上穿'].append(df['OpenTime'].iloc[i])
        elif df['DIF'].iloc[i] < 0 and df['DIF'].iloc[i - 1] >= 0:
            crossover_points['下穿'].append(df['OpenTime'].iloc[i])
    return crossover_points


def send_dingtalk_message(webhook, secret, message):
    # 獲取時(shí)間戳
    timestamp = str(round(time.time() * 1000))
    # 構(gòu)造簽名
    secret_enc = secret.encode('utf-8')
    string_to_sign = '{}\n{}'.format(timestamp, secret)
    string_to_sign_enc = string_to_sign.encode('utf-8')
    hmac_code = hmac.new(secret_enc, string_to_sign_enc, digestmod=hashlib.sha256).digest()
    sign = urllib.parse.quote_plus(base64.b64encode(hmac_code))
    # 構(gòu)造請(qǐng)求URL
    webhook = '{}&timestamp={}&sign={}'.format(webhook, timestamp, sign)

    # 構(gòu)造消息體
    headers = {'Content-Type': 'application/json'}
    data = {
        "msgtype": "text",
        "text": {
            "content": message
        }
    }

    # 發(fā)送請(qǐng)求
    response = requests.post(url=webhook, headers=headers, data=json.dumps(data))
    return response.text


# 初始化最后已知的金叉和死叉時(shí)間點(diǎn)
last_golden_cross = pd.Timestamp('2024-08-26 17:05:00+08:00')
last_death_cross = pd.Timestamp('2024-08-26 18:08:00+08:00')

webhook_url = "https://oapi.dingtalk.com/robot/send?access_token=8a6ddcf98d3b47c63333580bfe9d0bad55b17272eea05cc9c0af7f7be4de070d"
secret_key = "你的加簽密鑰"  # 如果沒(méi)有設(shè)置加簽暴氏，則這部分可以忽略
while True:  # 開(kāi)始一個(gè)死循環(huán)
    try:
        # 獲取K線數(shù)據(jù)
        symbol = 'BTCUSDT'  # 示例: BTC/USDT
        interval = '5m'  # 示例: 3分鐘
        data = get_klines(symbol, interval)

        # 計(jì)算MACD
        df_macd = calculate_macd(data)

        # 找到金叉和死叉
        crossovers = find_macd_crossovers(df_macd)

        # 檢查并打印新的金叉和死叉的時(shí)間點(diǎn)
        new_crossovers = False
        for cross_type, times in crossovers.items():
            for time_point in times:
                time_utc8 = pd.to_datetime(time_point, unit='ms').tz_localize('UTC').tz_convert('Asia/Shanghai')

                # 檢查時(shí)間點(diǎn)是否新
                if cross_type == '上穿' and time_utc8 > last_golden_cross:
                    message = f"{interval}穿越充石，新的{cross_type} at: {time_utc8}"
                    print(message)
                    result = send_dingtalk_message(webhook_url, secret_key, message)
                    print(result)
                    last_golden_cross = time_utc8
                    new_crossovers = True
                elif cross_type == '下穿' and time_utc8 > last_death_cross:
                    message = f"{interval}穿越厘唾，新的{cross_type} at: {time_utc8}"
                    print(message)
                    result = send_dingtalk_message(webhook_url, secret_key, message)
                    last_death_cross = time_utc8
                    new_crossovers = True

        if not new_crossovers:
            print("沒(méi)有新的金叉或死叉罪既。")

        print("Waiting for the next update...")
        time.sleep(2)  # 每10秒查詢(xún)一次狠毯，可以根據(jù)需要調(diào)整

    except KeyboardInterrupt:
        print("Program terminated by user.")
        break  # 當(dāng)用戶(hù)中斷程序(通常是通過(guò)Ctrl+C)時(shí)于置，退出循環(huán)
    except Exception as e:
        print(f"An error occurred: {e}")
        # 出現(xiàn)錯(cuò)誤時(shí)等待一段時(shí)間再重試劣光，避免因臨時(shí)的網(wǎng)絡(luò)API限制導(dǎo)致程序終止
        time.sleep(300)