jenkins-bot has submitted this change and it was merged.
Change subject: Port catall.py to core ......................................................................
Port catall.py to core
Change-Id: I92e6a4adf5276c55624fa14239c79ac4d5a282e8 --- A scripts/catall.py 1 file changed, 116 insertions(+), 0 deletions(-)
Approvals: Merlijn van Deen: Looks good to me, approved jenkins-bot: Verified
diff --git a/scripts/catall.py b/scripts/catall.py new file mode 100755 index 0000000..c99ac4d --- /dev/null +++ b/scripts/catall.py @@ -0,0 +1,116 @@ +# -*- coding: utf-8 -*- +""" +Add or change categories on a number of pages. Usage: catall.py name - goes +through pages, starting at 'name'. Provides the categories on the page and asks +whether to change them. If no starting name is provided, the bot starts at 'A'. + +Options: +-onlynew : Only run on pages that do not yet have a category. +""" +# +# (C) Rob W.W. Hooft, Andre Engels, 2004 +# (C) Pywikibot team, 2004-2013 +# +# Distributed under the terms of the MIT license. +# +__version__ = '$Id$' +# + +import sys +import pywikibot +from pywikibot import i18n + + +def choosecats(pagetext): + chosen = [] + done = False + length = 1000 + print("""Give the new categories, one per line. +Empty line: if the first, don't change. Otherwise: Ready. +-: I made a mistake, let me start over. +?: Give the text of the page with GUI. +??: Give the text of the page in console. +xx: if the first, remove all categories and add no new. +q: quit.""") + while not done: + choice = pywikibot.input(u"?") + if choice == "": + done = True + elif choice == "-": + chosen = choosecats(pagetext) + done = True + elif choice == "?": + from pywikibot import editor as editarticle + editor = editarticle.TextEditor() + newtext = editor.edit(pagetext) + elif choice == "??": + pywikibot.output(pagetext[0:length]) + length = length + 500 + elif choice == "xx" and chosen == []: + chosen = None + done = True + elif choice == "q": + print "quit..." + sys.exit() + else: + chosen.append(choice) + return chosen + + +def make_categories(page, list, site=None): + if site is None: + site = pywikibot.getSite() + pllist = [] + for p in list: + cattitle = "%s:%s" % (site.category_namespace(), p) + pllist.append(pywikibot.Page(site, cattitle)) + page.put_async(pywikibot.replaceCategoryLinks(page.get(), pllist), + comment=i18n.twtranslate(site.lang, 'catall-changing')) + + +def main(): + docorrections = True + start = [] + + for arg in pywikibot.handleArgs(): + if arg == '-onlynew': + docorrections = False + else: + start.append(arg) + + if not start: + start = 'A' + else: + start = ' '.join(start) + + mysite = pywikibot.getSite() + + for p in mysite.allpages(start=start): + try: + text = p.get() + cats = p.categories() + if not cats: + pywikibot.output(u"========== %s ==========" % p.title()) + print "No categories" + print "-" * 40 + newcats = choosecats(text) + if newcats != [] and newcats is not None: + make_categories(p, newcats, mysite) + elif docorrections: + pywikibot.output(u"========== %s ==========" % p.title()) + for c in cats: + pywikibot.output(c.title()) + print "-" * 40 + newcats = choosecats(text) + if newcats is None: + make_categories(p, [], mysite) + elif newcats != []: + make_categories(p, newcats, mysite) + except pywikibot.IsRedirectPage: + pywikibot.output(u'%s is a redirect' % p.title()) + +if __name__ == "__main__": + try: + main() + finally: + pywikibot.stopme()