TelegramModeratorBotAI/bot.py at master · GraphGrail/TelegramModeratorBotAI · 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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
import os
import telebot
import sqlite3
import settings
from time import time
from datetime import datetime, timedelta
from commentEvaluator import CommentEvaluator
from keras.models import load_model
import pickle

#spacy.load('xx_ent_wiki_sm')
script_dir = os.path.dirname(__file__)
bot = telebot.TeleBot(settings.BOT_TOKEN)
conn = sqlite3.connect(os.path.join(script_dir, 'universabot.db'), check_same_thread=False)
c = conn.cursor()

with open(os.path.join(script_dir, 'model_files/token.pickle'), 'rb') as handle:
    tok = pickle.load(handle)

model = load_model(os.path.join(script_dir, 'model_files/rus_weights2.hdf5'))
evaluator = CommentEvaluator(model, tok)


def get_language(lang_code):
    if not lang_code:
        return "en"
    if "-" in lang_code:
        lang_code = lang_code.split("-")[0]
    if lang_code == "ru":
        return "ru"
    else:
        return "en"


@bot.message_handler(func=lambda message: message.entities is not None)
def delete_links(message):
    for entity in message.entities:
        if entity.type in settings.RESTRICTED_LINKS:
            bot.delete_message(message.chat.id, message.message_id)
            bot.send_message(message.chat.id, settings.strings.get(get_language(message.from_user.language_code)).get("ro_link"), reply_to_message_id=message.message_id)
            bot.restrict_chat_member(message.chat.id, message.from_user.id, until_date=time()+60*settings.MINUTES)
        else:
            return


@bot.message_handler(func=lambda message: message.text and evaluator.analyze(message.text) == "negative")
def handle_negative(message):

    c.execute('SELECT ID FROM Userstat WHERE user_id=? and chat_id=?', (int(message.from_user.id),
                                                                        int(message.chat.id)))
    rows = c.fetchall()
    if len(rows) == 0:
        c.execute('INSERT INTO Userstat VALUES(NULL, ?, ?, ?, ?, ?, ?)', (message.from_user.id,
                                                                          message.from_user.first_name, 1,
                                                                          datetime.now(), 0, message.chat.id,))
        c.execute('INSERT INTO message VALUES(NULL, ?, ?, ?, ?, ?, ?)', (message.message_id, message.text, 0,
                                                                         message.from_user.id, datetime.now(),
                                                                         message.chat.id,))
        conn.commit()
    else:
        c.execute('UPDATE Userstat SET negative_total=negative_total+1')
        c.execute('Insert INTO message VALUES(NULL, ?, ?, ?, ?, ?, ?)', (message.message_id, message.text, 0,
                                                                         message.from_user.id, datetime.now(),
                                                                         message.chat.id,))
        conn.commit()
        c.execute("SELECT MESSAGE_ID FROM message WHERE CREATED_AT >= Datetime('now', '-60 minutes', 'localtime') AND SENITMENT=0 AND USER_ID=?", (message.from_user.id,))
        rows = c.fetchall()
        print(rows)
        if len(rows) >= 5:
            if not int(message.from_user.id) in settings.ADMINS:
                bot.restrict_chat_member(message.chat.id, message.from_user.id, until_date=time()+60*settings.MINUTES)
                bot.send_message(message.chat.id,
                                 settings.strings.get(get_language(message.from_user.language_code)).get("ro_msg"),
                                 reply_to_message_id=message.message_id)
                c.execute("INSERT INTO banned VALUES(NULL, ?, ?, ?)", (message.from_user.id, datetime.now(), datetime.now() + timedelta(minutes=180),))
                conn.commit()
            else:
                print('Admin can not be banned!')
            #bot.delete_message(message.chat.id, message.message_id)


@bot.message_handler(func=lambda message: message.text and evaluator.analyze(message.text) == "positive")
def handle_positive(message):
    c.execute('SELECT ID FROM Userstat WHERE user_id=? and chat_id=?',
              (int(message.from_user.id), int(message.chat.id)))
    rows = c.fetchall()
    if len(rows) == 0:
        c.execute('INSERT INTO Userstat VALUES(NULL, ?, ?, ?, ?, ?, ?)', (message.from_user.id,
                                                                          message.from_user.first_name, 0,
                                                                          datetime.now(), 1, message.chat.id,))
        c.execute('INSERT INTO message VALUES(NULL, ?, ?, ?, ?, ?, ?)', (message.message_id, message.text, 1,
                                                                         message.from_user.id, datetime.now(),
                                                                         message.chat.id,))
        conn.commit()
    else:
        c.execute('UPDATE Userstat SET positive_total=positive_total+1')
        c.execute('Insert INTO message VALUES(NULL, ?, ?, ?, ?, ?, ?)', (message.message_id, message.text, 1,
                                                                         message.from_user.id, datetime.now(),
                                                                         message.chat.id,))
        conn.commit()

@bot.message_handler(func=lambda message: message.text and evaluator.analyze(message.text) == "neutral")
def handle_neutral(message):
    c.execute('SELECT ID FROM Userstat WHERE user_id=? and chat_id=?',
              (int(message.from_user.id), int(message.chat.id)))
    rows = c.fetchall()
    if len(rows) == 0:
        c.execute('INSERT INTO Userstat VALUES(NULL, ?, ?, ?, ?, ?, ?)', (message.from_user.id,
                                                                          message.from_user.first_name, 0,
                                                                          datetime.now(), 0, message.chat.id,))
        c.execute('INSERT INTO message VALUES(NULL, ?, ?, ?, ?, ?, ?)', (message.message_id, message.text, 2,
                                                                         message.from_user.id, datetime.now(),
                                                                         message.chat.id,))
        conn.commit()
    else:
        c.execute('UPDATE Userstat SET positive_total=positive_total+1')
        c.execute('Insert INTO message VALUES(NULL, ?, ?, ?, ?, ?, ?)', (message.message_id, message.text, 2,
                                                                         message.from_user.id, datetime.now(),
                                                                         message.chat.id,))
        conn.commit()


if __name__ == "__main__":
    bot.polling(none_stop=True)