pythonthings
diff --git a/‎stream/personalization.py‎
Lines changed: 39 additions & 17 deletions b/‎stream/personalization.py‎
Lines changed: 39 additions & 17 deletions
@@ -1,42 +1,56 @@
-
 class Personalization(object):
     def __init__(self, client, token):
+        """
+
+        :param client: the api client
+        :param token: the token
+        """
 
         self.client = client
         self.token = token
 
     def get(self, url, **params):
         """
         Get personalized activities for this feed
+        :param url: personalized url endpoint i.e "follow recommendations"
+        :param params: params to pass to url i.e user_id = "user:123"
+        :return: personalized feed
 
-        :param params:
-        :return:
+        **Example**::
+
+            personalization.get('follow_recommendations', limit=10, offset=10)
         """
 
         response = self.client.get(url, personal=True, params=params,
                                    signature=self.token)
         return response
 
-    def post(self, url, *args, **params):
+    def post(self, url, **params):
         """
         "Generic function to post data to personalization endpoint
-        :param url: personalization endpoint ex: "meta"
-        :param args: If endpoint has required args insert them here.
-        :param kwargs: data is a reserved keyword to post to body
+        :param url: personalized url endpoint i.e "follow recommendations"
+        :param params: params to pass to url (data is a reserved keyword to post to body)
 
         """
 
-        args = args or None
         data = params['data'] or None
-        print(data)
-        if args is not None:
-            url = url + '/' + '/'.join(list(args))
 
         response = self.client.post(url, personal=True, params=params,
                                     signature=self.token, data=data)
         return response
 
-    def upsert_data(self, item_type, ids, data):
+    def upsert_data(self, feed_group, ids, data):
+        """
+
+        :param feed_group: Feed Group i.e 'user'
+        :param ids: list of ids of feed group i.e [123,456]
+        :param data: list of dictionaries
+        :return: http response, 201 if successful along with data posted.
+
+        **Example**::
+            personalization.upsert_data('user', [1, 2], [{"name": "Juniper", "hobbies": ["Playing", "Sleeping", "Eating"]},
+                                             {"name": "Ruby", "interests": ["Sunbeams", "Surprise Attacks"]}])
+        """
 
         if type(ids) != list:
             ids = [ids]
@@ -48,23 +62,31 @@ def upsert_data(self, item_type, ids, data):
         # format data to expected json blob
         data_json = {}
         for i in range(len(ids)):
-            data_json['%s:%s' % (item_type, ids[i])] = data[i]
+            data_json['%s:%s' % (feed_group, ids[i])] = data[i]
 
         response = self.post("meta", data={'data': data_json})
 
         return response
 
-    def select_data(self, item_type, ids):
+    def select_data(self, feed_group, ids):
+        """
+
+        :param feed_group: Feed Group i.e 'user'
+        :param ids: list of ids of feed group i.e [123,456]
+        :return: meta data as json blob
+
+        **Example**::
+            personalization.select_data('user', 1)
+            personalization.select_data('user', [1,2,3])
+        """
 
         if type(ids) != list:
             ids = [ids]
 
         foreign_ids = []
         for i in range(len(ids)):
-            foreign_ids.append('%s:%s' % (item_type, ids[i]))
+            foreign_ids.append('%s:%s' % (feed_group, ids[i]))
 
         response = self.get('meta', foreign_ids=foreign_ids)
 
         return response
-
-