Ladsgroup has uploaded a new change for review.
https://gerrit.wikimedia.org/r/239512
Change subject: Add basic support for badges
......................................................................
Add basic support for badges
Change-Id: I1daa7cb85cea390c89819c749696aadb39690bd2
---
M pywikibase/itempage.py
M tests/test_itempage.py
2 files changed, 11 insertions(+), 1 deletion(-)
git pull ssh://gerrit.wikimedia.org:29418/pywikibot/wikibase
refs/changes/12/239512/1
diff --git a/pywikibase/itempage.py b/pywikibase/itempage.py
index dd7b516..df7ee42 100644
--- a/pywikibase/itempage.py
+++ b/pywikibase/itempage.py
@@ -49,12 +49,16 @@
"""
data = super(ItemPage, self).get(*args, **kwargs)
- # sitelinks
+ # sitelinks and badges
self.sitelinks = {}
+ self.badges = {}
if 'sitelinks' in self._content:
for dbname in self._content['sitelinks']:
self.sitelinks[dbname] = self._content[
'sitelinks'][dbname]['title']
+ if self._content['sitelinks'][dbname]['badges']:
+ self.badges[dbname] = \
+ self._content['sitelinks'][dbname]['badges']
data['claims'] = self.claims
data['sitelinks'] = self.sitelinks
diff --git a/tests/test_itempage.py b/tests/test_itempage.py
index 24e2357..7f5a261 100644
--- a/tests/test_itempage.py
+++ b/tests/test_itempage.py
@@ -45,5 +45,11 @@
self.assertNotEqual(self.item_page.claims, old_claims)
self.assertNotIn('P31', self.item_page.claims)
+ def test_badges(self):
+ self.assertEqual(len(self.item_page.badges), 4)
+ self.assertEqual(self.item_page.badges['enwiki'], ['Q17437798'])
+ self.assertIn('enwiki', self.item_page.badges)
+ self.assertNotIn('fawiki', self.item_page.badges)
+
if __name__ == '__main__':
unittest.main()
--
To view, visit https://gerrit.wikimedia.org/r/239512
To unsubscribe, visit https://gerrit.wikimedia.org/r/settings
Gerrit-MessageType: newchange
Gerrit-Change-Id: I1daa7cb85cea390c89819c749696aadb39690bd2
Gerrit-PatchSet: 1
Gerrit-Project: pywikibot/wikibase
Gerrit-Branch: master
Gerrit-Owner: Ladsgroup <[email protected]>
_______________________________________________
MediaWiki-commits mailing list
[email protected]
https://lists.wikimedia.org/mailman/listinfo/mediawiki-commits
