Athalbota_bot/api.py at main · asirihewage/Athalbota_bot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
# -*- coding: utf-8 -*-
"""TelegramBot
Original file is located at hhttps://github.com/asirihewage/Friendly_CryptoBot
# Telegram Bot
"""

# importing all dependencies
import logging
import logging.handlers as handlers
import pymongo
from flask import Flask
from utils.Configs import Configs
from utils.Logs import log

api = Flask(__name__)


# connect to the database
def connect():
    try:
        if Configs.BOT_MONGODB_CONNECTION_URL:
            log("Database Client initialized.")
            client = pymongo.MongoClient(Configs.BOT_MONGODB_CONNECTION_URL)
            database = client[Configs.BOT_DATABASE_NAME]
            if database:
                log("Database Connected.")
                return database
            else:
                log("Database Connection failed.")
                return None
        else:
            log("Database Client Connection failed.")
            return None
    except Exception as e:
        log("Database Error : {}".format(e))
        return None


dbConnection = connect()


def promote_user(user_id):
    try:
        if user_id and dbConnection:
            usersCollection = dbConnection.get_collection("users")
            if usersCollection.find_one({"id": int(user_id)}):
                usersCollection.update_one({"id": int(user_id)}, {"$set": {"level": 1}})
                return True
            else:
                return False
        else:
            log("Database connection error")
            return False
    except Exception as er:
        print(er)
        return False


@api.route("/")
def index():
    html = '<HTMl><head><meta http-equiv="refresh" content="1; URL={}"></head>Oopz! The link is invalid</HTML>'.format(
        Configs.BOT_LINK)
    return html


@api.route("/verifyuser/<user_id>")
def verifyuser(user_id):
    html = '<HTMl><head><meta http-equiv="refresh" content="2; URL={}"></head>Account Verified!</HTML>'.format(Configs.BOT_LINK)
    if user_id and promote_user(int(user_id)):
        return html
    else:
        return '<HTMl><head><meta http-equiv="refresh" content="2; URL={}"></head>Something went wrong</HTML>'.format(
            Configs.BOT_LINK)