ptba_notes/main.py at main · barbax7/ptba_notes · 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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
from os import getenv

import telebot
from telebot.types import Message
from telebot.util import extract_arguments

from utils import autodelete, has_reply, log
from utils.db import Db

#pip install python-dotenv
try:
    from os.path import dirname, join

    from dotenv import load_dotenv
    dotenv_path = join(dirname(__file__), '.env')
    load_dotenv(dotenv_path)
except ModuleNotFoundError:
    print('dotenv non è installato')
except:
    pass

db = Db(getenv('DB_NAME'))

bot = telebot.TeleBot(getenv('BOT_TOKEN'), parse_mode = 'HTML')

GROUPID = int(getenv('GROUP_ID'))

@bot.message_handler(commands=['notes'])
def notes(message: Message):
    keys = db.select("SELECT Key, Description FROM Notes")
    if keys:
        keys.sort()
        getkeytext = "Use '/get key' - to get note.\n\n<b>Keys:</b>\n"
        for key in keys:
            getkeytext+=f"<code>{str(key[0])}</code> - <i>{str(key[1])}</i>\n"
        msg = bot.send_message(message.chat.id, getkeytext, reply_to_message_id = has_reply(message))
    else:
        msg = bot.send_message(message.chat.id, "There are no keys yet. Use\n'/add_note key description\ntext'")
    if message.chat.id == GROUPID: bot.delete_message(message.chat.id, message.message_id)
    autodelete(bot, msg)

@bot.message_handler(commands=['get'])
def get_value(message: Message):
    a = extract_arguments(message.text)
    if a:
        key = db.select("SELECT Text FROM Notes WHERE key = ?", str(a))
        if key:
            bot.send_message(message.chat.id, str(key[0][0]), reply_to_message_id = has_reply(message))
        else:
            msg = bot.send_message(message.chat.id, "Note with this key isn't available❌\n/notes")
    else:
        msg = bot.send_message(message.chat.id, "Use /get key - to get note.")
    if message.chat.id == GROUPID: bot.delete_message(message.chat.id, message.message_id)
    autodelete(bot, msg)

@bot.message_handler(commands=['add_note'])
def add_notes(message: Message):
    if message.chat.id == GROUPID:
        admin = db.select("SELECT Uid FROM Admins WHERE Uid = ?", message.from_user.id)
        if admin:
            text = message.html_text.split('\n', maxsplit=1)
            if len(text) > 1:
                nextext = text[0].split(maxsplit=2)
                if len(nextext) == 1:
                    msg = bot.send_message(message.chat.id, "Use '/add_note key description\ntext'")
                    bot.delete_message(message.chat.id, message.message_id)
                elif len(nextext) == 2:
                    msg = bot.send_message(message.chat.id, "You forgot the description❌")
                    autodelete(bot, message)
                elif len(nextext) == 3:
                    key = db.select("SELECT Text FROM Notes WHERE Key = ?", str(nextext[1]))
                if key:
                    msg = bot.send_message(message.chat.id, "Note with this key already exist❌\nPlease create note again with another key.")
                else:
                    db.query("INSERT INTO Notes(Key, Description, Text) VALUES(?, ?, ?)", str(nextext[1]), str(nextext[2]), str(text[1]), commit = True)
                    msg = bot.send_message(message.chat.id, "Note successfully added with key <code>{}</code> ✅".format(nextext[1]))
            else:
                msg = bot.send_message(message.chat.id, "You forgot the text❌")
        else:
            msg = bot.send_message(message.chat.id, "You're not a moderator❌")
        autodelete(bot, msg, 2)
        autodelete(bot, message, 2)
    else:
        bot.send_message(message.chat.id, text="<b>Notes can be modified ONLY in our group,\n\
            \nJoin ➤ <a href='https://telegram.me/joinchat/Bn4ixj84FIZVkwhk2jag6A'>pyTelegramBotAPI</a></b>")

@bot.message_handler(commands=['delete_note'])
def delete_notes(message):
    if message.chat.id == GROUPID:
        admin = db.select("SELECT Uid FROM Admins WHERE Uid = ?", message.from_user.id)
        if admin:
            a = extract_arguments(message.text)
            if a:
                key = db.select("SELECT Key FROM Notes WHERE Key = ?", str(a))
                if key:
                    db.query("DELETE FROM Notes WHERE Key = ?", str(a), commit = True)
                    msg = bot.send_message(message.chat.id, "Note with key <code>{}</code> successfully deleted ✅".format(a))
                else:
                    msg = bot.send_message(message.chat.id, "Note with key <code>{}</code> isn't available ❌".format(a))
            else:
                msg = bot.send_message(message.chat.id, "You forgot the key ❌")
        else:
            msg = bot.send_message(message.chat.id, "You're not a moderator ❌")
        autodelete(bot, msg, 2)
        autodelete(bot, message, 0)
    else:
        bot.send_message(message.chat.id, text="<b>Notes can be modified ONLY in our group,\n\
            \nJoin ➤ <a href='https://telegram.me/joinchat/Bn4ixj84FIZVkwhk2jag6A'>pyTelegramBotAPI</a></b>")

@bot.message_handler(commands=['add_admin'])
def add_admins(message: Message):
    if message.chat.id == GROUPID:
        admin = db.select("SELECT Uid FROM Admins WHERE Uid = ?", message.from_user.id)
        if admin:
            if message.reply_to_message:
                admin2 = db.select("SELECT Uid FROM Admins WHERE Uid = ?", message.reply_to_message.from_user.id)
                if admin2:
                    msg = bot.send_message(message.chat.id, "This user is already a moderator ❌")
                else:
                    db.query("INSERT INTO Admins(Uid) VALUES(?)", message.reply_to_message.from_user.id, commit = True)
                    msg = bot.send_message(message.chat.id, f"<a href='tg://user?id={message.reply_to_message.from_user.id}'>This</a> user has now successfully become a moderator ✅")
            else:
                msg = bot.send_message(message.chat.id, "This command must be a reply to the user ❌")
        else:
            msg = bot.send_message(message.chat.id, "You're not a moderator❌")
        autodelete(bot, msg, 2)
        autodelete(bot, message, 1)
    else:
        bot.send_message(message.chat.id, text="<b>Admins can be added ONLY in our group,\n\
            \nJoin ➤ <a href='https://telegram.me/joinchat/Bn4ixj84FIZVkwhk2jag6A'>pyTelegramBotAPI</a></b>")

@bot.message_handler(commands=['help'])
def help_func(message):
    msg = bot.send_message(message.chat.id, "Usage:\n/notes - list of notes\n/add_note - add new note\n/delete_note - delete note\n/get - get note\n/add_admin - add new moderator")
    autodelete(bot, msg, 2)
    if message.chat.id == GROUPID: autodelete(bot, message, 0)

@bot.message_handler(commands=['start'])
def welcome(message: Message):
    if message.chat.id == GROUPID:
        msg = bot.send_message(message.chat.id, "Hi, I'm online!\nSend /help")
        autodelete(bot, msg, 0.5)
        autodelete(bot, message, 0)
    else:
        bot.send_message(message.chat.id, "Join pyTelegramBotApi's Telegram Group to get code snippets\n\nhttps://telegram.me/joinchat/Bn4ixj84FIZVkwhk2jag6A")

msg = log.info_log(bot, "I'm started!", GROUPID)
autodelete(bot, msg, 1)
bot.infinity_polling()