SVN: [11240] branches/rewrite/pywikibot - Pywikipedia-svn

23 Mar 2013

http://www.mediawiki.org/wiki/Special:Code/pywikipedia/11240
Revision: 11240
Author:   legoktm
Date:     2013-03-23 08:09:19 +0000 (Sat, 23 Mar 2013)
Log Message:
-----------
Add support for adding sources to claims
Modified Paths:
--------------
    branches/rewrite/pywikibot/page.py
    branches/rewrite/pywikibot/site.py
Modified: branches/rewrite/pywikibot/page.py
===================================================================

--- branches/rewrite/pywikibot/page.py	2013-03-23 06:22:29 UTC (rev 11239)
+++ branches/rewrite/pywikibot/page.py	2013-03-23 08:09:19 UTC (rev 11240)
@@ -2613,12 +2613,14 @@
         """
         return self.sources
-    def addSource(self, source):
+    def addSource(self, source, **kwargs):
         """
         source is a Claim.
         adds it as a reference.
         """
-        raise NotImplementedError
+        data = self.repo.editSource(self, source, new=True, **kwargs)
+        source.hash = data['reference']['hash']
+        self.sources.append(source)
class Revision(object):
Modified: branches/rewrite/pywikibot/site.py
===================================================================
--- branches/rewrite/pywikibot/site.py	2013-03-23 06:22:29 UTC (rev 11239)
+++ branches/rewrite/pywikibot/site.py	2013-03-23 08:09:19 UTC (rev 11240)
@@ -3409,9 +3409,55 @@
         data = req.submit()
         return data
+    def editSource(self, claim, source, new=False, **kwargs):
+        """
+        Create/Edit a source.
+        @param claim A Claim object to add the source to
+        @type claim pywikibot.Claim
+        @param source A Claim object to be used as a source
+        @type source pywikibot.Claim
+        @param new Whether to create a new one if the "source" already exists
+        @type new bool
+        """
+        if claim.isReference:
+            raise ValueError("The claim cannot be a reference.")
+        params = dict(action='wbsetreference',
+                      statement=claim.snak,
+                      )
+        params['token'] = self.token(claim, 'edit')
+        if not new and hasattr(source, 'hash'):
+            params['reference'] = source.hash
+        #build up the snak
+        if source.getType() == 'wikibase-item':
+            datavalue = {'type': 'wikibase-entityid',
+                         'value': {'entity-type': 'item',
+                                   'numeric-id': source.getTarget().getID(numeric=True),
+                                   },
+                         }
+        elif source.getType() == 'string':
+            datavalue = {'type': 'string',
+                         'value': source.getTarget(),
+                         }
+        else:
+            raise NotImplementedError('%s datatype is not supported yet.' % claim.getType())
+        snak = {source.getID(): [{'snaktype': 'value',
+                                  'property': source.getID(),
+                                  'datavalue': datavalue,
+                                  },
+                                 ],
+                }
+        params['snak'] = json.dumps(snak)
+        for arg in kwargs:
+            if arg in ['bot', 'lastrevid']:
+                params[arg] = kwargs[arg]
+        req = api.Request(site=self, **params)
+        data = req.submit()
+        return data
+
+
     # deprecated BaseSite methods
     def fam(self):
         raise NotImplementedError

    