From: Moritz Kiefer <moritz.kiefer@gmail.com>
Date: Sat, 2 Feb 2013 22:32:12 +0000 (+0100)
Subject: Add functions to add and remove users to/from aspects
X-Git-Url: https://vcs.fsf.org/?a=commitdiff_plain;h=5c2b6162f5e8df4b4593493c8da03a19d72b9f3d;p=diaspy.git

Add functions to add and remove users to/from aspects
---

diff --git a/diaspy/client.py b/diaspy/client.py
index 7a8fa73..8aa7e0c 100644
--- a/diaspy/client.py
+++ b/diaspy/client.py
@@ -178,3 +178,46 @@ class Client:
             posts.append(diaspy.models.Post(str(post['id']), self))
 
         return posts
+
+    def add_user_to_aspect(self, user_id, aspect_id):
+        """ this function adds a user to an aspect.
+
+        :param user_id: User ID
+        :type user_id: str
+        :param aspect_id: Aspect ID
+        :type aspect_id: str
+
+        """
+
+        data = {'authenticity_token': self.get_token(),
+                'aspect_id': aspect_id,
+                'person_id': user_id}
+
+        r = self.session.post(self.pod + '/aspect_memberships.json',
+                              data=data)
+
+        if r.status_code != 201:
+            raise Exception('wrong status code: ' + str(r.status_code))
+        return r.json()
+
+    def remove_user_from_aspect(self, user_id, aspect_id):
+        """ this function removes a user from an aspect.
+
+        :param user_id: User ID
+        :type user_id: str
+        :param aspect_id: Aspect ID
+        :type aspect_id: str
+
+        """
+
+        data = {'authenticity_token': self.get_token(),
+                'aspect_id': aspect_id,
+                'person_id': user_id}
+
+        r = self.session.delete(self.pod + '/aspect_memberships/42.json',
+                                data=data)
+
+        if r.status_code != 200:
+            raise Exception('wrong status code: ' + str(r.status_code))
+
+        return r.json()
diff --git a/diaspy/models.py b/diaspy/models.py
index 20001ce..44f2762 100644
--- a/diaspy/models.py
+++ b/diaspy/models.py
@@ -157,5 +157,3 @@ class Post:
                                         self.post_id,
                                         data=data,
                                         headers={'accept': 'application/json'})
-
-        return r