editarticle.py
29125188e56dmaster

editarticle.py
View Options

			#!/usr/bin/python
			# -- coding: utf-8 --
			"""
			Edit a Wikipedia article with your favourite editor.

			TODO: - non existing pages
			- edit conflicts
			- minor edits
			- watch/unwatch
			- ...
			"""
			#
			# (C) Gerrit Holl, 2004
			# (C) Pywikibot team, 2004-2014
			#
			# Distributed under the terms of the MIT license.
			#
			__version__ = '$Id$'
			#

			import os
			import optparse
			import tempfile

			import pywikibot
			from pywikibot import i18n
			from pywikibot.editor import TextEditor


			class ArticleEditor(object):

			"""Edit a wiki page."""

			# join lines if line starts with this ones
			# TODO: No apparent usage
			# joinchars = string.letters + '[]' + string.digits

			def __init__(self, *args):
			self.set_options(*args)
			self.setpage()
			self.site = pywikibot.Site()

			def set_options(self, *args):
			"""Parse commandline and set options attribute."""
			my_args = []
			for arg in pywikibot.handle_args(args):
			my_args.append(arg)
			parser = optparse.OptionParser()
			parser.add_option("-r", "--edit_redirect", action="store_true",
			default=False, help="Ignore/edit redirects")
			parser.add_option("-p", "--page", help="Page to edit")
			parser.add_option("-w", "--watch", action="store_true", default=False,
			help="Watch article after edit")
			(self.options, args) = parser.parse_args(args=my_args)

			# for convenience, if we have an arg, stuff it into the opt, so we
			# can act like a normal editor.
			if (len(args) == 1):
			self.options.page = args[0]

			def setpage(self):
			"""Set page and page title."""
			site = pywikibot.Site()
			pageTitle = self.options.page or pywikibot.input(u"Page to edit:")
			self.page = pywikibot.Page(pywikibot.Link(pageTitle, site))
			if not self.options.edit_redirect and self.page.isRedirectPage():
			self.page = self.page.getRedirectTarget()

			def handle_edit_conflict(self, new):
			fn = os.path.join(tempfile.gettempdir(), self.page.title())
			fp = open(fn, 'w')
			fp.write(new)
			fp.close()
			pywikibot.output(
			u"An edit conflict has arisen. Your edit has been saved to %s. Please try again."
			% fn)

			def run(self):
			self.site.login()
			try:
			old = self.page.get(get_redirect=self.options.edit_redirect)
			except pywikibot.NoPage:
			old = ""
			textEditor = TextEditor()
			new = textEditor.edit(old)
			if new and old != new:
			pywikibot.showDiff(old, new)
			changes = pywikibot.input(u"What did you change?")
			comment = i18n.twtranslate(pywikibot.Site(), 'editarticle-edit',
			{'description': changes})
			try:
			self.page.put(new, comment=comment, minorEdit=False,
			watchArticle=self.options.watch)
			except pywikibot.EditConflict:
			self.handle_edit_conflict(new)
			else:
			pywikibot.output(u"Nothing changed")


			def main(*args):
			"""
			Process command line arguments and invoke bot.

			If args is an empty list, sys.argv is used.

			@param args: command line arguments
			@type args: list of unicode
			"""
			app = ArticleEditor(*args)
			app.run()


			if __name__ == "__main__":
			main()

rPWBC/scripts/editarticle.py29125188e56dmaster

editarticle.pyView Options

rPWBC/scripts/editarticle.py
29125188e56dmaster

editarticle.py
View Options