Added support for merchant user IDs to manual ident URLs

Author: MarkusH

CHANGELOG.md

@@ -12,6 +12,8 @@
   installation requirements as it's required for the added manual ident URLs
   feature.
 
+* Added support for `muid` when creating manual ident URLs.
+
 * Added support to check for access based on `muid` instead of `lptoken`.
 
 ## 5.2.0

laterpay/__init__.py

@@ -407,19 +407,21 @@ def get_access_data(self, article_ids, lptoken=None, muid=None):
 
         return response.json()
 
-    def get_manual_ident_url(self, article_url, article_ids):
+    def get_manual_ident_url(self, article_url, article_ids, muid=None):
         """
         Return a URL to allow users to claim previous purchase content.
         """
-        token = self._get_manual_ident_token(article_url, article_ids)
+        token = self._get_manual_ident_token(article_url, article_ids, muid=muid)
         return '%s/ident/%s/%s/' % (self.web_root, self.cp_key, token)
 
-    def _get_manual_ident_token(self, article_url, article_ids):
+    def _get_manual_ident_token(self, article_url, article_ids, muid=None):
         """
         Return the token data for ``get_manual_ident_url()``.
         """
         data = {
             'back': compat.stringify(article_url),
             'ids': [compat.stringify(article_id) for article_id in article_ids],
         }
+        if muid:
+            data['muid'] = compat.stringify(muid)
         return jwt.encode(data, self.shared_secret).decode()

tests/test_client.py

@@ -526,7 +526,7 @@ def test_get_manual_ident_url(self):
         article_url = u'http://example.com/news?id=10&emoji=😄'
         article_ids = ['aid≠1', b'aid\xe2\x89\xa02']
 
-        url = self.lp.get_manual_ident_url(article_url, article_ids)
+        url = self.lp.get_manual_ident_url(article_url, article_ids, muid='blụb')
 
         url_info = urlparse(url)
 
@@ -555,6 +555,7 @@ def test_get_manual_ident_url(self):
         self.assertEqual(data, {
             'back': u'http://example.com/news?id=10&emoji=\U0001f604',
             'ids': [u'aid\u22601', u'aid\u22602'],
+            'muid': u'bl\u1ee5b'
         })
 
     def test_get_manual_ident_token(self):
@@ -569,6 +570,20 @@ def test_get_manual_ident_token(self):
             'ids': [u'aid\u22601', u'aid\u22602'],
         })
 
+    def test_get_manual_ident_token_muid(self):
+        article_url = 'http://example.com/news'
+        article_ids = ['aid=1']
+        muid = u'😄'
+
+        token = self.lp._get_manual_ident_token(article_url, article_ids, muid=muid)
+        data = jwt.decode(token, self.lp.shared_secret)
+
+        self.assertEqual(data, {
+            'back': 'http://example.com/news',
+            'ids': ['aid=1'],
+            'muid': u'\U0001f604',
+        })
+
 
 if __name__ == '__main__':
     unittest.main()