kopia lustrzana https://github.com/fabston/TradingView-Webhook-Bot
47 wiersze
1.5 KiB
Python
47 wiersze
1.5 KiB
Python
# ----------------------------------------------- #
|
|
# Plugin Name : TradingView-Webhook-Bot #
|
|
# Author Name : fabston #
|
|
# File Name : main.py #
|
|
# ----------------------------------------------- #
|
|
|
|
from handler import send_alert
|
|
import config
|
|
import time
|
|
from flask import Flask, request, jsonify
|
|
|
|
app = Flask(__name__)
|
|
|
|
|
|
def get_timestamp():
|
|
timestamp = time.strftime("%Y-%m-%d %X")
|
|
return timestamp
|
|
|
|
|
|
@app.route("/webhook", methods=["POST"])
|
|
def webhook():
|
|
whitelisted_ips = ['52.89.214.238', '34.212.75.30', '54.218.53.128', '52.32.178.7']
|
|
client_ip = request.headers.get('X-Forwarded-For', request.remote_addr)
|
|
if client_ip not in whitelisted_ips:
|
|
return jsonify({'message': 'Unauthorized'}), 401
|
|
try:
|
|
if request.method == "POST":
|
|
data = request.get_json()
|
|
if data["key"] == config.sec_key:
|
|
print(get_timestamp(), "Alert Received & Sent!")
|
|
send_alert(data)
|
|
return jsonify({'message': 'Webhook received successfully'}), 200
|
|
|
|
else:
|
|
print("[X]", get_timestamp(), "Alert Received & Refused! (Wrong Key)")
|
|
return jsonify({'message': 'Unauthorized'}), 401
|
|
|
|
except Exception as e:
|
|
print("[X]", get_timestamp(), "Error:\n>", e)
|
|
return jsonify({'message': 'Error'}), 400
|
|
|
|
|
|
if __name__ == "__main__":
|
|
from waitress import serve
|
|
|
|
serve(app, host="0.0.0.0", port=8080)
|