Spaces:
Sleeping
Sleeping
File size: 2,642 Bytes
ea508b7 |
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 |
from threading import RLock
from Powers import LOGGER
from Powers.database import MongoDB
INSERTION_LOCK = RLock()
class LOCKS(MongoDB):
"""Class to store locks"""
db_name = "locks"
def __init__(self) -> None:
super().__init__(self.db_name)
def insert_lock_channel(self, chat: int, locktype: str):
"""
locktypes: anti_c_send, anti_fwd, anti_fwd_u, anti_fwd_c, anti_links
"""
curr = self.find_one({"chat_id":chat,"locktype":locktype})
if curr:
return False
else:
with INSERTION_LOCK:
hmm = self.merge_u_and_c(chat,locktype)
if not hmm:
self.insert_one({"chat_id":chat,"locktype":locktype})
return True
def remove_lock_channel(self, chat: int, locktype: str):
"""
locktypes: anti_c_send, anti_fwd, anti_fwd_u, anti_fwd_c, anti_links
"""
curr = self.find_one({"chat_id":chat,"locktype":locktype})
if curr:
with INSERTION_LOCK:
self.delete_one({"chat_id":chat,"locktype":locktype})
return True
else:
return False
def get_lock_channel(self, locktype: str="all"):
"""
locktypes: anti_c_send, anti_fwd, anti_fwd_u, anti_fwd_c, anti_links
"""
if locktype not in ["anti_c_send","anti_fwd","anti_fwd_u","anti_fwd_c","anti_links", "all"]:
return False
else:
if locktype == "all":
find = {}
else:
find = {"locktype":locktype}
curr = self.find_all(find)
if not curr:
list_ = []
else:
list_ = [i["chat_id"] for i in curr]
return list_
def merge_u_and_c(self, chat: int, locktype: str):
if locktype == "anti_fwd_u":
curr = self.find_one({"chat_id":chat,"locktype":"anti_fwd_c"})
elif locktype == "anti_fwd_c":
curr = self.find_one({"chat_id":chat,"locktype":"anti_fwd_u"})
else:
return False
if curr:
self.delete_one({"chat_id":chat,"locktype":locktype})
self.insert_one({"chat_id":chat,"locktype":"anti_fwd"})
return True
else:
return False
def is_particular_lock(self, chat: int, locktype: str):
"""
locktypes: anti_c_send, anti_fwd, anti_fwd_u, anti_fwd_c, anti_links
"""
curr = self.find_one({"chat_id":chat,"locktype":locktype})
if curr:
return True
else:
return False
|