提交: 81e075a1d916532ec67ee60b93bd2f0dac8c3e2d - sfjpmag2ools (git)

maintaince tools for sfjp magazine

Commit MetaInfo

• 修订版: 81e075a1d916532ec67ee60b93bd2f0dac8c3e2d (tree)
• 时间: 2013-05-01 20:28:17
• 作者: hylom <hylom@hylo...>
• Commiter: hylom

Log Message

add: upload_media method

更改概述

• modified: sfjpmagclient/sfjpmagclient.py
• add: test/sample.png
• modified: test/test_sfjpmagclient.py
• add: test/test_upload_media.py

差异

--- a/sfjpmagclient/sfjpmagclient.py
+++ b/sfjpmagclient/sfjpmagclient.py
@@ -7,14 +7,17 @@ import sys
 HOST_PATH = 'magazine-admin.sourceforge.jp/magazine/xmlrpc.php'
 
 class MagClient(object):
-    def __init__(self, username, password, auth_user="", auth_pass=""):
+    def __init__(self, username, password, auth_user="", auth_pass="", endpoint=None):
         self.username = username
         self.password = password
+        if endpoint == None:
+            endpoint = HOST_PATH
+
         if auth_user and auth_pass:
-#            self.uri = "http://{username}:{password}@{url}".format(username=auth_user, password=auth_pass, url=HOST_PATH)
-            self.uri = "http://%s:%s@%s" % (auth_user, auth_pass, HOST_PATH)
+#            self.uri = "http://{username}:{password}@{url}".format(username=auth_user, password=auth_pass, url=endpoint)
+            self.uri = "http://%s:%s@%s" % (auth_user, auth_pass, endpoint)
         else:
-            self.uri = "http://" + HOST_PATH
+            self.uri = "http://" + endpoint
 
     def _get_blog_id(self):
         try:
@@ -42,6 +45,23 @@ class MagClient(object):
                                 self.password,
                                 content)
 
+    def upload_media(self, filename, data, mimetype, overwrite=True, parent=None):
+        proxy = self._get_proxy()
+        post_data = {
+            "name": filename,
+            "type": mimetype,
+            "bits": xmlrpclib.Binary(data),
+            "overwrite": overwrite
+            }
+
+        if parent != None:
+            post_data["post_id"] = parent
+
+        return proxy.wp.uploadFile(self._get_blog_id(),
+                                   self.username,
+                                   self.password,
+                                   post_data)
+
     def get_users_blogs(self):
         proxy = self._get_proxy()
         return proxy.wp.getUsersBlogs(self.username, self.password)

Binary files /dev/null and b/test/sample.png differ

--- a/test/test_sfjpmagclient.py
+++ b/test/test_sfjpmagclient.py
@@ -9,10 +9,11 @@ from sfjpmagclient import MagClient
 
 username = config["username"]
 password = config["password"]
-auth_user = config["auth_user"]
-auth_pass = config["auth_pass"]
+auth_user = config.get("auth_user", "")
+auth_pass = config.get("auth_pass", "")
+endpoint = config["endpoint"]
 
-c = MagClient(username, password, auth_user, auth_pass)
+c = MagClient(username, password, auth_user, auth_pass, endpoint)
 ret = c.get_users_blogs()
 print ret
 

--- /dev/null
+++ b/test/test_upload_media.py
@@ -0,0 +1,36 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+import json
+import sys
+import xmlrpclib
+sys.path.append('.')
+
+from testconfig import config
+from sfjpmagclient import MagClient
+
+username = config["username"]
+password = config["password"]
+auth_user = config.get("auth_user", "")
+auth_pass = config.get("auth_pass", "")
+endpoint = config["endpoint"]
+
+POST_ID = 1
+TEST_FILE = "test/sample.png"
+
+c = MagClient(username, password, auth_user, auth_pass, endpoint)
+
+f = open(TEST_FILE, "r")
+test_data = f.read()
+f.close()
+
+try:
+    ret = c.upload_media("This_is_test.png", test_data, "image/png",
+                     True, 1)
+except xmlrpclib.Fault, e:
+    for k in e.__dict__:
+        print k, e.__getattribute__(k)
+    sys.exit(-1)
+print ret
+
+