add rss

worker/rss.py

@@ -0,0 +1,107 @@
+from zzcore import StdAns
+
+import feedparser, json
+
+allowCMD = ['sub','unsub','list','get']
+
+allSub = [
+    {
+        'url' : 'https://farseerfc.me/feeds/atom.xml',
+        'title' : 'Farseerfc的小窩',
+        'lastUpdated' : 'Sat, 12 Dec 2020 22:55:52 +0800',
+    }
+]
+
+class Ans(StdAns):
+
+    def push(self,title,posttitle,posturl):
+        msg = f'{title}\n{posttitle} {posturl}'
+        self.sendmsg(msg)
+    
+    def GETMSG(self):
+
+        if len(self.parms) < 2:
+            return '不加参数是坏文明！'
+        
+        gid = str(self.gid)
+        cmd = self.parms[1]
+        if cmd not in allowCMD:
+            return f'咱不知道 {cmd} 是什么东西 ●﹏●'
+
+        try:
+            nowdata = json.loads(self.DATAGET()[gid])
+        except:
+            nowdata ={
+                'allSub' : {},
+                'status' : False,
+            }
+        
+        if cmd == 'sub':
+            if len(self.parms) < 3:
+                return '汝想订阅什么呢？'
+            
+            suburl = self.parms[2]
+            try:
+                d = feedparser.parse(suburl)
+                title = d.feed.title
+                lastUpdated = d.feed.updated
+            
+            except:
+                return "咱好像没能成功订阅 (╥_╥)"
+            
+            newsub = {
+                'url': suburl,
+                'title': title,
+                'lastUpdated' : lastUpdated,
+            }
+
+            nowdata['allSub'][suburl] = newsub
+
+            msg = f'《{title}》订阅成功'
+
+        elif cmd == 'unsub':
+            if len(self.parms) < 3:
+                return '汝想退订什么呢？'
+
+            suburl = self.parms[2]
+            if suburl not in nowdata['allSub'].keys():
+                return '汝没有订阅过这个！'
+            
+            unsub = nowdata['allSub'].pop(suburl)
+            
+            msg = f'《{unsub.title}》已退订'
+
+        elif cmd == 'list':
+            msg = '订阅列表'
+
+            for sub in nowdata['allSub']:
+                msg += f'\n{sub.title}  {sub.url}'
+
+        elif cmd == 'enable':
+            nowdata['status'] = True
+            self.DATASET({gid:json.dumps(nowdata)})
+            self.sendmsg("订阅已启用，咱会每隔五分钟抓去订阅，有新内容就会推送哦")
+            while(nowdata['status']):
+                nowdata = json.loads(self.DATAGET()[gid])
+                for sub in nowdata['allSub']:
+                    try:
+                        d = feedparser.parse(sub['url'])
+                    except:
+                        self.sendmsg(f"咱抓取不到 《{sub['title']}》!")
+                        continue
+                    if d.feed.updated == sub['lastUpdated']:
+                        continue
+                    
+                    newfeedtitle = d.entries[0].title
+                    newfeedlink = d.entries[0].link
+                    self.push(sub['title'], newfeedtitle, newfeedlink)
+                    nowdata['allSub'][sub['url']]['lastUpdated'] = d.feed.updated
+                self.DATASET({gid:json.dumps(nowdata)})
+
+        elif cmd == 'disable':
+            nowdata['status'] = False
+            msg = '咱不会再推送了！'
+
+        self.DATASET({gid:json.dumps(nowdata)})
+        return msg
+