M4tx has uploaded a new change for review.
https://gerrit.wikimedia.org/r/101642
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(-)
git pull ssh://gerrit.wikimedia.org:29418/pywikibot/core
refs/changes/42/101642/1
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()
--
To view, visit https://gerrit.wikimedia.org/r/101642
To unsubscribe, visit https://gerrit.wikimedia.org/r/settings
Gerrit-MessageType: newchange
Gerrit-Change-Id: I92e6a4adf5276c55624fa14239c79ac4d5a282e8
Gerrit-PatchSet: 1
Gerrit-Project: pywikibot/core
Gerrit-Branch: master
Gerrit-Owner: M4tx <[email protected]>
_______________________________________________
MediaWiki-commits mailing list
[email protected]
https://lists.wikimedia.org/mailman/listinfo/mediawiki-commits
