Revision: 7739
Author: alexsh
Date: 2009-12-07 17:09:36 +0000 (Mon, 07 Dec 2009)
Log Message:
-----------
*API upload back (confirmed by imagetransfer.py)
* add multiple warning handle
* add session key after ignore warning, will not re-upload file contents again.
Modified Paths:
--------------
trunk/pywikipedia/upload.py
Modified: trunk/pywikipedia/upload.py
===================================================================
--- trunk/pywikipedia/upload.py 2009-12-07 07:24:45 UTC (rev 7738)
+++ trunk/pywikipedia/upload.py 2009-12-07 17:09:36 UTC (rev 7739)
@@ -214,18 +214,18 @@
self.description = newDescription
return filename
- def upload_image(self, debug=False):
+ def upload_image(self, debug=False, sessionKey = 0):
"""Gets the image at URL self.url, and uploads it to the target wiki.
Returns the filename which was used to upload the image.
If the upload fails, the user is asked whether to try again or not.
If the user chooses not to retry, returns null.
"""
try:
- #if config.use_api and self.targetSite.versionnumber() >= 16:
- # x = self.targetSite.api_address()
- # del x
- #else:
- raise NotImplementedError
+ if config.use_api and self.targetSite.versionnumber() >= 16:
+ x = self.targetSite.api_address()
+ del x
+ else:
+ raise NotImplementedError
except NotImplementedError:
return self._uploadImageOld(debug)
@@ -241,9 +241,11 @@
'filename': filename,
#'': '',
}
+ if sessionKey:
+ params['sessionkey'] = sessionKey
if self.uploadByUrl:
params['url'] = self.url
- else:
+ elif not self.uploadByUrl and not sessionKey:
params['file'] = self._contents
if self.ignoreWarning:
@@ -262,28 +264,27 @@
else:
data = data['upload']
if data['result'] == u'Warning': #upload success but return
warning.
- warn = data['warnings'].keys()[0]
- wikipedia.output("We got a warning message:", newline=False)
- warFn = data['warnings'][warn]
- if warn == 'duplicate-archive':
- wikipedia.output("The file is duplicate a deleted file
%s." % warFn)
- elif warn == 'was-deleted':
- wikipedia.output("This file was deleted for %s." % warFn)
- elif warn == 'emptyfile':
- wikipedia.output("File %s is an empty file." % warFn)
- elif warn == 'exists':
- wikipedia.output("File %s is exists." % warFn)
- elif warn == 'duplicate':
- wikipedia.output("Uploaded file is duplicate with %s." %
warFn)
- elif warn == 'badfilename':
- wikipedia.output("Target filename is invaild.")
- elif warn == 'filetype-unwanted-type':
- wikipedia.output("File %s type is unwatched type." % warFn)
+ wikipedia.output("Got warning message:")
+ for k,v in data['warnings'].iteritems():
+ if k == 'duplicate-archive':
+ wikipedia.output("\tThe file is duplicate a deleted
file %s." % v)
+ elif k == 'was-deleted':
+ wikipedia.output("\tThis file was deleted for %s." % v)
+ elif k == 'emptyfile':
+ wikipedia.output("\tFile %s is an empty file." % v)
+ elif k == 'exists':
+ wikipedia.output("\tFile %s is exists." % v)
+ elif k == 'duplicate':
+ wikipedia.output("\tUploaded file is duplicate with
%s." % v)
+ elif k == 'badfilename':
+ wikipedia.output("\tTarget filename is invalid.")
+ elif k == 'filetype-unwanted-type':
+ wikipedia.output("\tFile %s type is unwatched type." %
v)
answer = wikipedia.inputChoice(u"Do you want to ignore?",
['Yes', 'No'], ['y', 'N'], 'N')
if answer == "y":
self.ignoreWarning = 1
self.keepFilename = True
- return self.upload_image(debug)
+ return self.upload_image(debug, sessionKey =
data['sessionkey'])
else:
wikipedia.output("Upload aborted.")
return
_______________________________________________
Pywikipedia-svn mailing list
[email protected]
https://lists.wikimedia.org/mailman/listinfo/pywikipedia-svn
