add /banlist, /ban, /unban

blackList.py

@@ -15,21 +15,96 @@
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 from re import search
 from typing import List
+from textc import textc, timeToStr
+from enum import Enum, unique
+from math import ceil, floor
 
 
 class BlackInfo:
-    uid: int = None
+    __uid: int = None
     op_uid: int = None
     from_config: bool = False
     add_time: int = None
     reason: str = None
 
-    def __init__(self, uid: int, op_uid: int = None, add_time: int = None, reason: str = None):
-        self.uid = uid
+    @property
+    def uid(self) -> int:
+        return self.__uid
+
+    def __init__(self, uid: int, op_uid: int = None, add_time: int = None, reason: str = None, title: str = None):
+        self.__uid = uid
         self.op_uid = op_uid
         self.from_config = True if self.op_uid is None else False
         self.add_time = add_time
         self.reason = reason
+        self.title = title
+
+    @property
+    def name(self) -> str:
+        return f"{self.uid}" if self.title is None else self.title
+
+
+class BlackInfoListIter:
+    def __init__(self, li: List[BlackInfo]):
+        self.__l = li
+        self.__start = 0
+
+    def __next__(self) -> BlackInfo:
+        if self.__start >= len(self.__l):
+            raise StopIteration
+        v = self.__l[self.__start]
+        self.__start += 1
+        return v
+
+
+class BlackInfoList:
+    def __add__(self, b):
+        o = BlackInfoList(self.__l)
+        if isinstance(b, (BlackInfoList, list)):
+            for i in b:
+                o.append(i)
+        elif isinstance(b, BlackInfo):
+            o.append(b)
+        return o
+
+    def __getitem__(self, i) -> BlackInfo:
+        return self.__l[i]
+
+    def __iadd__(self, b):
+        if isinstance(b, (BlackInfoList, list)):
+            for i in b:
+                self.append(i)
+        elif isinstance(b, BlackInfo):
+            self.append(b)
+        return self
+
+    def __init__(self, li: List[BlackInfo] = None):
+        self.__t = []
+        self.__l = []
+        if li is not None:
+            for i in li:
+                self.append(i)
+
+    def __iter__(self):
+        it = BlackInfoListIter(self.__l)
+        return it
+
+    def __len__(self):
+        return len(self.__l)
+
+    def append(self, info: BlackInfo):
+        if info.uid not in self.__t:
+            self.__l.append(info)
+            self.__t.append(info.uid)
+
+    def find(self, uid: int):
+        if uid in self.__t:
+            return self.__t.index(uid)
+        else:
+            return -1
+
+    def isInBlackList(self, chatId: int) -> bool:
+        return chatId in self.__t
 
 
 class ConfigBlackList:
@@ -43,8 +118,8 @@ class ConfigBlackList:
                 if search(r'^[\+-]?[0-9]+$', i) is not None:
                     self.__list.append(int(i))
 
-    def getBlackList(self) -> List[BlackInfo]:
-        r = []
+    def getBlackList(self) -> BlackInfoList:
+        r = BlackInfoList()
         for i in self.__list:
             r.append(BlackInfo(i))
         return r
@@ -58,23 +133,129 @@ class BlackList:
         from rssbot import main
         self._main: main = m
         self._configBlackList = ConfigBlackList(m, self._main._setting._blackList)
+        self._blackList = self._main._db.getBlackList()
 
-    def getBlackList(self) -> List[BlackInfo]:
-        m = self._configBlackList.getBlackList()
-        t = []
-        r = []
-        for i in m:
-            if i.uid not in t:
-                t.append(i.uid)
-                r.append(i)
+    def ban(self, i: BlackInfo):
+        re = self._main._db.addBlackInfo(i)
+        if re:
+            self._blackList = self._main._db.getBlackList()
+            self.checkRSSList()
+        return re
+
+    def getBlackList(self) -> BlackInfoList:
+        r = self._configBlackList.getBlackList()
+        r += self._blackList
         return r
 
     def isInBlackList(self, chatId: int) -> bool:
         if self._configBlackList.isInBlackList(chatId):
             return True
+        if self._blackList.isInBlackList(chatId):
+            return True
         return False
 
     def checkRSSList(self):
         li = self.getBlackList()
         for i in li:
             self._main._db.removeChatInChatList(i.uid)
