jenkins-bot has submitted this change and it was merged.
Change subject: (bug 64174) Bot to illustrate Wikidata. Happy pyflakes and happy pep8 ......................................................................
(bug 64174) Bot to illustrate Wikidata. Happy pyflakes and happy pep8
Change-Id: I8db93b3e9af90c5057b2d9b35c0fd85da41d3343 --- A scripts/illustrate_wikidata.py 1 file changed, 109 insertions(+), 0 deletions(-)
Approvals: Ladsgroup: Looks good to me, approved jenkins-bot: Verified
diff --git a/scripts/illustrate_wikidata.py b/scripts/illustrate_wikidata.py new file mode 100644 index 0000000..14e0898 --- /dev/null +++ b/scripts/illustrate_wikidata.py @@ -0,0 +1,109 @@ +#!/usr/bin/python +# -*- coding: utf-8 -*- +""" +Bot to add images to Wikidata items. The image is extracted from the page_props. +For this to be available the PageImages extension (https://www.mediawiki.org/wiki/Extension:PageImages) needs to be installed + +Usage: + +python illustrate_wikidata.py <some generator> + +python harvest_template.py -lang:en -catr:Category:Railway_stations_in_New_York + +¶ms; +""" +# +# (C) Multichill 2014 +# (C) Pywikibot team, 2013-2014 +# +# Distributed under the terms of MIT License. +# +__version__ = '$Id$' +# + +import pywikibot +from pywikibot import pagegenerators as pg, WikidataBot + +docuReplacements = {'¶ms;': pywikibot.pagegenerators.parameterHelp} + + +class IllustrateRobot(WikidataBot): + """ + A bot to add Wikidata image claims + """ + def __init__(self, generator, wdproperty=u'P18'): + """ + Arguments: + * generator - A generator that yields Page objects. + * wdproperty - The property to add. Should be of type commonsMedia + + """ + self.generator = pg.PreloadingGenerator(generator) + self.wdproperty = wdproperty + self.repo = pywikibot.Site().data_repository() + self.cacheSources() + + claim = pywikibot.Claim(self.repo, self.wdproperty) + if not claim.getType() == 'commonsMedia': + raise ValueError(u'%s is of type %s, should be commonsMedia' % (self.wdproperty, claim.getType())) + + def run(self): + """ + Starts the robot. + """ + for page in self.generator: + pywikibot.output(u'Working on %s' % page.title()) + item = pywikibot.ItemPage.fromPage(page) + + if item.exists(): + pywikibot.output(u'Found %s' % item.title()) + imagename = page.properties().get('page_image') + + if imagename: + claims = item.get().get('claims') + if self.wdproperty in claims: + pywikibot.output(u'Item %s already contains image (%s)' % (item.title(), self.wdproperty)) + else: + newclaim = pywikibot.Claim(self.repo, self.wdproperty) + commonssite = pywikibot.Site("commons", "commons") + imagelink = pywikibot.Link(imagename, source=commonssite, defaultNamespace=6) + image = pywikibot.ImagePage(imagelink) + if image.isRedirectPage(): + image = pywikibot.ImagePage(image.getRedirectTarget()) + if not image.exists(): + pywikibot.output('[[%s]] doesn't exist so I can't link to it' % (image.title(),)) + continue + newclaim.setTarget(image) + pywikibot.output('Adding %s --> %s' % (newclaim.getID(), newclaim.getTarget())) + item.addClaim(newclaim) + + # A generator might yield pages from multiple sites + source = self.getSource(page.site) + if source: + newclaim.addSource(source, bot=True) + + +def main(): + gen = pg.GeneratorFactory() + wdproperty = u'P18' + + for arg in pywikibot.handleArgs(): + if arg.startswith('-property'): + if len(arg) == 9: + wdproperty = pywikibot.input( + u'Please enter the property you want to add:') + else: + wdproperty = arg[10:] + elif gen.handleArg(arg): + continue + + generator = gen.getCombinedGenerator() + if not generator: + pywikibot.output('I need a generator with pages to work on') + return + + bot = IllustrateRobot(generator, wdproperty) + bot.run() + +if __name__ == "__main__": + main()