weboob-devel/modules/feedly/browser.py

103 lines
3.8 KiB
Python

# -*- coding: utf-8 -*-
# Copyright(C) 2014 Bezleputh
#
# This file is part of weboob.
#
# weboob is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# weboob is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with weboob. If not, see <http://www.gnu.org/licenses/>.
from weboob.tools.json import json
from weboob.capabilities.collection import Collection
from weboob.browser2 import LoginBrowser, URL, need_login
from .pages import EssentialsPage, TokenPage, ContentsPage, PreferencesPage
__all__ = ['FeedlyBrowser']
class FeedlyBrowser(LoginBrowser):
BASEURL = 'http://www.feedly.com'
essentials = URL('http://s3.feedly.com/essentials/essentials_fr.json', EssentialsPage)
token = URL('v3/auth/token', TokenPage)
contents = URL('v3/streams/contents', ContentsPage)
preferences = URL('v3/preferences', PreferencesPage)
marker = URL('v3/markers')
def __init__(self, username, password, login_browser, *args, **kwargs):
super(FeedlyBrowser, self).__init__(username, password, *args, **kwargs)
self.login_browser = login_browser
self.user_id = None
def do_login(self):
if self.login_browser.code is None or self.user_id is None:
self.login_browser.do_login()
params = {'code': self.login_browser.code,
'client_id': 'feedly',
'client_secret': '0XP4XQ07VVMDWBKUHTJM4WUQ',
'redirect_uri': 'http://dev.feedly.com/feedly.html',
'grant_type': 'authorization_code'}
token, self.user_id = self.token.go(data=params).get_token()
self.session.headers['X-Feedly-Access-Token'] = token
@need_login
def iter_threads(self):
params = {'streamId': 'user/%s/category/global.all' % self.user_id,
'unreadOnly': 'true',
'ranked': 'newest',
'count': '100'}
return self.contents.go(params=params).get_articles()
def get_unread_feed(self, url):
params = {'streamId': url,
'backfill': 'true',
'boostMustRead': 'true',
'unreadOnly': 'true'}
return self.contents.go(params=params).get_articles()
def get_categories(self):
if self.username is not None and self.password is not None:
return self.get_logged_categories()
return self.essentials.go().get_categories()
@need_login
def get_logged_categories(self):
user_categories = list(self.preferences.go().get_categories())
user_categories.append(Collection(['global.saved'], 'Saved'))
return user_categories
def get_feeds(self, category):
if self.username is not None and self.password is not None:
return self.get_logged_feeds(category)
return self.essentials.go().get_feeds(category)
@need_login
def get_logged_feeds(self, category):
if category == 'global.saved':
type = 'tag'
else:
type = 'category'
url = 'user/%s/%s/%s' % (self.user_id, type, category)
return self.get_unread_feed(url)
def get_feed_url(self, category, feed):
return self.essentials.go().get_feed_url(category, feed)
@need_login
def set_message_read(self, _id):
datas = {'action': 'markAsRead',
'type': 'entries',
'entryIds': [_id]}
self.marker.open(data=json.dumps(datas))