From 3ec82c9f4c653e0ad3d90389d4f5e698d873ff91 Mon Sep 17 00:00:00 2001
From: Romain Bignon <romain@symlink.me>
Date: Tue, 15 Mar 2011 16:33:20 +0100
Subject: [PATCH] ability to add a tag on a content

---
 weboob/backends/dlfp/browser.py    | 29 ++++++++++++++++++++++++++---
 weboob/backends/dlfp/pages/news.py | 12 ++++++++++++
 2 files changed, 38 insertions(+), 3 deletions(-)

diff --git a/weboob/backends/dlfp/browser.py b/weboob/backends/dlfp/browser.py
index 8f3e581b..ee99cd54 100644
--- a/weboob/backends/dlfp/browser.py
+++ b/weboob/backends/dlfp/browser.py
@@ -22,7 +22,7 @@ from weboob.tools.browser import BaseBrowser, BrowserHTTPError, BrowserIncorrect
 from weboob.capabilities.messages import CantSendMessage
 
 from .pages.index import IndexPage, LoginPage
-from .pages.news import ContentPage, NewCommentPage, NodePage, CommentPage
+from .pages.news import ContentPage, NewCommentPage, NodePage, CommentPage, NewTagPage
 from .pages.board import BoardIndexPage
 from .tools import id2url, url2id
 
@@ -37,6 +37,7 @@ class DLFP(BaseBrowser):
              'https://linuxfr.org/nodes/(\d+)/comments/(\d+)$': CommentPage,
              'https://linuxfr.org/nodes/(\d+)/comments/nouveau': NewCommentPage,
              'https://linuxfr.org/nodes/(\d+)/comments$': NodePage,
+             'https://linuxfr.org/nodes/(\d+)/tags/nouveau': NewTagPage,
              'https://linuxfr.org/board/index.xml': BoardIndexPage,
             }
 
@@ -45,16 +46,25 @@ class DLFP(BaseBrowser):
     def home(self):
         return self.location('https://linuxfr.org')
 
-    def get_content(self, _id):
+    def parse_id(self, _id):
         url = id2url(_id)
         if url is None:
             if url2id(_id) is not None:
                 url = _id
                 _id = url2id(url)
             else:
-                return None
+                return None, None
+
+        return url, _id
+
+    def get_content(self, _id):
+        url, _id = self.parse_id(_id)
+
+        if url is None:
+            return None
 
         self.location(url)
+        assert self.is_on_page(ContentPage)
         content = self.page.get_article()
         content.id = _id
         return content
@@ -153,3 +163,16 @@ class DLFP(BaseBrowser):
                                      urllib.urlencode({'board[message]': msg}),
                                      {'Referer': self.absurl('/')})
         self.readurl(request)
+
+    def add_tag(self, _id, tag):
+        url, _id = self.parse_id(_id)
+        if url is None:
+            return None
+
+        self.location(url)
+        assert self.is_on_page(ContentPage)
+
+        self.location(self.page.get_tag_url())
+        assert self.is_on_page(NewTagPage)
+
+        self.page.tag(tag)
diff --git a/weboob/backends/dlfp/pages/news.py b/weboob/backends/dlfp/pages/news.py
index 898f5c12..295bf72b 100644
--- a/weboob/backends/dlfp/pages/news.py
+++ b/weboob/backends/dlfp/pages/news.py
@@ -151,9 +151,21 @@ class ContentPage(DLFPPage):
     def get_post_comment_url(self):
         return select(self.document.getroot(), 'p#send-comment', 1).find('a').attrib['href']
 
+    def get_tag_url(self):
+        return select(self.document.getroot(), 'div.tag_in_place', 1).find('a').attrib['href']
+
 class NewCommentPage(DLFPPage):
     pass
 
+class NewTagPage(DLFPPage):
+    def _is_tag_form(self, form):
+        return form.action.endswith('/tags')
+
+    def tag(self, tag):
+        self.browser.select_form(predicate=self._is_tag_form)
+        self.browser['tags'] = tag
+        self.browser.submit()
+
 class NodePage(DLFPPage):
     def get_errors(self):
         try: