Revision: 8698
Author: xqt
Date: 2010-11-05 15:30:44 +0000 (Fri, 05 Nov 2010)
Log Message:
-----------
UserEditFilterGenerator() and options -user -skipuser
yield a page if a given user is or is not author of that page. Yet it looks at the last
100 edits only.
Modified Paths:
--------------
trunk/pywikipedia/template.py
Modified: trunk/pywikipedia/template.py
===================================================================
--- trunk/pywikipedia/template.py 2010-11-05 06:52:20 UTC (rev 8697)
+++ trunk/pywikipedia/template.py 2010-11-05 15:30:44 UTC (rev 8698)
@@ -24,6 +24,16 @@
-namespace: Only process templates in the given namespace number (may be used
multiple times).
+-user: Only process pages edited by a given user
+
+-skipuser: Only process pages not edited by a given user
+
+-timestamp: (With -user or -skipuser). Only check for a user where his edit is
+ not older than the given timestamp. Timestamp must be writen in
+ MediaWiki timestamp format which is "%Y%m%d%H%M%S"
+ If this parameter is missed, all edits are checked but this is
+ restricted to the last 100 edits.
+
-summary: Lets you pick a custom edit summary. Use quotes if edit summary contains
spaces.
@@ -94,6 +104,35 @@
import replace, pagegenerators
import re, sys, string, catlib
+def UserEditFilterGenerator(generator, username, timestamp=None, skip=False):
+ """
+ Generator which will yield Pages depending of user:username is an Author of
+ that page (only looks at the last 100 editors).
+ If timestamp is set in MediaWiki format JJJJMMDDhhmmss, older edits are
+ ignored
+ If skip is set, pages edited by the given user are ignored otherwise only
+ pages edited by this user are given back
+
+ """
+ if timestamp:
+ ts = pywikibot.Timestamp.fromtimestampformat(timestamp)
+ for page in generator:
+ editors = page.getLatestEditors(limit=100)
+ found = False
+ for ed in editors:
+ uts = pywikibot.Timestamp.fromISOformat(ed['timestamp'])
+ if not timestamp or uts>=ts:
+ if username == ed['user']:
+ found = True
+ break
+ else:
+ break
+ if found and not skip or not found and skip:
+ yield page
+ else:
+ pywikibot.output(u'Skipping %s' % page.title(asLink=True))
+
+
class XmlDumpTemplatePageGenerator:
"""
Generator which will yield Pages to pages that might contain the chosen
@@ -381,12 +420,17 @@
genFactory = pagegenerators.GeneratorFactory()
# If xmlfilename is None, references will be loaded from the live wiki.
xmlfilename = None
+ user = None
+ skip = False
+ timestamp = None
# read command line parameters
for arg in pywikibot.handleArgs():
if arg == '-remove':
remove = True
elif arg == '-subst':
subst = True
+ elif arg == ('-always'):
+ acceptAll = True
elif arg.startswith('-xml'):
if len(arg) == 4:
xmlfilename = pywikibot.input(u'Please enter the XML dump\'s
filename: ')
@@ -401,8 +445,13 @@
addedCat = arg[len('-category:'):]
elif arg.startswith('-summary:'):
editSummary = arg[len('-summary:'):]
- elif arg.startswith('-always'):
- acceptAll = True
+ elif arg.startswith('-user:'):
+ user = arg[len('-user:'):]
+ elif arg.startswith('-skipuser:'):
+ user = arg[len('-skipuser:'):]
+ skip = True
+ elif arg.startswith('-timestamp:'):
+ timestamp = arg[len('-timestamp:'):]
else:
if not genFactory.handleArg(arg):
templateNames.append(pywikibot.Page(pywikibot.getSite(), arg,
defaultNamespace=10).titleWithoutNamespace())
@@ -436,7 +485,8 @@
if namespaces:
gen = pagegenerators.NamespaceFilterPageGenerator(gen, namespaces)
-
+ if user:
+ gen = UserEditFilterGenerator(gen, user, timestamp, skip)
preloadingGen = pagegenerators.PreloadingGenerator(gen)
bot = TemplateRobot(preloadingGen, templates, subst, remove, editSummary, acceptAll,
addedCat)