hiprotect/cache.py at main · TerminalPR0/hiprotect · 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
from mongo import db

class Collection:
    def __init__(self, collection):
        self.collection = db[collection]
        print(collection, "loaded")
        self.cached = {}

    def add(self, id, data):
        idict = {"_id": id}
        self.collection.update_one(idict, {"$set": data}, upsert=True)
        if not id in self.cached:
            self.cached[id] = {}
        for i in data:
            self.cached[id][i] = data[i]

    def remove(self, id):
        idict = {"_id": id}
        self.collection.delete_one(idict)
        del self.cached[id]

    def delete(self, id, data):
        idict = {"_id": id}
        self.collection.update_one(idict, {"$unset": data})
        for i in data:
            del self.cached[id][i]

    def load_data(self):
        results = self.collection.find({})
        for res in results:
            self.cached[res["_id"]] = res
        return self.cached


configs = Collection(collection = "config")
configs_data = configs.load_data()

antiflood = Collection(collection = "antiflood")
antiflood_data = antiflood.load_data()

antiinvite = Collection(collection = "antiinvite")
antiinvite_data = antiinvite.load_data()

antiraid = Collection(collection = "antiraid")
antiraid_data = antiraid.load_data()

bans = Collection(collection = "bans")
bans_data = bans.load_data()

locks = Collection(collection = "locks")
locks_data = locks.load_data()

mutes = Collection(collection = "mutes")
mutes_data = mutes.load_data()

warns = Collection(collection = "warns")
warns_data = warns.load_data()

invited = Collection(collection = "invited")
invited_data = invited.load_data()

perms = Collection(collection = "perms")
perms_data = perms.load_data()

rr = Collection(collection = "rr-new")
rr_data = rr.load_data()

whitelist = Collection(collection = "whitelist")
whitelist_data = whitelist.load_data()

quarantine = Collection(collection = "quarantine")
quarantine_data = quarantine.load_data()

bl = Collection(collection = "bot-bl")
bl_data = bl.load_data()

logs = Collection(collection = "logs")
logs_data = logs.load_data()

bonus = Collection(collection = "bonus")
bonus_data = bonus.load_data()

antinuke = Collection(collection = "antinuke")
antinuke_data = antinuke.load_data()

botstats = Collection(collection = "bot_stats")
botstats_data = botstats.load_data()

premium = Collection(collection = "premium")
premium_data = premium.load_data()

invoices = Collection(collection = "invoices")
invoices_data = invoices.load_data()

allowed = Collection(collection = "allowed")
allowed_data = allowed.load_data()