source: python/newsblur.py @ b09ce17

#!/usr/bin/python

import operator
import re

import requests

import implugin

class NewsBlurPlugin(implugin.BitlBeeIMPlugin):
        NAME = "newsblur"
        ACCOUNT_FLAGS = 0
        SETTINGS = {
                "backlog": {
                        "default": 20,
                        "type": "int",
                },
        }
        
        BASE_URL = "https://newsblur.com"
        
        def url(self, path):
                return (self.BASE_URL + path)

        def login(self, account):
                super(NewsBlurPlugin, self).login(account)
                self.ua = requests.Session()
                creds = {"username": account["user"], "password": account["pass"]}
                r = self.ua.post(self.url("/api/login"), creds)
                self.bee.log("You're running BitlBee %d.%d.%d" % self.bitlbee_version)
                if r.status_code != 200:
                        self.bee.error("HTTP error %d" % r.status_code)
                        self.bee.logout(True)
                elif r.json()["errors"]:
                        self.bee.error("Authentication error")
                        self.bee.logout(False)
                else:
                        self.bee.add_buddy("rss", None)
                        self.bee.connected()
                        self.seen_hashes = set()
                        self.keepalive()

        def logout(self):
                self.bee.log("Ok bye!")

        def buddy_msg(self, handle, msg, flags):
                feed = self.feeds[handle]
                cmd = re.split(r"\s+", msg)
        
        def set_set_backlog(self, setting, value):
                print "Setting %s changed to %r" % (setting, value)
        
        # BitlBee will call us here every minute which is actually a neat way
        # to get periodic work (like RSS polling) scheduled. :-D
        def keepalive(self):
                r = self.ua.post(
                        self.url("/reader/unread_story_hashes"),
                        {"include_timestamps": True})
                if r.status_code != 200:
                        self.bee.error("HTTP error %d" % r.status_code)
                        return

                # Throw all unread-post hashes in a flat list and sort it by posting time.
                feed_hashes = r.json()["unread_feed_story_hashes"]
                all_hashes = []
                for feed, hashes in feed_hashes.iteritems():
                        all_hashes += [tuple(h) for h in hashes]
                all_hashes.sort(key=operator.itemgetter(1))
                
                # Look at the most recent posts, grab the ones we haven't shown yet.
                req_hashes = []
                for hash, _ in all_hashes[-self.setting("backlog"):]:
                        if hash not in self.seen_hashes:
                                req_hashes.append(hash)
                
                if not req_hashes:
                        return
                
                # Grab post details.
                r = self.ua.post(self.url("/reader/river_stories"), {"h": req_hashes})
                if r.status_code != 200:
                        self.bee.error("HTTP error %d" % r.status_code)
                        return
                
                # Response is not in the order we requested. :-( Make it a hash
                # and reconstruct order from our request.
                stories = {s["story_hash"]: s for s in r.json()["stories"]}
                for s in (stories[hash] for hash in req_hashes):
                        line = "%(story_title)s <%(story_permalink)s>" % s
                        ts = int(s.get("story_timestamp", "0"))
                        self.bee.buddy_msg("rss", line, 0, ts)
                        self.seen_hashes.add(s["story_hash"])


implugin.RunPlugin(NewsBlurPlugin, debug=True)