Strainu has submitted this change and it was merged.
Change subject: Porting revertbot.py from compat to core ......................................................................
Porting revertbot.py from compat to core
For Google Code-In Replaced imports and deprecated query.GetData calls
Change-Id: If77ba53e6905b3f1cc2b4dba75af492fd2bb5d18 --- A scripts/revertbot.py 1 file changed, 118 insertions(+), 0 deletions(-)
Approvals: Strainu: Verified; Looks good to me, approved
diff --git a/scripts/revertbot.py b/scripts/revertbot.py new file mode 100644 index 0000000..21d3de5 --- /dev/null +++ b/scripts/revertbot.py @@ -0,0 +1,118 @@ +#!/usr/bin/python +# -*- coding: utf-8 -*- +""" +""" +# +# (C) Bryan Tong Minh, 2008 +# (C) Pywikipedia bot team, 2008-2013 +# +# Ported by Geoffrey "GEOFBOT" Mon - User:Sn1per +# for Google Code-In 2013 +# +# Distributed under the terms of the MIT license. +# +__version__ = '$Id$' +# + +import re +import pywikibot +from pywikibot import i18n +import query + + +class BaseRevertBot(object): + """ Base revert bot + + Subclass this bot and override callback to get it to do something useful. + + """ + def __init__(self, site, comment=None): + self.site = site + self.comment = comment + + def get_contributions(self, max=500, ns=None): + count = 0 + iterator = iter(xrange(0)) + never_continue = False + while count != max or never_continue: + try: + item = iterator.next() + except StopIteration: + self.log(u'Fetching new batch of contributions') + data = list(pywikibot.getSite().usercontribs(user=self.site.username(), namespaces=ns, total=max)) + never_continue = True + iterator = iter(data) + else: + count += 1 + yield item + + def revert_contribs(self, callback=None): + self.site.forceLogin() + + if callback is None: + callback = self.callback + + contribs = self.get_contributions() + for item in contribs: + try: + if callback(item): + result = self.revert(item) + if result: + self.log(u'%s: %s' % (item['title'], result)) + else: + self.log(u'Skipped %s' % item['title']) + else: + self.log(u'Skipped %s by callback' % item['title']) + except StopIteration: + return + + def callback(self, item): + return 'top' in item + + def revert(self, item): + if len(pywikibot.Page(pywikibot.getSite(), item['title']).fullVersionHistory()) > 1: + rev = pywikibot.Page(pywikibot.getSite(), item['title']).fullVersionHistory()[1] + else: + return False + + comment = i18n.twtranslate(pywikibot.getSite(), 'revertbot-revert', {'revid': rev[0], 'author': rev[2], 'timestamp': rev[1]}) + + if self.comment: + comment += ': ' + self.comment + + page = pywikibot.Page(self.site, item['title']) + pywikibot.output(u"\n\n>>> \03{lightpurple}%s\03{default} <<<" + % page.title(asLink=True, forceInterwiki=True, + textlink=True)) + old = page.get() + new = rev[3] + pywikibot.showDiff(old, new) + page.put(new, comment) + return comment + + def log(self, msg): + pywikibot.output(msg) + + +class myRevertBot(BaseRevertBot): + + def callback(self, item): + if 'top' in item: + page = pywikibot.Page(self.site, item['title']) + text = page.get() + pattern = re.compile(u'[[.+?:.+?..+?]]', re.UNICODE) + return pattern.search(text) >= 0 + return False + + +def main(): + for arg in pywikibot.handleArgs(): + continue + bot = myRevertBot(site=pywikibot.getSite()) + bot.revert_contribs() + +if __name__ == "__main__": + try: + main() + finally: + pywikibot.stopme()