Commit 4065f7ef authored by Romain Bignon's avatar Romain Bignon

renamed dlfp2mail to weboob

parent 7906433c
# -*- coding: utf-8 -*-
"""
Copyright(C) 2010 Romain Bignon
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, version 3 of the License.
This program 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 General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
"""
class DLFPUnavailable(Exception):
pass
class DLFPBanned(DLFPUnavailable):
pass
class DLFPIncorrectPassword(Exception):
pass
class DLFPRetry(Exception):
pass
......@@ -18,21 +18,23 @@ Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
"""
from mechanize import FormNotFoundError
from weboob.tools.browser import Browser
from weboob.backends.dlfp.pages.index import IndexPage, LoginPage
class BasePage:
def __init__(self, dlfp, document, url=''):
self.dlfp = dlfp
self.document = document
self.url = url
class DLFP(Browser):
def loaded(self):
pass
DOMAIN = 'linuxfr.org'
PAGES = {'https://linuxfr.org/': IndexPage,
'https://linuxfr.org/pub/': IndexPage,
'https://linuxfr.org/my/': IndexPage,
'https://linuxfr.org/login.html': LoginPage,
}
def isLogged(self):
forms = self.document.getElementsByTagName('form')
for form in forms:
if form.getAttribute('id') == 'formulaire':
return False
def home(self):
return self.location('https://linuxfr.org')
def login(self):
self.location('/login.html', 'login=%s&passwd=%s&isauto=1' % (self.username, self.password))
return True
def isLogged(self):
return (self.page and self.page.isLogged())
......@@ -24,14 +24,15 @@ import re
class Article:
RSS = None
def __init__(self, _id, title, author, datetime):
def __init__(self, _id, url, title, author, datetime):
self._id = _id
self.url = url
self.title = title
self.author = author
self.datetime = datetime
class Newspaper(Article):
RSS = 'http://linuxfr.org/backend/news/rss20.rss'
RSS = 'https://linuxfr.org/backend/news/rss20.rss'
class Telegram(Article):
RSS = 'https://linuxfr.org/backend/journaux/rss20.rss'
......@@ -58,5 +59,5 @@ class ArticlesList:
warning('Unable to parse ID from link \'%s\'' % item['link'])
continue
_id = m.group(1)
article = klass(_id, item['title'], item['author'], item['date_parsed'])
article = klass(_id, item['link'], item['title'], item['author'], item['date_parsed'])
print _id, item['author'], item['title']
......@@ -18,12 +18,26 @@ Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
"""
from dlfp.pages.base import BasePage
from weboob.tools.browser import BrowserIncorrectPassword, BasePage
class IndexPage(BasePage):
class DLFPPage(BasePage):
def isLogged(self):
forms = self.document.getElementsByTagName('form')
for form in forms:
if form.getAttribute('id') == 'formulaire':
return False
return True
class IndexPage(DLFPPage):
pass
class LoginPage(BasePage):
class LoginPage(DLFPPage):
def loaded(self):
if self.hasError():
raise BrowserIncorrectPassword()
def hasError(self):
plist = self.document.getElementsByTagName('p')
for p in plist:
......
......@@ -18,7 +18,8 @@ Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
"""
from mechanize import Browser, response_seek_wrapper, BrowserStateError
import mechanize.Browser
from mechanize import response_seek_wrapper, BrowserStateError
import urllib2
import html5lib
from html5lib import treebuilders
......@@ -27,9 +28,16 @@ import time
from logging import warning, error
from copy import copy
from dlfp.pages.login import IndexPage, LoginPage
from dlfp.exceptions import DLFPIncorrectPassword, DLFPUnavailable, DLFPRetry
from dlfp.firefox_cookies import FirefoxCookieJar
from weboob.tools.firefox_cookies import FirefoxCookieJar
class BrowserIncorrectPassword(Exception):
pass
class BrowserUnavailable(Exception):
pass
class BrowserRetry(Exception):
pass
class NoHistory:
def __init__(self): pass
......@@ -38,48 +46,67 @@ class NoHistory:
def clear(self): pass
def close(self): pass
class DLFP(Browser):
class BasePage:
def __init__(self, browser, document, url=''):
self.browser = browser
self.document = document
self.url = url
def loaded(self):
pass
class Browser(mechanize.Browser):
# ------ Class attributes --------------------------------------
DOMAIN = None
PAGES = {}
USER_AGENT = 'Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.9.0.4) Gecko/2008111318 Ubuntu/8.10 (intrepid) Firefox/3.0.3'
# ------ Abstract methods --------------------------------------
# Go to home
def home(self):
raise NotImplementedError()
# Login to the website
def login(self):
raise NotImplementedError()
# Return True if we are logged on website
def isLogged(self):
raise NotImplementedError()
pages = {'https://linuxfr.org/': IndexPage,
'https://linuxfr.org/pub/': IndexPage,
'https://linuxfr.org/my/': IndexPage,
'https://linuxfr.org/login.html': LoginPage,
}
# ------ Browser methods ---------------------------------------
def __init__(self, username, password=None, firefox_cookies=None):
Browser.__init__(self, history=NoHistory())
mechanize.Browser.__init__(self, history=NoHistory())
self.addheaders = [
['User-agent', 'Mozilla/5.0 (X11; U; Linux i686; en-US; rv:1.9.0.4) Gecko/2008111318 Ubuntu/8.10 (intrepid) Firefox/3.0.3']
['User-agent', self.USER_AGENT]
]
# Share cookies with firefox
if firefox_cookies:
self.__cookie = FirefoxCookieJar(firefox_cookies)
self.__cookie = FirefoxCookieJar(self.DOMAIN, firefox_cookies)
self.__cookie.load()
self.set_cookiejar(self.__cookie)
else:
self.__cookie = None
self.__parser = html5lib.HTMLParser(tree=treebuilders.getTreeBuilder("dom"))
self.__page = None
self.__last_update = 0.0
self.page = None
self.last_update = 0.0
self.username = username
self.password = password
if self.password:
try:
self.home()
except DLFPUnavailable:
except BrowserUnavailable:
pass
def page(self):
return self.__page
def home(self):
return self.location('https://linuxfr.org')
def pageaccess(func):
def inner(self, *args, **kwargs):
if not self.__page or not self.__page.isLogged() and self.password:
if not self.page or not self.page.isLogged() and self.password:
self.home()
return func(self, *args, **kwargs)
......@@ -89,57 +116,65 @@ class DLFP(Browser):
def keepalive(self):
self.home()
def login(self):
self.location('/login.html', 'login=%s&passwd=%s&isauto=1' % (self.username, self.password))
def change_location(func):
def inner(self, *args, **kwargs):
if args and args[0][0] == '/' and (not self.request or self.request.host != self.DOMAIN):
args = ('%s://%s%s' % (self.PROTOCOL, self.DOMAIN, args[0]),) + args[1:]
print args
return func(self, *args, **kwargs)
return inner
@change_location
def openurl(self, *args, **kwargs):
try:
return Browser.open(self, *args, **kwargs)
return mechanize.Browser.open(self, *args, **kwargs)
except (response_seek_wrapper, urllib2.HTTPError, urllib2.URLError), e:
error(e)
raise DLFPUnavailable()
raise BrowserUnavailable()
except BrowserStateError:
self.home()
return Browser.open(self, *args, **kwargs)
return mechanize.Browser.open(self, *args, **kwargs)
def submit(self, *args, **kwargs):
try:
self.__changeLocation(Browser.submit(self, *args, **kwargs))
self.__changeLocation(mechanize.Browser.submit(self, *args, **kwargs))
except (response_seek_wrapper, urllib2.HTTPError, urllib2.URLError), e:
error(e)
self.__page = None
raise DLFPUnavailable()
except (BrowserStateError,DLFPRetry):
self.page = None
raise BrowserUnavailable()
except (BrowserStateError,BrowserRetry):
self.home()
raise DLFPUnavailable()
raise BrowserUnavailable()
def isOnPage(self, pageCls):
return isinstance(self.__page, pageCls)
return isinstance(self.page, pageCls)
def follow_link(self, *args, **kwargs):
try:
self.__changeLocation(Browser.follow_link(self, *args, **kwargs))
self.__changeLocation(mechanize.Browser.follow_link(self, *args, **kwargs))
except (response_seek_wrapper, urllib2.HTTPError, urllib2.URLError), e:
error(e)
self.__page = None
raise DLFPUnavailable()
except (BrowserStateError,DLFPRetry):
self.page = None
raise BrowserUnavailable()
except (BrowserStateError,BrowserRetry):
self.home()
raise DLFPUnavailable()
raise BrowserUnavailable()
@change_location
def location(self, *args, **kwargs):
keep_args = copy(args)
keep_kwargs = kwargs.copy()
try:
self.__changeLocation(Browser.open(self, *args, **kwargs))
except DLFPRetry:
if not self.__page or not args or self.__page.url != args[0]:
self.__changeLocation(mechanize.Browser.open(self, *args, **kwargs))
except BrowserRetry:
if not self.page or not args or self.page.url != args[0]:
self.location(keep_args, keep_kwargs)
except (response_seek_wrapper, urllib2.HTTPError, urllib2.URLError), e:
error(e)
self.__page = None
raise DLFPUnavailable()
self.page = None
raise BrowserUnavailable()
except BrowserStateError:
self.home()
self.location(*keep_args, **keep_kwargs)
......@@ -147,7 +182,7 @@ class DLFP(Browser):
def __changeLocation(self, result):
# Find page from url
pageCls = None
for key, value in self.pages.items():
for key, value in self.PAGES.items():
regexp = re.compile('^%s$' % key)
m = regexp.match(result.geturl())
if m:
......@@ -156,7 +191,7 @@ class DLFP(Browser):
# Not found
if not pageCls:
self.__page = None
self.page = None
r = result.read()
if isinstance(r, unicode):
r = r.encode('iso-8859-15', 'replace')
......@@ -165,18 +200,13 @@ class DLFP(Browser):
return
print '[%s] Gone on %s' % (self.username, result.geturl())
self.__last_update = time.time()
self.last_update = time.time()
document = self.__parser.parse(result, encoding='iso-8859-1')
self.__page = pageCls(self, document, result.geturl())
self.__page.loaded()
# Special pages
if isinstance(self.__page, LoginPage):
if self.__page.hasError():
raise DLFPIncorrectPassword()
raise DLFPRetry()
if not self.__page.isLogged() and self.password:
self.page = pageCls(self, document, result.geturl())
self.page.loaded()
if not self.isLogged() and self.password:
print '!! Relogin !!'
self.login()
return
......
......@@ -31,11 +31,10 @@ from mechanize import CookieJar, Cookie
#logger.setLevel(logging.DEBUG)
class FirefoxCookieJar(CookieJar):
def __init__(self, sqlite_file=None, policy=None):
def __init__(self, domain, sqlite_file=None, policy=None):
CookieJar.__init__(self, policy)
self.domain = domain
self.sqlite_file = sqlite_file
def __connect(self):
......@@ -47,7 +46,6 @@ class FirefoxCookieJar(CookieJar):
return db
def load(self):
db = self.__connect()
......@@ -55,7 +53,7 @@ class FirefoxCookieJar(CookieJar):
cookies = db.execute("""SELECT host, path, name, value, expiry, lastAccessed, isSecure
FROM moz_cookies
WHERE host LIKE '%linuxfr%'""")
WHERE host LIKE '%%%s%%'""" % self.domain)
for entry in cookies:
......@@ -91,7 +89,7 @@ class FirefoxCookieJar(CookieJar):
db = self.__connect()
if not db: return
db.execute("DELETE FROM moz_cookies WHERE host LIKE '%linuxfr%'")
db.execute("DELETE FROM moz_cookies WHERE host LIKE '%%%s%%'" % self.domain)
for cookie in self:
if cookie.secure: secure = 1
else: secure = 0
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment