karaGroupManagement/dbManagement.py at main · Muhammad-Mirab/karaGroupManagement · 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
import sqlite3


class dbManager:
    def __init__(self) -> None:
        self.con = sqlite3.connect("KaraGroups.db", check_same_thread = False)
        self.cur = self.con.cursor()
        self.cur.execute("CREATE TABLE IF NOT EXISTS groupSettings(groupID STRING, admins STRING, isWelcomeEnabled STRING, welcomeMessage STRING, userPositions STRING, emoji INTEGER, link INTEGER, gif INTEGER, sticker INTEGER, picture INTEGER, video INTEGER, music INTEGER, file INTEGER, english INTEGER, bad_words INTEGER)")
        self.cur.execute("CREATE TABLE IF NOT EXISTS firstMessageQueue(chatID STRING, messageID STRING)")
        # self.cur.execute("INSERT OR IGNORE INTO groupSettings(groupID, welcomeMessage) VALUES ('1', 'سلام، همکار / دوست گرامی [یوزر جدید] به تیم [اسم گروه] خوش اومدی🌹 \n من ربات کارا هستم، یک ربات مدیریت گروه و پروژه از راه دور که میتونی از طریق دکمه زیر باهاش آشنا بشی و طرز کار باهاش رو یاد بگیری :)') ")
        # self.con.commit()

    # converts numeral chat id into sth the database can handle
    def getChatID(self, groupId):
        chat_idDB = str(groupId).split("-")
        # checks if the chat id value is negative if not will return the exact same thing
        return f"Minus{chat_idDB[1]}" if chat_idDB[0] == "" else chat_idDB[0]

    # adds the bot's first message to firstMessageQueue table in the db
    def firstMessageEditID(self, chatID, messageID):
        self.cur.execute(f"INSERT or IGNORE INTO firstMessageQueue VALUES ('{chatID}', '{messageID}')")
        self.con.commit()

    # returns the message id of the first message using the group's chat id
    def getFirstMessageEditID(self, chatID):
        self.cur.execute(f"SELECT messageID From firstMessageQueue WHERE chatID='{chatID}'")
        return [i[0] for i in self.cur.fetchall()]

    # it removes the first message's entry in database after editing the message and confirming the admin process
    def removeFirstMessageEditID(self, chatID):
        self.cur.execute(f"DELETE FROM firstMessageQueue WHERE chatID = '{chatID}'")
        self.con.commit()

    # return a list of the groups the bot is active in
    def getAvailableGroupsID(self):
        self.cur.execute("SELECT groupID FROM groupSettings")
        return [i[0] for i in self.cur.fetchall()]

    def getUserPositionDict(self, chatID):
        self.cur.execute(f"SELECT userPositions FROM groupSettings WHERE groupID='{chatID}'")
        return [i[0] for i in self.cur.fetchall()]

    def updateUserPositionDict(self, chatID, positionTitles):
        self.cur.execute(f'UPDATE groupSettings SET userPositions = "{positionTitles}" WHERE groupID=\'{self.getChatID(chatID)}\'')
        self.con.commit()

    # adds the group into groupSettings table which will hold all the admin related settings of the group
    def addToGroupSettings(self, groupID):
        defaultIsWelcomeEnabled = 1
        defaultWelcomeMessage = f'سلام، همکار / دوست گرامی [کاربر] به تیم خوش اومدی🌹 \n من ربات کارا هستم، یک ربات مدیریت گروه و پروژه از راه دور که میتونی از طریق دکمه زیر باهاش آشنا بشی و طرز کار باهاش رو یاد بگیری :)'
        userPositions = {}
        self.cur.execute(f"INSERT or IGNORE INTO groupSettings VALUES ('{groupID}', 'No one', {defaultIsWelcomeEnabled}, '{defaultWelcomeMessage}', '{str(userPositions)}', 1, 1, 1, 1, 1, 1, 0, 0, 1, 0)")
        self.con.commit()

    # using the chatID and attribute(which is the name of the value in db e.g. "isWelcomeEnabled") will check if the admin has disabled a feature
    def checkWetherGroupSettingIsSet(self, chatID, attribute):
        self.cur.execute(f"SELECT {attribute} FROM groupSettings WHERE groupID='{chatID}'")
        return [i[0] for i in self.cur.fetchall()]

    #Just return settings values
    def GetSettings(self, chat_ID):
        self.cur.execute(f"SELECT *  FROM  groupSettings Where groupID='{self.getChatID(chat_ID)}'")
        return self.cur.fetchall()[0][4:]

    def GetAllValues(self, chat_ID):
        self.cur.execute(f"SELECT *  FROM  groupSettings Where groupID='{self.getChatID(chat_ID)}'")
        return self.cur.fetchall()[0]

    def UpdateSettins(self, chat_id, subject):
        settings = ['emoji', 'link', 'gif', 'sticker', 'picture', 'video', 'music', 'file', 'english', 'bad_words']
        if subject in settings:
            # print(dbManager.GetSettings(self, chat_id)[settings.index(subject)])
            # print(f'{1 if dbManager.GetSettings(self, chat_id)[settings.index(subject) - 4] == 0 else 0}')
            self.cur.execute(f"UPDATE groupsettings SET {subject} = {1 if dbManager.GetSettings(self, chat_id)[settings.index(subject)] == 0 else 0} WHERE groupID='Minus{str(chat_id).replace('-', '')}'")
            self.con.commit()

        else:
            return 'subject does not exists'

    def AddAdmins(self, chat_id, Admins):
        self.cur.execute(f"UPDATE groupsettings SET admins = '{Admins}' WHERE groupID='Minus{str(chat_id).replace('-', '')}'")
        self.con.commit()

if __name__ == "__main__":
    pass