http://www.mediawiki.org/wiki/Special:Code/pywikipedia/9544

Revision: 9544 Author: xqt Date: 2011-09-25 10:38:14 +0000 (Sun, 25 Sep 2011) Log Message: ----------- limit to 80 chars (PEP 8 styling guide); use default sites namespace alias.

Modified Paths: -------------- trunk/pywikipedia/template.py

Modified: trunk/pywikipedia/template.py =================================================================== --- trunk/pywikipedia/template.py 2011-09-25 09:08:55 UTC (rev 9543) +++ trunk/pywikipedia/template.py 2011-09-25 10:38:14 UTC (rev 9544) @@ -98,6 +98,8 @@ # # (C) Daniel Herding, 2004 # (C) Rob W.W. Hooft, 2003 +# (C) xqt, 2009-2011 +# (C) Pywikipedia team, 2004-2010 # # Distributed under the terms of the MIT license. # @@ -206,23 +208,24 @@ self.editSummary = editSummary self.acceptAll = acceptAll self.addedCat = addedCat + site = pywikibot.getSite() if self.addedCat: - self.addedCat = catlib.Category(pywikibot.getSite(), 'Category:' + self.addedCat) + self.addedCat = catlib.Category( + site, u'%s:%s' % (site.namespace(14), self.addedCat))

# get edit summary message if it's empty if (self.editSummary==''): Param = {'list': (', ').join(self.templates.keys()), 'num' : len(self.templates)} - mysite = pywikibot.getSite() if self.remove: self.editSummary = i18n.twntranslate( - mysite, 'template-removing', Param) + site, 'template-removing', Param) elif self.subst: self.editSummary = i18n.twntranslate( - mysite, 'template-substituting', Param) + site, 'template-substituting', Param) else: self.editSummary = i18n.twntranslate( - mysite, 'template-changing', Param) + site, 'template-changing', Param)

def run(self): """ @@ -240,7 +243,10 @@ for old, new in self.templates.iteritems(): namespaces = list(site.namespace(10, all=True)) if not site.nocapitalize: - pattern = '[' + re.escape(old[0].upper()) + re.escape(old[0].lower()) + ']' + re.escape(old[1:]) + pattern = '[' + \ + re.escape(old[0].upper()) + \ + re.escape(old[0].lower()) + \ + ']' + re.escape(old[1:]) else: pattern = re.escape(old) pattern = re.sub(r'_|\ ', r'[_ ]', pattern) @@ -252,12 +258,17 @@ if self.remove: replacements.append((templateRegex, '')) elif self.subst: - replacements.append((templateRegex, '{{subst:' + old + '\g<parameters>}}')) + replacements.append((templateRegex, + '{{subst:%s\g<parameters>}}' % old)) exceptions['inside-tags']=['ref', 'gallery'] else: - replacements.append((templateRegex, '{{' + new + '\g<parameters>}}')) + replacements.append((templateRegex, + '{{%s\g<parameters>}}' % new))

- replaceBot = replace.ReplaceRobot(self.generator, replacements, exceptions, acceptall = self.acceptAll, addedCat=self.addedCat, editSummary=self.editSummary) + replaceBot = replace.ReplaceRobot(self.generator, replacements, + exceptions, acceptall=self.acceptAll, + addedCat=self.addedCat, + editSummary=self.editSummary) replaceBot.run()

def main(*args): @@ -276,7 +287,7 @@ skip = False timestamp = None # read command line parameters - for arg in pywikibot.handleArgs(): + for arg in pywikibot.handleArgs(*args): if arg == '-remove': remove = True elif arg == '-subst': @@ -285,7 +296,8 @@ acceptAll = True elif arg.startswith('-xml'): if len(arg) == 4: - xmlfilename = pywikibot.input(u'Please enter the XML dump's filename: ') + xmlfilename = pywikibot.input( + u'Please enter the XML dump's filename: ') else: xmlfilename = arg[5:] elif arg.startswith('-namespace:'): @@ -306,7 +318,9 @@ timestamp = arg[len('-timestamp:'):] else: if not genFactory.handleArg(arg): - templateNames.append(pywikibot.Page(pywikibot.getSite(), arg, defaultNamespace=10).titleWithoutNamespace()) + templateNames.append( + pywikibot.Page(pywikibot.getSite(), arg, + defaultNamespace=10).titleWithoutNamespace())

if subst or remove: for templateName in templateNames: @@ -316,7 +330,8 @@ for i in range(0, len(templateNames), 2): templates[templateNames[i]] = templateNames[i + 1] except IndexError: - pywikibot.output(u'Unless using -subst or -remove, you must give an even number of template names.') + pywikibot.output( +u'Unless using -subst or -remove, you must give an even number of template names.') return

oldTemplates = [] @@ -331,7 +346,8 @@ gen = genFactory.getCombinedGenerator() if not gen: gens = [] - gens = [pagegenerators.ReferringPageGenerator(t, onlyTemplateInclusion = True) for t in oldTemplates] + gens = [pagegenerators.ReferringPageGenerator( + t, onlyTemplateInclusion=True) for t in oldTemplates] gen = pagegenerators.CombinedPageGenerator(gens) gen = pagegenerators.DuplicateFilterPageGenerator(gen)

@@ -341,7 +357,8 @@ gen = UserEditFilterGenerator(gen, user, timestamp, skip) preloadingGen = pagegenerators.PreloadingGenerator(gen)

- bot = TemplateRobot(preloadingGen, templates, subst, remove, editSummary, acceptAll, addedCat) + bot = TemplateRobot(preloadingGen, templates, subst, remove, editSummary, + acceptAll, addedCat) bot.run()

if __name__ == "__main__":