Fixed assignees bug in private patterns

pyistanbul · May 7, 2013 · 61c6187 · 61c6187
1 parent a105d7e
commit 61c6187
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/web/dbpatterns/documents/mixins.py b/web/dbpatterns/documents/mixins.py
@@ -12,6 +12,7 @@ class DocumentMixin(object):
     def get_document(self):
         resource = DocumentResource()
         try:
-            return resource.obj_get(request=self.request, pk=self.kwargs.get("slug"))
-        except (TypeError, InvalidId, ImmediateHttpResponse):
+            return resource.obj_get(request=self.request,
+                                    pk=self.kwargs.get("slug"))
+        except (InvalidId, ImmediateHttpResponse):
             raise Http404("Document is not found.")
diff --git a/web/dbpatterns/documents/models.py b/web/dbpatterns/documents/models.py
@@ -106,10 +106,7 @@ def get_stars(self):
     @property
     def assignees(self):
         """Returns assigned users or empty list"""
-        try:
-            return self['assignees']
-        except KeyError:
-            return []
+        return self.get('assignees') or []
 
     @property
     def star_count(self):