Материал из Википедии — свободной энциклопедии
# -*- coding: utf-8 -*-
"""
Script to upload images to wikipedia.
Arguments:
-keep Keep the filename as is
-noverify Do not ask for verification of the upload description if one is given
If any other arguments are given, the first is the URL or filename
to upload, and the rest is a proposed description to go with the
upload. If none of these are given, the user is asked for the
file or URL to upload. The bot will then upload the image to the wiki.
The script will ask for the location of an image, if not given as a parameter,
and for a description.
"""
#
# (C) Rob W.W. Hooft, Andre Engels 2003-2004
#
# Distributed under the terms of the MIT license.
#
__version__='$Id: upload.py,v 1.36 2006/03/25 08:45:46 a_engels Exp $'
import os, sys, re
import urllib, httplib
import wikipedia, config, mediawiki_messages
def post_multipart(host, selector, fields, files, cookies):
"""
Post fields and files to an http host as multipart/form-data.
fields is a sequence of (name, value) elements for regular form fields.
files is a sequence of (name, filename, value) elements for data to be uploaded as files
Return the server's response page.
"""
content_type, body = encode_multipart_formdata(fields, files)
conn = httplib.HTTPConnection(host)
conn.putrequest('POST', selector)
conn.putheader('content-type', content_type)
conn.putheader('content-length', str(len(body)))
conn.putheader("User-agent", wikipedia.useragent)
if cookies:
conn.putheader('Cookie',cookies)
conn.endheaders()
conn.send(body)
response = conn.getresponse()
returned_html = response.read()
conn.close()
return response, returned_html
def encode_multipart_formdata(fields, files):
"""
fields is a sequence of (name, value) elements for regular form fields.
files is a sequence of (name, filename, value) elements for data to be uploaded as files
Return (content_type, body) ready for httplib.HTTP instance
"""
BOUNDARY = '----------ThIs_Is_tHe_bouNdaRY_$'
CRLF = '\r\n'
L = []
for (key, value) in fields:
L.append('--' + BOUNDARY)
L.append('Content-Disposition: form-data; name="%s"' % key)
L.append('')
L.append(value)
for (key, filename, value) in files:
L.append('--' + BOUNDARY)
L.append('Content-Disposition: form-data; name="%s"; filename="%s"' % (key, filename))
L.append('Content-Type: %s' % get_content_type(filename))
L.append('')
L.append(value)
L.append('--' + BOUNDARY + '--')
L.append('')
body = CRLF.join(L)
content_type = 'multipart/form-data; boundary=%s' % BOUNDARY
return content_type, body
def get_content_type(filename):
import mimetypes
return mimetypes.guess_type(filename)[0] or 'application/octet-stream'
class UploadRobot:
def __init__(self, url, description = u'', keepFilename = False, verifyDescription = True, ignoreWarning = False, targetSite = None, urlEncoding = None):
"""
ignoreWarning - Set this to True if you want to upload even if another
file would be overwritten or another mistake would be
risked.
Attention: This parameter doesn't work yet for unknown reason.
"""
self.url = url
self.urlEncoding = urlEncoding
self.description = description
self.keepFilename = keepFilename
self.verifyDescription = verifyDescription
self.ignoreWarning = ignoreWarning
if config.upload_to_commons:
self.targetSite = targetSite or wikipedia.getSite('commons', 'commons')
else:
self.targetSite = targetSite or wikipedia.getSite()
self.targetSite.forceLogin()
def urlOK(self):
'''
Returns true iff the URL references an online site or an
existing local file.
'''
return self.url != '' and ('://' in self.url or os.path.exists(self.url))
def upload_image(self, debug=False):
"""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 u''.
"""
# Get file contents
if '://' in self.url:
uo = wikipedia.MyURLopener()
file = uo.open(self.url,"rb")
else:
# Opening local files with MyURLopener would be possible, but we
# don't do it because it only accepts ASCII characters in the
# filename.
file = open(self.url,"rb")
wikipedia.output(u'Reading file %s' % self.url)
contents = file.read()
if contents.find("The requested URL was not found on this server.") != -1:
print "Couldn't download the image."
return u''
file.close()
# Isolate the pure name
filename = self.url
if '/' in filename:
filename = filename.split('/')[-1]
if '\\' in filename:
filename = filename.split('\\')[-1]
if self.urlEncoding:
filename = urllib.unquote(filename)
filename = filename.decode(self.urlEncoding)
# MediaWiki doesn't allow spaces in the file name.
# Replace them here to avoid an extra confirmation form
filename = filename.replace(' ', '_')
# Convert the filename (currently Unicode) to the encoding used on the
# target wiki
encodedFilename = filename.encode(self.targetSite.encoding())
# A proper description for the submission.
wikipedia.output(u"The suggested description is:")
wikipedia.output(self.description)
if self.verifyDescription:
newDescription = u''
choice = wikipedia.inputChoice(u'Do you want to change this description?', ['Yes', 'No'], ['y', 'N'], 'n')
if choice == 'y':
import editarticle
editor = editarticle.TextEditor()
newDescription = editor.edit(self.description)
# if user saved / didn't press Cancel
if newDescription:
self.description = newDescription
formdata = {}
formdata["wpUploadDescription"] = self.description
# if self.targetSite.version() >= '1.5':
# formdata["wpUploadCopyStatus"] = wikipedia.input(u"Copyright status: ")
# formdata["wpUploadSource"] = wikipedia.input(u"Source of image: ")
formdata["wpUploadAffirm"] = "1"
formdata["wpUpload"] = "upload bestand"
# This somehow doesn't work.
if self.ignoreWarning:
formdata["wpIgnoreWarning"] = "1"
else:
formdata["wpIgnoreWarning"] = "0"
# try to encode the strings to the encoding used by the target site.
# if that's not possible (e.g. because there are non-Latin-1 characters and
# the home Wikipedia uses Latin-1), convert all non-ASCII characters to
# HTML entities.
for key in formdata:
assert isinstance(key, basestring), "ERROR: %s is not a string but %s" % (key, type(key))
try:
formdata[key] = formdata[key].encode(self.targetSite.encoding())
except (UnicodeEncodeError, UnicodeDecodeError):
formdata[key] = wikipedia.UnicodeToAsciiHtml(formdata[key]).encode(self.targetSite.encoding())
# don't upload if we're in debug mode
if not debug:
wikipedia.output(u'Uploading file to %s...' % self.targetSite)
response, returned_html = post_multipart(self.targetSite.hostname(),
self.targetSite.upload_address(),
formdata.items(),
(('wpUploadFile', encodedFilename, contents),),
cookies = self.targetSite.cookies()
)
returned_html = returned_html.decode(self.targetSite.encoding())
# There are 2 ways MediaWiki can react on success: either it gives
# a 200 with a success message, or it gives a 302 (redirection).
# Do we know how the "success!" HTML page should look like?
# ATTENTION: if you changed your Wikimedia Commons account not to show
# an English interface, this detection will fail!
success_msg = mediawiki_messages.get('successfulupload', site = self.targetSite)
if success_msg in returned_html or response.status == 302:
wikipedia.output(u"Upload successful.")
# The following is not a good idea, because the server also gives a 200 when
# something went wrong.
#if response.status in [200, 302]:
# wikipedia.output(u"Upload successful.")
else:
return u''
return filename
def run(self):
while not self.urlOK():
if not self.url:
wikipedia.output(u'No input filename given')
else:
wikipedia.output(u'Invalid input filename given. Try again.')
self.url = wikipedia.input(u'File or URL where image is now:')
return self.upload_image()
def main(args):
url = u''
description = []
keepFilename = False
verifyDescription = True
# call wikipedia.py function to process all global wikipedia args
# returns a list of non-global args, i.e. args for upload.py
args = wikipedia.handleArgs()
for arg in args:
if arg:
if arg.startswith('-keep'):
keepFilename = True
elif arg.startswith('-noverify'):
verifyDescription = False
elif url == u'':
url = arg
else:
description.append(arg)
description = u' '.join(description)
bot = UploadRobot(url, description, keepFilename, verifyDescription)
bot.run()
if __name__ == "__main__":
try:
main(sys.argv[1:])
finally:
wikipedia.stopme()
