You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.2 KiB
44 lines
1.2 KiB
2 years ago
|
#-*-coding utf-8-*-
|
||
|
|
# Общественное достояние, 2023, Алексей Безбородов (Alexei Bezborodov) <AlexeiBv+mirocod_platform_bot@narod.ru>
|
||
|
|
|
||
|
|
# Работа с подписками
|
||
|
|
|
||
|
|
from enum import Enum
|
||
|
|
from enum import auto
|
||
|
|
|
||
|
|
# Тип поля в таблице
|
||
|
|
class SubscribeType(Enum):
|
||
|
|
ADD = auto()
|
||
|
|
ANY_ITEM_DEL = auto()
|
||
|
|
ANY_ITEM_EDIT = auto()
|
||
|
|
ITEM_DEL = auto()
|
||
|
|
ITEM_EDIT = auto()
|
||
|
|
|
||
|
|
class BotSubscribes
|
||
|
|
def __init__(self):
|
||
|
|
self.Clear()
|
||
|
|
|
||
|
|
def GetSubscribes(self):
|
||
|
|
return self.m_Subscribes
|
||
|
|
|
||
|
|
def Clear(self):
|
||
|
|
self.m_Subscribes = {}
|
||
|
|
|
||
|
|
def CheckSubscribe(self, a_UserID, a_ModuleName, a_Type, a_ItemID = -1):
|
||
|
|
s = self.GetSubscribes()
|
||
|
|
su = s.get(a_UserID, None)
|
||
|
|
if su:
|
||
|
|
sub_um = su.get(a_ModuleName, None)
|
||
|
|
if sub_um:
|
||
|
|
t = su.get(a_ItemID, None)
|
||
|
|
return t
|
||
|
|
return None
|
||
|
|
|
||
|
|
def AddSubscribe(self, a_UserID, a_ModuleName, a_Type, a_ItemID = -1):
|
||
|
|
s = self.GetSubscribes()
|
||
|
|
if not s.get(a_UserID, None):
|
||
|
|
s[a_UserID] = {}
|
||
|
|
if not s[a_UserID].get(a_ModuleName, None):
|
||
|
|
s[a_UserID][a_ModuleName] = {}
|
||
|
|
s[a_UserID][a_ModuleName][a_ItemID] = a_Type
|