User:P-bot/blp.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
import wikipedia, pagegenerators
import re
summary = u'机器人：添加{{Blp}}模板'
genFactory = pagegenerators.GeneratorFactory()
for arg in wikipedia.handleArgs():
    generator = genFactory.handleArg(arg)
    if generator:
        gen = generator
preloadingGen = pagegenerators.PreloadingGenerator(gen)
for page in preloadingGen:
    if page.namespace() == 0:
        talktitle = 'talk:' + page.title()
        talkpage = wikipedia.Page(wikipedia.getSite(), talktitle)
        try:
            text = talkpage.get()
            if re.search('\{\{\s*([Bb]lp|BLP|生者[传傳][记記])\s*\}\}', text) == None:
                text = '{{Blp}}\n' + text
                talkpage.put(text, summary)
            else:
                wikipedia.output(u'条目[[%s]]的{{Blp}}模板已存在。' % page.title())
                continue
        except wikipedia.NoPage:
            text = '{{Blp}}'
            talkpage.put(text, summary)
        except wikipedia.IsRedirectPage:
            text = '{{Blp}}'
            talkpage.put(text, summary)
    else:
        wikipedia.output(u'[[%s]]不是一个条目。' % page.title())
        continue