Added discussions to the notification API

udata/core/user/api.py

@@ -6,7 +6,7 @@
 
 from udata import search
 from udata.api import api, ModelAPI, ModelListAPI, API
-from udata.models import User, FollowUser, Reuse, Dataset, Issue
+from udata.models import User, FollowUser, Reuse, Dataset, Issue, Discussion
 
 from udata.core.dataset.api_fields import dataset_fields
 from udata.core.followers.api import FollowAPI
@@ -103,13 +103,13 @@ def get(self):
         user = current_user._get_current_object()
         notifications = []
 
-        orgs = [o for o in user.organizations if o.is_admin(user)]
+        orgs = [o for o in user.organizations if o.is_member(user)]
+        datasets = Dataset.objects.owned_by(user, *orgs)
+        reuses = Reuse.objects.owned_by(user, *orgs)
 
         # TODO: use polymorph field
 
         # Fetch user open issues
-        datasets = Dataset.objects.owned_by(user, *orgs)
-        reuses = Reuse.objects.owned_by(user, *orgs)
         for issue in Issue.objects(subject__in=list(datasets)+list(reuses)):
             notifications.append({
                 'type': 'issue',
@@ -124,6 +124,21 @@ def get(self):
                 }
             })
 
+        # Fetch user open discussions
+        for discussion in Discussion.objects(subject__in=list(datasets)+list(reuses)):
+            notifications.append({
+                'type': 'discussion',
+                'created_on': discussion.created,
+                'details': {
+                    'id': str(discussion.id),
+                    'title': discussion.title,
+                    'subject': {
+                        'id': str(discussion.subject.id),
+                        'type': discussion.subject.__class__.__name__.lower(),
+                    }
+                }
+            })
+
         # Fetch pending membership requests
         for org in orgs:
             for request in org.pending_requests: