Revision: 7729 Author: xqt Date: 2009-12-02 16:50:10 +0000 (Wed, 02 Dec 2009)

Log Message: ----------- -titlecase option added

Modified Paths: -------------- trunk/pywikipedia/capitalize_redirects.py

Modified: trunk/pywikipedia/capitalize_redirects.py =================================================================== --- trunk/pywikipedia/capitalize_redirects.py 2009-12-02 15:39:17 UTC (rev 7728) +++ trunk/pywikipedia/capitalize_redirects.py 2009-12-02 16:50:10 UTC (rev 7729) @@ -1,7 +1,9 @@ +#!/usr/bin/python # -*- coding: utf-8 -*-

''' -Bot to create redirects. +Bot to create capitalized redirects where the first character of the first +word is uppercase and the remainig characters and words are lowercase.

Command-line arguments:

@@ -28,16 +30,24 @@

-always Don't prompt to make changes, just do them.

+ -titlecase creates a titlecased redirect version of a given page + where all words of the title start with an uppercase + character and the remaining characters are lowercase + Example: "python capitalize_redirects.py -start:B -always" ''' # # (C) Yrithinnd +# (C) Pywikipedia bot team, 2007-2009 +# # Class licensed under terms of the MIT license # __version__ = '$Id$' +#

import time, sys -import wikipedia, pagegenerators, catlib +import wikipedia as pywikibot +import pagegenerators, catlib

msg = { 'ar': u'روبوت: إنشاء تحويلة إلى [[%s]]', @@ -55,9 +65,11 @@ }

class CapitalizeBot: - def __init__(self, generator, acceptall): + def __init__(self, generator, acceptall, titlecase): self.generator = generator self.acceptall = acceptall + self.titlecase = titlecase + self.site = pywikibot.getSite()

def run(self): for page in self.generator: @@ -66,31 +78,32 @@ page_t = page.title() # Show the title of the page we're working on. # Highlight the title in purple. - wikipedia.output(u"\n>>> \03{lightpurple}%s\03{default} <<<" + pywikibot.output(u"\n>>> \03{lightpurple}%s\03{default} <<<" % page_t) - page_cap = wikipedia.Page(wikipedia.getSite(), page_t.title().capitalize()) + if self.titlecase: + page_cap = pywikibot.Page(self.site, page_t.title()) + else: + page_cap = pywikibot.Page(self.site, page_t.capitalize()) if not page_cap.exists(): - wikipedia.output(u'%s doesn't exist' % page_cap.title()) + pywikibot.output(u'[[%s]] doesn't exist' % page_cap.title()) if not self.acceptall: - choice = wikipedia.inputChoice( + choice = pywikibot.inputChoice( u'Do you want to create a redirect?', - ['Yes', 'No', 'All'], ['y', 'N', 'a'], 'N') + ['Yes', 'No', 'All', 'Quit'], ['y', 'N', 'a', 'q'], 'N') if choice == 'a': self.acceptall = True + elif choice == 'q': + break if self.acceptall or choice == 'y': + comment = pywikibot.translate(self.site, msg) % page_t try: - wikipedia.setAction( - wikipedia.translate(wikipedia.getSite(), msg) - % page_t) - page_cap.put(u"#REDIRECT [[%s]]" % page_t) + page_cap.put(u"#REDIRECT [[%s]]" % page_t, comment) print except: - wikipedia.output( - u"An error occurred. Retrying in 15 seconds...") - time.sleep(15) + pywikibot.output(u"An error occurred, skipping...") continue else: - wikipedia.output(u'%s already exists, skipping...\n' + pywikibot.output(u'%s already exists, skipping...\n' % page_t.title())

def main(): @@ -103,44 +116,47 @@ acceptall = False namespaces = [] startpage = None + titlecase = False

- for arg in wikipedia.handleArgs(): + for arg in pywikibot.handleArgs(): if arg.startswith('-file'): if len(arg) == 5: - textfilename = wikipedia.input(u'Please enter the filename:') + textfilename = pywikibot.input(u'Please enter the filename:') else: textfilename = arg[6:] source = 'textfile' elif arg.startswith('-cat'): if len(arg) == 4: - categoryname = wikipedia.input( + categoryname = pywikibot.input( u'Please enter the category name:') else: categoryname = arg[5:] source = 'category' elif arg.startswith('-page'): if len(arg) == 5: - pageNames.append(wikipedia.input( + pageNames.append(pywikibot.input( u'Which page do you want to change?')) else: pageNames.append(arg[6:]) source = 'singlepage' elif arg.startswith('-ref'): if len(arg) == 4: - referredPageName = wikipedia.input( + referredPageName = pywikibot.input( u'Links to which page should be processed?') else: referredPageName = arg[5:] source = 'ref' elif arg.startswith('-start'): if len(arg) == 6: - firstPageTitle = wikipedia.input( + firstPageTitle = pywikibot.input( u'Which page do you want to change?') else: firstPageTitle = arg[7:] source = 'allpages' elif arg == '-always': acceptall = True + elif arg == '-titlecase': + titlecase = True elif arg.startswith('-namespace:'): try: namespaces.append(int(arg[11:])) @@ -152,30 +168,30 @@ if source == 'textfile': gen = pagegenerators.TextfilePageGenerator(textfilename) elif source == 'category': - cat = catlib.Category(wikipedia.getSite(), categoryname) + cat = catlib.Category(pywikibot.getSite(), categoryname) gen = pagegenerators.CategorizedPageGenerator(cat) elif source == 'singlepage': - pages = [wikipedia.Page(wikipedia.getSite(), pageName) + pages = [pywikibot.Page(pywikibot.getSite(), pageName) for pageName in pageNames] gen = iter(pages) elif source == 'allpages': - namespace = wikipedia.Page(wikipedia.getSite(), + namespace = pywikibot.Page(pywikibot.getSite(), firstPageTitle).namespace() gen = pagegenerators.AllpagesPageGenerator(firstPageTitle, namespace) elif source == 'ref': - referredPage = wikipedia.Page(wikipedia.getSite(), referredPageName) + referredPage = pywikibot.Page(pywikibot.getSite(), referredPageName) gen = pagegenerators.ReferringPageGenerator(referredPage) elif source == None or len(commandline_replacements) not in [0, 2]: - wikipedia.showHelp(u'capitalize_redirects') + pywikibot.showHelp(u'capitalize_redirects') return if namespaces != []: gen = pagegenerators.NamespaceFilterPageGenerator(gen, namespaces) preloadingGen = pagegenerators.PreloadingGenerator(gen, pageNumber = 20) - bot = CapitalizeBot(preloadingGen, acceptall) + bot = CapitalizeBot(preloadingGen, acceptall, titlecase) bot.run()

if __name__ == "__main__": try: main() finally: - wikipedia.stopme() + pywikibot.stopme()