post HN news ..

1 files changed, 34 insertions, 5 deletions

diff --git a/main.py b/main.py
index 73584f5..a589057 100644
--- a/main.py
+++ b/main.py
@@ -5,6 +5,7 @@ import re
 import random
 import os
 from patterns import *
+import hnnews
 
 
 def login_refresh_token():
@@ -15,7 +16,7 @@ def login_refresh_token():
 
 
 def restore_session():
-    return Mastodon(client_id=config.CLIENTID, access_token=config.TOKEN)
+    return Mastodon(client_id=config.CLIENTID, access_token=config.TOKEN, feature_set='pleroma')
 
 
 class VnilCat:
@@ -23,6 +24,7 @@ class VnilCat:
         self.session = restore_session()
         self.tl_lastseen_sid = None
         self.config = config
+        self.news = []
         try:
             self.session.account_verify_credentials()
         except:
@@ -48,20 +50,43 @@ class VnilCat:
         return status['account']['acct'] != self.config.UNAME
 
     def is_mine(self, status):
-        return status['account']['acct'] != self.config.UNAME
+        return status['account']['acct'] == self.config.UNAME
 
     def reply_meow(self, ori_status):
         print("replying meow to ", ori_status["id"])
         self.session.status_reply(
             to_status=ori_status, status=random.choice(cat_sounds))
 
+    def post_hn_news(self, amount=3):
+        if amount <= 0:
+            return
+        if amount > len(self.news):
+            self.news = hnnews.get_topnews(20)
+        if amount > len(self.news):
+            amount = len(self.news)
+        status = "Hear ye, hear ye \n\n"
+        for i in range(amount):
+            n = self.news.pop(random.randrange(len(self.news)))
+            status += n["title"]
+            status += "\n"
+            status += n["url"]
+            status += "\n\n"
+        self.session.toot(status)
+
+
     def handle_command(self, status, content):
         print("we have a command", content)
         cmd = content[1:].strip().split()
+        actor = status["account"]["acct"]
+        # handle cmds
+        if cmd[0] == "news":
+            if actor not in config.ADMINS:
+                return
+            self.post_hn_news()
 
     def handle_home_status(self, status):
         self.tl_lastseen_sid = status["id"]
-        if not self.is_mine(status):
+        if self.is_mine(status):
             return
         acc = status["account"]
         content = cleanhtml(status["content"])
@@ -74,7 +99,7 @@ class VnilCat:
                 print("fail to post")
 
     def scantimeline(self):
-        # print("scanning timeline, lastseen=",self.tl_lastseen_sid)
+        print("scanning timeline, lastseen=",self.tl_lastseen_sid)
         tl = self.session.timeline_home(since_id=self.tl_lastseen_sid)
         # important! make sure to iterate from older status to newer
         # otherwise the 'last_seen' won't be updated correctly
@@ -132,17 +157,21 @@ def init_bot():
                         to_file=config.CLIENTID)
     session = login_refresh_token()
 
-
+# TODO use sched
 def run():
     cat = VnilCat(config)
+    epoch = 1
     while True:
         try:
             cat.handle_notification()
             cat.scantimeline()
+            if epoch % 360 == 1 :
+                cat.post_hn_news(3)
         except Exception as e:
             print("something wrong...")
             print(e)
         time.sleep(config.POLL_INTERVAL)
+        epoch += 1
 
 
 if __name__ == "__main__":