update

RSSEntry.py

@@ -15,6 +15,9 @@
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 from json import loads
 from config import RSSConfig
+from time import time_ns
+from typing import List
+from hashl import sha256WithBase64
 
 
 class ChatEntry:
@@ -28,8 +31,66 @@ class ChatEntry:
             self.config = RSSConfig()
 
 
+class HashEntry:
+    def __init__(self, data=None, id: str = None, hash: str = None):
+        self.id = data[0] if data is not None and data[0] is not None else None
+        self.hash = data[1] if data is not None and data[1] is not None else None
+        self.time = data[2] if data is not None and data[2] is not None else time_ns()
+        if id is not None:
+            self.id = id
+        if hash is not None:
+            self.hash = hash
+
+
+def calHash(url: dict, item: dict) -> HashEntry:
+    hashd = sha256WithBase64(url)
+    hasht = url
+    if 'title' in item and item['title'] is not None:
+        hasht = hasht + item['title']
+    if 'link' in item and item['link'] is not None:
+        hasht = hasht + item['link']
+    if 'description' in item and item['description'] is not None:
+        hasht = hasht + item['description']
+    hashed = sha256WithBase64(hasht)
+    return HashEntry(id=hashd, hash=hashed)
+
+
+class HashEntries:
+    def __init__(self, maxCount: int = 100):
+        self.__list = []
+        self.__maxCount = maxCount if maxCount is not None and maxCount >= 1 else 100
+
+    def __removeMax(self):
+        self.__sort()
+        while len(self.__list) > self.__maxCount:
+            t = self.__list[0]
+            self.__list.remove(t)
+
+    def __sort(self, reverse: bool = False):
+        self.__list.sort(key=lambda d: d.time, reverse=reverse)
+
+    def add(self, d: HashEntry):
+        if d.hash is not None and d.id is not None:
+            for v in self.__list:
+                if v.hash == d.hash and v.id == d.id:
+                    return
+            self.__list.append(d)
+            self.__removeMax()
+
+    def getList(self) -> List[HashEntry]:
+        self.__removeMax()
+        r = []
+        for i in self.__list:
+            r.append(i)
+        return r
+
+    def setMaxCount(self, maxCount: int):
+        self.__maxCount = maxCount if maxCount >= 1 else 100
+        self.__removeMax()
+
+
 class RSSEntry:
-    def __init__(self, data=None):
+    def __init__(self, data=None, maxCount: int = 100):
         self.title = None
         if data is not None and data[0] is not None:
             self.title = data[0]
@@ -46,3 +107,4 @@ class RSSEntry:
         if data is not None and data[4] is not None:
             self.id = data[4]
         self.chatList = []
+        self.hashList = HashEntries(maxCount)