filter organizations admin models for related user organizations if not super_user

picampaign/organization/admin.py

@@ -2,6 +2,34 @@ from django.contrib import admin
 from picampaign.organization.models import (Organization, Group,
                                             FeedbackCategory)
 
-admin.site.register(Organization)
-admin.site.register(Group)
-admin.site.register(FeedbackCategory)
+
+class OrganizationAdmin(admin.ModelAdmin):
+    def get_queryset(self, request):
+        qs = super(OrganizationAdmin, self).get_queryset(request)
+        if request.user.is_superuser:
+            return qs
+        user_orgs = [x.id for x in request.user.organizations.all()]
+        return qs.filter(id__in=user_orgs)
+
+
+class GroupAdmin(admin.ModelAdmin):
+    def get_queryset(self, request):
+        qs = super(GroupAdmin, self).get_queryset(request)
+        if request.user.is_superuser:
+            return qs
+        user_orgs = [x.id for x in request.user.organizations.all()]
+        return qs.filter(organization__in=user_orgs)
+
+
+class CategoryAdmin(admin.ModelAdmin):
+    def get_queryset(self, request):
+        qs = super(CategoryAdmin, self).get_queryset(request)
+        if request.user.is_superuser:
+            return qs
+        user_orgs = [x.id for x in request.user.organizations.all()]
+        return qs.filter(organization__in=user_orgs)
+
+
+admin.site.register(Organization, OrganizationAdmin)
+admin.site.register(Group, GroupAdmin)
+admin.site.register(FeedbackCategory, CategoryAdmin)

picampaign/organization/models.py

@@ -7,7 +7,8 @@ class Organization(models.Model):
     id = models.AutoField(primary_key=True)
     name = models.CharField(max_length=64)
     sip_key = models.CharField(max_length=255)
-    users = models.ManyToManyField(User, null=True, blank=True)
+    users = models.ManyToManyField(User, null=True, blank=True,
+                                   related_name='organizations')
 
     def __unicode__(self):
         return self.name