+
+    def unban(self, userId: int):
+        re = self._main._db.removeFromBlackList(userId)
+        if re:
+            self._blackList = self._main._db.getBlackList()
+        return re
+
+
+@unique
+class InlineKeyBoardForBlackList(Enum):
+    FirstPage = 0
+    LastPage = 1
+    PrevPage = 2
+    NextPage = 3
+    Close = 4
+    BlackInfo = 5
+    BackToList = 6
+    Unban = 7
+    ConfirmUnban = 8
+    CancleUnban = 9
+
+
+def getTextContentForBlackInfo(i: BlackInfo):
+    t = textc()
+    link = '' if i.uid < 0 else f'tg://user?id={i.uid}'
+    t += f'被封禁用户: <a href="{link}">{i.name}</a>'
+    if i.from_config:
+        t += '来源：配置文件'
+    else:
+        t += '来源：数据库'
+        t += f'封禁操作人：<a href="tg://user?id={i.op_uid}">{i.op_uid}</a>'
+        t += f'封禁时间：{timeToStr(i.add_time)}'
+        t += f'封禁理由：{i.reason}'
+    return t.tostr()
+
+
+def getTextContentForUnbanBlackInfo(i: BlackInfo):
+    link = '' if i.uid < 0 else f'tg://user?id={i.uid}'
+    return f'是否要取消封禁<a href="{link}">{i.name}</a>？'
+
+
+def getInlineKeyBoardForBlackList(bl: BlackInfoList, page: int = 1, lastPage: bool = False, itemIndex: int = None):
+    d = []
+    i = -1
+    lineLimit = 7
+    l = len(bl)
+    pn = ceil(l / lineLimit)
+    if lastPage:
+        page = pn
+    if itemIndex is not None and itemIndex >= 0:
+        page = floor(itemIndex / lineLimit) + 1
+    if l != 0:
+        page = max(min(pn, page), 1)
+        s = max(lineLimit * (page - 1), 0)
+        n = min(lineLimit * page, l)
+        while s < n:
+            d.append([])
+            i += 1
+            d[i].append({'text': bl[s].name, 'callback_data': f'2,{InlineKeyBoardForBlackList.BlackInfo.value},{s},{bl[s].uid}'})
+            s += 1
+        if pn != 1:
+            d.append([])
+            i += 1
+            if page != 1:
+                d[i].append({'text': '上一页', 'callback_data': f'2,{InlineKeyBoardForBlackList.PrevPage.value},{page}'})
+            if page != pn:
+                d[i].append({'text': '下一页', 'callback_data': f'2,{InlineKeyBoardForBlackList.NextPage.value},{page}'})
+            d.append([])
+            i += 1
+            if page != 1:
+                d[i].append({'text': '首页', 'callback_data': f'2,{InlineKeyBoardForBlackList.FirstPage.value}'})
+            if page != pn:
+                d[i].append({'text': '尾页', 'callback_data': f'2,{InlineKeyBoardForBlackList.LastPage.value}'})
+    d.append([])
+    i += 1
+    d[i].append({'text': '关闭', 'callback_data': f'2,{InlineKeyBoardForBlackList.Close.value}'})
+    return {'inline_keyboard': d}
+
+
+def getInlineKeyBoardForBlackInfo(b: BlackInfo, index: int):
+    d = []
+    i = -1
+    if not b.from_config:
+        d.append([])
+        i += 1
+        d[i].append({'text': '取消封禁', 'callback_data': f'2,{InlineKeyBoardForBlackList.Unban.value},{index},{b.uid}'})
+    d.append([])
+    i += 1
+    d[i].append(
+        {'text': '返回', 'callback_data': f'2,{InlineKeyBoardForBlackList.BackToList.value},{index}'})
+    return {'inline_keyboard': d}
+
+
+def getInlineKeyBoardForUnbanBlackInfo(b: BlackInfo, index: int):
+    d = []
+    i = -1
+    d.append([])
+    i += 1
+    d[i].append({'text': '是', 'callback_data': f'2,{InlineKeyBoardForBlackList.ConfirmUnban.value},{index},{b.uid}'})
+    d[i].append({'text': '否', 'callback_data': f'2,{InlineKeyBoardForBlackList.CancleUnban.value},{index},{b.uid}'})
+    return {'inline_keyboard': d}