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

View file

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