Revision: 6575 Author: nicdumz Date: 2009-04-04 17:39:06 +0000 (Sat, 04 Apr 2009)

Log Message: ----------- Adding -create option, to create pages when adding a category to a non-existing page, instead of skipping the said page. (Think Image descriptions pages!)

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

Modified: trunk/pywikipedia/category.py =================================================================== --- trunk/pywikipedia/category.py 2009-04-04 16:37:34 UTC (rev 6574) +++ trunk/pywikipedia/category.py 2009-04-04 17:39:06 UTC (rev 6575) @@ -45,6 +45,8 @@ * -recurse - Recurse through all subcategories of categories. * -match - Only work on pages whose titles match the given regex (for move and remove actions). + * -create - An option for add: if a page doesn't exist, do not skip it, + create it instead

If action is "add", the following options are supported:

@@ -318,7 +320,7 @@ else: return wikipedia.Page(site, catlink.title())

-def add_category(sort_by_last_name = False): +def add_category(sort_by_last_name = False, create_pages = False): '''A robot to mass-add a category to a list of pages.''' site = wikipedia.getSite() if gen: @@ -352,42 +354,46 @@ try: text = page.get() except wikipedia.NoPage: - wikipedia.output(u"%s doesn't exist yet. Ignoring." - % (page.title())) - pass + if create_pages: + wikipedia.output(u"%s doesn't exist yet. Creating." + % (page.title())) + text = '' + else: + wikipedia.output(u"%s doesn't exist yet. Ignoring." + % (page.title())) + continue except wikipedia.IsRedirectPage, arg: redirTarget = wikipedia.Page(site, arg.args[0]) wikipedia.output( u"WARNING: %s is redirect to %s. Ignoring." % (page.title(), redirTarget.title())) + continue + cats = page.categories() + # Show the title of the page we're working on. + # Highlight the title in purple. + wikipedia.output( + u"\n\n>>> \03{lightpurple}%s\03{default} <<<" + % page.title()) + wikipedia.output(u"Current categories:") + for cat in cats: + wikipedia.output(u"* %s" % cat.title()) + catpl = wikipedia.Page(site, + cat_namespace + ':' + newcatTitle) + if sort_by_last_name: + catpl = sorted_by_last_name(catpl, page) + if catpl in cats: + wikipedia.output(u"%s is already in %s." + % (page.title(), catpl.title())) else: - cats = page.categories() - # Show the title of the page we're working on. - # Highlight the title in purple. - wikipedia.output( - u"\n\n>>> \03{lightpurple}%s\03{default} <<<" - % page.title()) - wikipedia.output(u"Current categories:") - for cat in cats: - wikipedia.output(u"* %s" % cat.title()) - catpl = wikipedia.Page(site, - cat_namespace + ':' + newcatTitle) - if sort_by_last_name: - catpl = sorted_by_last_name(catpl, page) - if catpl in cats: - wikipedia.output(u"%s is already in %s." - % (page.title(), catpl.title())) - else: - wikipedia.output(u'Adding %s' % catpl.aslink()) - cats.append(catpl) - text = page.get() - text = wikipedia.replaceCategoryLinks(text, cats) - try: - page.put(text, comment = editSummary) - except wikipedia.EditConflict: - wikipedia.output( - u'Skipping %s because of edit conflict' - % (page.title())) + wikipedia.output(u'Adding %s' % catpl.aslink()) + cats.append(catpl) + text = wikipedia.replaceCategoryLinks(text, cats) + try: + page.put(text, comment = editSummary) + except wikipedia.EditConflict: + wikipedia.output( + u'Skipping %s because of edit conflict' + % (page.title()))

class CategoryMoveRobot: """Robot to move pages from one category to another.""" @@ -899,6 +905,7 @@ action = None sort_by_last_name = False restore = False + create_pages = False for arg in wikipedia.handleArgs(): if arg == 'add': action = 'add' @@ -946,6 +953,8 @@ talkPages = True elif arg == '-recurse': recurse = True + elif arg == '-create': + create_pages = True else: genFactory.handleArg(arg)

@@ -960,7 +969,7 @@ # The preloading generator is responsible for downloading multiple # pages from the wiki simultaneously. gen = pagegenerators.PreloadingGenerator(genFactory.getCombinedGenerator()) - add_category(sort_by_last_name) + add_category(sort_by_last_name, create_pages) elif action == 'remove': if (fromGiven == False): oldCatTitle = wikipedia.input(u'Please enter the name of the category that should be removed:')