Commit: b5561a04bfec320b5c99de8c76e25c0e7d05be85 - sfjplib (git)

sfjplib for python

Commit MetaInfo

• Révision: b5561a04bfec320b5c99de8c76e25c0e7d05be85 (tree)
• l'heure: 2011-08-25 20:33:18
• Auteur: Hiromichi MATSUSHIMA <hirom@offi...>
• Commiter: Hiromichi MATSUSHIMA

Message de Log

add Wiki support

Change Summary

• modified: crawlerlib.py
• modified: sfjplib.py

Modification

--- a/crawlerlib.py
+++ b/crawlerlib.py
@@ -63,7 +63,7 @@ class Crawler(object):
         p = urllib2.HTTPCookieProcessor(c)
         opener = urllib2.build_opener(p)
 
-        res = opener.open(url, params)
+        res = opener.open(url, urllib.urlencode(params))
         self._set_cookie(c)
         return res
 

--- a/sfjplib.py
+++ b/sfjplib.py
@@ -4,6 +4,8 @@
 
 import crawlerlib
 import urllib
+import htmltree
+import re
 
 def login(uname, passwd):
     """
@@ -51,4 +53,47 @@ class docman2(object):
         return res.read()
 
 
-    
+class Wiki(object):
+    """Wiki manipulation library"""
+    def __init__(self, sfjp_user = None):
+        self._user = sfjp_user
+
+    def retrive_wikitext(self, project_uid, name):
+        c = crawlerlib.Crawler(self._user)
+        url = "http://sourceforge.jp/projects/%s/wiki/%s?action=edit" % (project_uid, name)
+        res = c.get(url)
+        html = res.read()
+
+        tree = htmltree.parse(html).root()
+        title = tree.get_element_by_id("title").attr("value")
+        wikitext = tree.get_element_by_id("text").inner_text()
+        comment = tree.get_element_by_id("comment").attr("value")
+
+        # get postkey
+    #document.write('<inp'+'ut type="hidden" name="postkey" value="Alqh7yg">');
+
+        m = re.search(r'''type="hidden" name="postkey" value="(.*?)"''', html)
+        if m:
+            postkey = m.group(1)
+        else:
+            postkey = ""
+        return (title, wikitext, comment, postkey)
+
+    def post_wikitext(self, project_uid, name, title, wikitext, comment, postkey):
+        c = crawlerlib.Crawler(self._user)
+        url = "http://sourceforge.jp/projects/%s/wiki/%s?action=update" % (project_uid, name)
+        
+        title = title.encode("utf_8")
+        wikitext = wikitext.encode("utf_8")
+        comment = comment.encode("utf_8")
+
+        params = {
+            "title": title,
+            "textarea_height": 24,
+            "text": wikitext,
+            "comment": comment,
+            "postkey": postkey
+            }
+        res = c.post_form(url, params)
+        return res
+