new method get_comment

This commit is contained in:
Romain Bignon 2011-03-07 08:47:17 +01:00
commit 73b6b1e655
2 changed files with 13 additions and 8 deletions

View file

@ -103,13 +103,7 @@ class DLFP(BaseBrowser):
def close_session(self):
self.openurl('/compte/deconnexion')
def plusse(self, url):
return self.relevance(url, 'for')
def moinse(self, url):
return self.relevance(url, 'against')
def relevance(self, url, what):
def get_comment(self, url):
self.location(url)
comment = None
@ -119,6 +113,17 @@ class DLFP(BaseBrowser):
ignored, id = url.rsplit('#comment-', 1)
comment = self.page.get_comment(int(id))
return comment
def plusse(self, url):
return self.relevance(url, 'for')
def moinse(self, url):
return self.relevance(url, 'against')
def relevance(self, url, what):
comment = self.get_comment(url)
if comment is None:
raise ValueError('The given URL isn\'t a comment.')

View file

@ -83,7 +83,7 @@ class Article(object):
self.date = None
self.comments = []
if not tree:
if tree is None:
return
header = tree.find('header')