Spaces:
Sleeping
Sleeping
File size: 2,175 Bytes
9d5781e |
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 |
from threading import RLock
from Powers.database import MongoDB
from Powers.utils.msg_types import Types
INSERTION_LOCK = RLock()
class Floods(MongoDB):
"""Class to store flood limit and action of a chat"""
db_name = "flood"
def __init__(self):
super().__init__(self.db_name)
def save_flood(
self,
chat_id: int,
limit: int,
within: int,
action: str,
):
with INSERTION_LOCK:
curr = self.find_one({"chat_id": chat_id, "limit": limit, "within": within, "action": action})
if curr:
if not(limit == int(curr['limit']) or within == int(curr['within']) or action == str(curr['action'])):
return self.update(
{
"_id": chat_id,
"limit": limit,
"within": within,
"action": action,
}
)
return self.insert_one(
{
"chat_id" : chat_id,
"limit": limit,
"within": within,
"action" : action
},
)
def is_chat(self, chat_id: int):
with INSERTION_LOCK:
curr = self.find_all({"chat_id": chat_id})
if curr:
action = [str(curr['limit']), str(curr['within']), str(curr['action'])]
return action
return False
def get_action(self, chat_id: int, limit: int, within: int, action: str):
with INSERTION_LOCK:
curr = self.find_one({"chat_id": chat_id, "limit": limit, "within": within, "action": action})
if curr:
return curr
return "Flood haven't set"
def rm_flood(self, chat_id: int, limit: int, within: int, action: str):
with INSERTION_LOCK:
curr = self.find_one({"chat_id": chat_id, "limit": limit, "within": within, "action": action})
if curr:
self.delete_one(curr)
return True
return False
|