Merge pull request GetStream#61 from GetStream/change-to-targets

tbarbugli · web-flow · commit fe4e51b41ae2 · 2017-08-31T12:45:44.000+02:00
Change to targets
diff --git a/CHANGELOG b/CHANGELOG
@@ -2,6 +2,13 @@
  Change history
 ================
 
+2.4.0
+======
+:release-date: 2017-08-31
+:by: Tommaso Barbugli
+
+* Added support for To target update endpoint
+
 2.3.11
 ======
 :release-date: 2017-05-22
diff --git a/stream/exceptions.py b/stream/exceptions.py
@@ -70,6 +70,77 @@ class SiteSuspendedException(StreamApiException):
     status_code = 401
     code = 7
 
+class InvalidPaginationException(StreamApiException):
+
+    '''
+    Raised when there is an issue with your Access Key
+    '''
+    status_code = 401
+    code = 8
+
+
+class MissingRankingException(FeedConfigException):
+    '''
+    Raised when you didn't configure the ranking for the given feed
+    '''
+    status_code = 400
+    code = 12
+
+
+class MissingUserException(MissingRankingException):
+    status_code = 400
+    code = 10
+
+
+class RankingException(FeedConfigException):
+    '''
+    Raised when there is a runtime issue with ranking the feed
+    '''
+    status_code = 400
+    code = 11
+
+
+class RateLimitReached(StreamApiException):
+
+    '''
+    Raised when too many requests are performed
+    '''
+    status_code = 429
+    code = 9
+
+
+class OldStorageBackend(StreamApiException):
+    '''
+    Raised if you try to perform an action which only works with the new storage
+    '''
+    status_code = 400
+    code = 13
+
+
+class BestPracticeException(StreamApiException):
+    '''
+    Raised if best practices are enforced and you do something that
+    would break a high volume integration
+    '''
+    status_code = 400
+    code = 15
+
+
+class DoesNotExistException(StreamApiException):
+    '''
+    Raised when the requested resource could not be found.
+    '''
+    status_code = 404
+    code = 16
+
+
+class NotAllowedException(StreamApiException):
+    '''
+    Raised when the requested action is not allowed for some reason.
+    '''
+    status_code = 403
+    code = 17
+
 
 def get_exceptions():
     from stream import exceptions
diff --git a/stream/feed.py b/stream/feed.py
@@ -19,6 +19,7 @@ def __init__(self, client, feed_slug, user_id, token):
         self.token = token
 
         self.feed_url = 'feed/%s/' % self.id.replace(':', '/')
+        self.feed_targets_url = 'feed_targets/%s/' % self.id.replace(':', '/')
         self.feed_together = self.id.replace(':', '')
         self.signature = self.feed_together + ' ' + self.token
 
@@ -211,3 +212,23 @@ def add_to_signature(self, recipients):
             feed = self.client.feed(feed_slug, user_id)
             data.append("%s %s" % (recipient, feed.token))
         return data
+
+    def update_activity_to_targets(self, foreign_id, time,
+                                   new_targets=None, added_targets=None,
+                                   removed_targets=None):
+        data = {
+            'foreign_id': foreign_id,
+            'time': time,
+        }
+
+        if new_targets is not None:
+            data['new_targets'] = new_targets
+        if added_targets is not None:
+            data['added_targets'] = added_targets
+        if removed_targets is not None:
+            data['removed_targets'] = removed_targets
+
+        url = self.feed_targets_url + 'activity_to_targets/'
+
+        token = self.create_scope_token('feed_targets', 'write')
+        return self.client.post(url, data=data, signature=token)
diff --git a/stream/tests.py b/stream/tests.py
@@ -565,6 +565,31 @@ def test_do_i_follow(self):
         self.assertEqual(followings['results'][0]['feed_id'], social.id)
         self.assertEqual(followings['results'][0]['target_id'], 'user:apy')
 
+    def test_update_activity_to_targets(self):
+        time = datetime.datetime.utcnow().isoformat()
+        foreign_id = 'user:1'
+        activity_data = {
+            'actor': 1,
+            'verb': 'tweet',
+            'object': 1,
+            'foreign_id': foreign_id,
+            'time': time,
+        }
+        activity_data['to'] = ['user:1', 'user:2']
+        self.user1.add_activity(activity_data)
+
+        ret = self.user1.update_activity_to_targets(foreign_id, time, new_targets=['user:3', 'user:2'])
+        self.assertEqual(len(ret['activity']['to']), 2)
+        self.assertTrue('user:2' in ret['activity']['to'])
+        self.assertTrue('user:3' in ret['activity']['to'])
+
+        ret = self.user1.update_activity_to_targets(foreign_id, time, added_targets=['user:4', 'user:5'], removed_targets=['user:3'])
+        self.assertEqual(len(ret['activity']['to']), 3)
+        self.assertTrue('user:2' in ret['activity']['to'])
+        self.assertTrue('user:4' in ret['activity']['to'])
+        self.assertTrue('user:5' in ret['activity']['to'])
+
+
     def test_get(self):
         activity_data = {'actor': 1, 'verb': 'tweet', 'object': 1}
         activity_id = self.user1.add_activity(activity_data)['id']
