[Pywikipedia-l] SVN: [6575] trunk/pywikipedia/category.py

[nicdumz at svn.wikimedia.org]

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:')