http://www.mediawiki.org/wiki/Special:Code/pywikipedia/9294

Revision: 9294 Author: multichill Date: 2011-06-12 15:23:51 +0000 (Sun, 12 Jun 2011) Log Message: ----------- Simple tool to create a lot of categories.

Added Paths: ----------- trunk/pywikipedia/create_categories.py

Added: trunk/pywikipedia/create_categories.py =================================================================== --- trunk/pywikipedia/create_categories.py (rev 0) +++ trunk/pywikipedia/create_categories.py 2011-06-12 15:23:51 UTC (rev 9294) @@ -0,0 +1,100 @@ +# -*- coding: utf-8 -*- +""" +Program to batch create categories. + +The program expects a generator containing a list of page titles to be used as base. + +The following command line parameters are supported: + +-alway (not implemented yet) Don't ask, just do the edit. + +-overwrite (not implemented yet). + +-parent The name of the parent category. + +-basename The base to be used for the new category names. + +Example: +create_categories.py + -lang:commons + -family:commons + -links:User:Multichill/Wallonia + -parent:"Cultural heritage monuments in Wallonia" + -basename:"Cultural heritage monuments in" + +""" +__version__ = '$Id: create_categories.py 9042 2011-03-13 10:14:47Z xqt $' +# +# (C) Multichill 2011 +# +# Distributed under the terms of the MIT license. +# +# +import os, sys, re, codecs +import urllib, httplib, urllib2 +import catlib +import time +import socket +import StringIO +import wikipedia as pywikibot +import config +import pagegenerators + + +def createCategory(page, parent, basename): + title = page.titleWithoutNamespace() + + newpage = pywikibot.Page(pywikibot.getSite(u'commons', u'commons'), + u'Category:' + basename + u' ' + title) + newtext = u'' + newtext += u'[[Category:' + parent + u'|' + title + u']]\n' + newtext += u'[[Category:' + title + u']]\n' + + if not newpage.exists(): + #FIXME: Add user prompting and -always option + pywikibot.output(newpage.title()) + pywikibot.showDiff(u'', newtext) + + comment = u'Creating new category' + #FIXME: Add exception handling + newpage.put(newtext, comment) + else: + #FIXME: Add overwrite option + pywikibot.output(u'%s already exists, skipping' % (newpage.title(),)) + + +def main(args): + ''' + Main loop. Get a generator and options. + ''' + generator = None + parent = u'' + basename = u'' + always = False + + genFactory = pagegenerators.GeneratorFactory() + + for arg in pywikibot.handleArgs(): + if arg == '-always': + always = True + elif arg.startswith('-parent:'): + parent = arg [len('-parent:'):].strip() + elif arg.startswith('-basename'): + basename = arg [len('-basename:'):].strip() + else: + genFactory.handleArg(arg) + + generator = genFactory.getCombinedGenerator() + if generator: + for page in generator: + createCategory(page, parent, basename) + else: + pywikibot.output(u'No pages to work on') + + pywikibot.output(u'All done') + +if __name__ == "__main__": + try: + main(sys.argv[1:]) + finally: + pywikibot.stopme()