Add document accesor method to display a document folders

mayan/apps/folders/__init__.py

@@ -48,6 +48,11 @@ class_permissions(Document, [
     PERMISSION_FOLDER_REMOVE_DOCUMENT,
 ])
 
+def document_folders(self):
+    return Folder.objects.filter(folderdocument__document=self)
+
+Document.add_to_class('folders', property(document_folders))
+
 endpoint = APIEndPoint('folders')
 endpoint.register_urls(api_urls)
 endpoint.add_endpoint('folder-list', _(u'Returns a list of all the folders.'))

mayan/apps/folders/views.py

@@ -230,7 +230,7 @@ def document_folder_list(request, document_id):
 
     return folder_list(
         request,
-        queryset=Folder.objects.filter(folderdocument__document=document),
+        queryset=document.folders.all(),
         extra_context={
             'title': _(u'folders containing: %s') % document,
             'object': document,